wip: Fix types

Author: sarahmonod

.pre-commit-config.yaml

@@ -27,11 +27,14 @@ repos:
     hooks:
     - id: pyupgrade
       args: [--py37-plus]
-  #- repo: https://github.com/pre-commit/mirrors-mypy
-  #  rev: v0.910
-  #  hooks:
-  #  - id: mypy
-  #    exclude: ^(docs/|tests/)
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v1.5.1
+    hooks:
+    - id: mypy
+      additional_dependencies:
+      - types-python-dateutil
+      - types-requests
+      exclude: ^(docs/|tests/)
   - repo: https://github.com/jorisroovers/gitlint
     rev: v0.19.1
     hooks:

pyproject.toml

@@ -86,3 +86,10 @@ exclude = '''
   )/
 )
 '''
+
+[tool.mypy]
+files = ["."]
+exclude = [
+    "^docs/",
+    "^tests/",
+]

src/github3/checks.py

@@ -45,7 +45,7 @@ def _update_attributes(self, pull):
     def _repr(self):
         return f"<CheckPullRequest [#{self.number}]>"
 
-    def to_pull(self):
+    def to_pull(self, conditional: bool = False):
         """Retrieve a full PullRequest object for this CheckPullRequest.
 
         :returns:
@@ -119,7 +119,7 @@ def _repr(self):
             self.name, str(self.owner["login"])
         )
 
-    def to_app(self):
+    def to_app(self, conditional: bool = False) -> models.GitHubCore:
         """Retrieve a full App object for this CheckApp.
 
         :returns:

src/github3/events.py

@@ -41,7 +41,7 @@ def _update_attributes(self, user):
         self.login = user["login"]
         self._api = self.url = user["url"]
 
-    def to_user(self):
+    def to_user(self, conditional: bool = False) -> models.GitHubCore:
         """Retrieve a full User object for this EventUser.
 
         :returns:
@@ -93,7 +93,7 @@ def _update_attributes(self, org):
         self.login = org["login"]
         self._api = self.url = org["url"]
 
-    def to_org(self):
+    def to_org(self, conditional: bool = False) -> models.GitHubCore:
         """Retrieve a full Organization object for this EventOrganization.
 
         :returns:
@@ -148,7 +148,7 @@ def _update_attributes(self, pull):
         self.locked = pull["locked"]
         self._api = self.url = pull["url"]
 
-    def to_pull(self):
+    def to_pull(self, conditional: bool = False) -> models.GitHubCore:
         """Retrieve a full PullRequest object for this EventPullRequest.
 
         :returns:
@@ -258,7 +258,9 @@ def _update_attributes(self, comment):
         self.updated_at = self._strptime(comment["updated_at"])
         self.user = users.ShortUser(comment["user"], self)
 
-    def to_review_comment(self):
+    def to_review_comment(
+        self, conditional: bool = False
+    ) -> models.GitHubCore:
         """Retrieve a full ReviewComment object for this EventReviewComment.
 
         :returns:
@@ -269,7 +271,7 @@ def to_review_comment(self):
         from . import pulls
 
         comment = self._json(self._get(self._api), 200)
-        return pulls.ReviewComment(comment, self)
+        return pulls.ReviewComment(comment, self.session)
 
     refresh = to_review_comment
 
@@ -285,7 +287,7 @@ def _update_attributes(self, issue):
         self.locked = issue["locked"]
         self._api = self.url = issue["url"]
 
-    def to_issue(self):
+    def to_issue(self, conditional: bool = False) -> models.GitHubCore:
         """Retrieve a full Issue object for this EventIssue."""
         from . import issues
 
@@ -352,7 +354,9 @@ def _update_attributes(self, comment):
         self.updated_at = self._strptime(comment["updated_at"])
         self.user = users.ShortUser(comment["user"], self)
 
-    def to_issue_comment(self):
+    def to_issue_comment(
+        self, conditional: bool = False
+    ) -> models.GitHubCore:
         """Retrieve the full IssueComment object for this comment.
 
         :returns:

src/github3/gists/gist.py

@@ -1,4 +1,5 @@
 """This module contains the Gist, ShortGist, and GistFork objects."""
+import typing as t
 from json import dumps
 
 from . import comment
@@ -31,7 +32,9 @@ class _Gist(models.GitHubCore):
     """
 
     class_name = "_Gist"
