Add type annotation for HTTPError

omarryhan · Jun 1, 2024 · 2698370 · 2698370
1 parent b6d8e56
commit 2698370
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/aiogoogle/excs.py b/aiogoogle/excs.py
@@ -1,5 +1,11 @@
+from __future__ import annotations
+from typing import TYPE_CHECKING
+
 __all__ = ["AiogoogleError", "ValidationError", "HTTPError", "AuthError"]
 
+if TYPE_CHECKING:
+    from .models import Response
+
 
 class AiogoogleError(Exception):
     pass
@@ -14,7 +20,7 @@ class ValidationError(AiogoogleError):
 
 
 class HTTPError(AiogoogleError):
-    def __init__(self, msg, req=None, res=None):
+    def __init__(self, msg, req, res: Response):
         self.req = req
         self.res = res
         super().__init__(msg)

diff --git a/aiogoogle/models.py b/aiogoogle/models.py
@@ -286,7 +286,7 @@ class Response:
 
     def __init__(
         self,
-        status_code=None,
+        status_code: int,
         headers=None,
         url=None,
         json=None,