Merge pull request #1215 from PokestarFan/add-types-message

(#1164) Add types to reddit/message.py

praw/models/reddit/message.py

@@ -1,10 +1,14 @@
 """Provide the Message class."""
+from typing import Any, Dict, TypeVar
+
 from ...const import API_PATH
 from .base import RedditBase
 from .mixins import FullnameMixin, InboxableMixin, ReplyableMixin
 from .redditor import Redditor
 from .subreddit import Subreddit
 
+Reddit = TypeVar("Reddit")
+
 
 class Message(InboxableMixin, ReplyableMixin, FullnameMixin, RedditBase):
     """A class for private messages.
@@ -42,7 +46,7 @@ class Message(InboxableMixin, ReplyableMixin, FullnameMixin, RedditBase):
     STR_FIELD = "id"
 
     @classmethod
-    def parse(cls, data, reddit):
+    def parse(cls, data: Dict[str, Any], reddit: Reddit):
         """Return an instance of Message or SubredditMessage from ``data``.
 
         :param data: The structured data.
@@ -76,7 +80,7 @@ def _kind(self):
         """Return the class's kind."""
         return self._reddit.config.kinds["message"]
 
-    def __init__(self, reddit, _data):
+    def __init__(self, reddit: Reddit, _data: Dict[str, Any]):
         """Construct an instance of the Message object."""
         super().__init__(reddit, _data=_data)
         self._fetched = True