-    _file_class = gistfile.ShortGistFile
+    _file_class: t.Type[
+        t.Union[gistfile.ShortGistFile, gistfile.GistFile]
+    ] = gistfile.ShortGistFile
 
     def _update_attributes(self, gist):
         self.comments_count = gist["comments"]
@@ -265,7 +268,7 @@ def _update_attributes(self, fork):
     def _repr(self):
         return f"<GistFork [{self.id}]>"
 
-    def to_gist(self):
+    def to_gist(self, conditional: bool = False) -> models.GitHubCore:
         """Retrieve the full Gist representation of this fork.
 
         :returns:

src/github3/gists/history.py

@@ -51,7 +51,7 @@ class GistHistory(models.GitHubCore):
     def _update_attributes(self, history) -> None:
         self.url = self._api = history["url"]
         self.version = history["version"]
-        self.user = users.ShortUser(history["user"], self)
+        self.user = users.ShortUser(history["user"], self.session)
         self.change_status = history["change_status"]
         self.additions = self.change_status.get("additions")
         self.deletions = self.change_status.get("deletions")

src/github3/git.py

@@ -297,42 +297,6 @@ def _repr(self):
         return f"<Tag [{self.tag}]>"
 
 
-class CommitTree(models.GitHubCore):
-    """This object represents the abbreviated tree data in a commit.
-
-    The API returns different representations of different objects. When
-    representing a :class:`~github3.git.ShortCommit` or
-    :class:`~github3.git.Commit`, the API returns an abbreviated
-    representation of a git tree.
-
-    This object has the following attributes:
-
-    .. attribute:: sha
-
-        The SHA1 of this tree in the git repository.
-    """
-
-    def _update_attributes(self, tree):
-        self._api = tree["url"]
-        self.sha = tree["sha"]
-
-    def _repr(self):
-        return f"<CommitTree [{self.sha}]>"
-
-    def to_tree(self):
-        """Retrieve a full Tree object for this CommitTree.
-
-        :returns:
-            The full git data about this tree
-        :rtype:
-            :class:`~github3.git.Tree`
-        """
-        json = self._json(self._get(self._api), 200)
-        return self._instance_or_null(Tree, json)
-
-    refresh = to_tree
-
-
 class Tree(models.GitHubCore):
     """This represents a tree object from a git repository.
 
@@ -382,6 +346,42 @@ def recurse(self):
         return self._instance_or_null(Tree, json)
 
 
+class CommitTree(models.GitHubCore):
+    """This object represents the abbreviated tree data in a commit.
+
+    The API returns different representations of different objects. When
+    representing a :class:`~github3.git.ShortCommit` or
+    :class:`~github3.git.Commit`, the API returns an abbreviated
+    representation of a git tree.
+
+    This object has the following attributes:
+
+    .. attribute:: sha
+
+        The SHA1 of this tree in the git repository.
+    """
+
+    def _update_attributes(self, tree):
+        self._api = tree["url"]
+        self.sha = tree["sha"]
+
+    def _repr(self):
+        return f"<CommitTree [{self.sha}]>"
+
+    def to_tree(self, conditional: bool = False) -> models.GitHubCore:
+        """Retrieve a full Tree object for this CommitTree.
+
+        :returns:
+            The full git data about this tree
+        :rtype:
+            :class:`~github3.git.Tree`
+        """
+        json = self._json(self._get(self._api), 200)
+        return self._instance_or_null(Tree, json)
+
+    refresh = to_tree
+
+
 class Hash(models.GitHubCore):
     """This is used to represent the elements of a tree.
 

src/github3/github.py

@@ -3,7 +3,7 @@
 import re
 import typing as t
 
-import uritemplate
+import uritemplate  # type: ignore
 
 from . import apps
 from . import auths
@@ -544,7 +544,7 @@ def authorize(
     @requires_auth
     def blocked_users(
         self, number: int = -1, etag: t.Optional[str] = None
-    ) -> t.Generator[users.ShortUser, None, None]:
+    ) -> t.Iterator[users.ShortUser]:
         """Iterate over the users blocked by this organization.
 
         .. versionadded:: 2.1.0

src/github3/issues/issue.py

@@ -1,7 +1,7 @@
 """Module containing the Issue logic."""
 from json import dumps
 
-from uritemplate import URITemplate
+from uritemplate import URITemplate  # type: ignore
 
 from . import comment
 from . import event

src/github3/models.py

@@ -16,9 +16,6 @@
 LOG = logging.getLogger(__package__)
 
 
-T = t.TypeVar("T")
-
-
 class GitHubCore:
     """The base object for all objects that require a session.
 
@@ -28,7 +25,7 @@ class GitHubCore:
     """
 
     _ratelimit_resource = "core"
-    _refresh_to: t.Optional["GitHubCore"] = None
+    _refresh_to: t.Optional[t.Type["GitHubCore"]] = None
 
     def __init__(self, json, session: session.GitHubSession):
         """Initialize our basic object.
@@ -240,26 +237,28 @@ def _api(self):
             value += f"?{self._uri.query}"
         return value
 
-    @staticmethod
-    def _uri_parse(uri):
-        return requests.compat.urlparse(uri)
-
     @_api.setter
     def _api(self, uri):
         if uri:
             self._uri = self._uri_parse(uri)
         self.url = uri
 
+    @staticmethod
+    def _uri_parse(uri):
+        return requests.compat.urlparse(uri)
+
     def _iter(
         self,
         count: int,
         url: str,
-        cls: t.Type[T],
-        params: t.Optional[t.Mapping[str, t.Optional[str]]] = None,
+        cls: t.Type["GitHubCore"],
+        params: t.Optional[
+            t.MutableMapping[str, t.Union[str, int, None]]
+        ] = None,
         etag: t.Optional[str] = None,
         headers: t.Optional[t.Mapping[str, str]] = None,
         list_key: t.Optional[str] = None,
-    ) -> "structs.GitHubIterator[T]":
+    ) -> "structs.GitHubIterator":
         """Generic iterator for this project.
 
         :param int count: How many items to return.
@@ -276,7 +275,7 @@ def _iter(
         from .structs import GitHubIterator
 
         return GitHubIterator(
-            count, url, cls, self, params, etag, headers, list_key
+            count, url, cls, self.session, params, etag, headers, list_key
         )
 
     @property
@@ -329,7 +328,7 @@ def refresh(self, conditional: bool = False) -> "GitHubCore":
                 self._json_data = json
                 self._update_attributes(json)
             else:
-                return self._refresh_to(json, self)
+                return self._refresh_to(json, self.session)
         return self
 
     def new_session(self):

src/github3/orgs.py

@@ -2,7 +2,7 @@
 import typing as t
 from json import dumps
 
-from uritemplate import URITemplate
+from uritemplate import URITemplate  # type: ignore
 
 from . import models
 from . import users
@@ -211,7 +211,7 @@ def permissions_for(
         headers = {"Accept": "application/vnd.github.v3.repository+json"}
         url = self._build_url("repos", repository, base_url=self._api)
         json = self._json(self._get(url, headers=headers), 200)
-        return ShortRepositoryWithPermissions(json, self)
+        return ShortRepositoryWithPermissions(json, self.session)
 
     @requires_auth
     def repositories(self, number=-1, etag=None):
@@ -491,7 +491,7 @@ def add_repository(self, repository, team_id):  # FIXME(jlk): add perms
     @requires_auth
     def blocked_users(
         self, number: int = -1, etag: t.Optional[str] = None
-    ) -> t.Generator[users.ShortUser, None, None]:
+    ) -> t.Iterator[users.ShortUser]:
         """Iterate over the users blocked by this organization.
 
         .. versionadded:: 2.1.0
@@ -749,7 +749,7 @@ def create_team(
                 getattr(r, "full_name", r) for r in (repo_names or [])
             ],
             "maintainers": [
-                getattr(m, "login", m) for m in (maintainers or [])
+                str(getattr(m, "login", m)) for m in (maintainers or [])
             ],
             "permission": permission,
             "privacy": privacy,

src/github3/pulls.py

@@ -1,7 +1,7 @@
 """This module contains all the classes relating to pull requests."""
 from json import dumps
 
-from uritemplate import URITemplate
+from uritemplate import URITemplate  # type: ignore
 
 from . import models
 from . import users