Add the missing typehints

Author: ArtyomVancyan

src/fastapi_oauth2/client.py

@@ -22,7 +22,7 @@ def __init__(
             client_secret: str,
             redirect_uri: Optional[str] = None,
             scope: Optional[Sequence[str]] = None,
-    ):
+    ) -> None:
         self.backend = backend
         self.client_id = client_id
         self.client_secret = client_secret

src/fastapi_oauth2/config.py

@@ -22,7 +22,7 @@ def __init__(
             jwt_expires: Union[int, str] = 900,
             jwt_algorithm: str = "HS256",
             clients: List[OAuth2Client] = None,
-    ):
+    ) -> None:
         if allow_http:
             os.environ["OAUTHLIB_INSECURE_TRANSPORT"] = "1"
         self.allow_http = allow_http

src/fastapi_oauth2/core.py

@@ -26,17 +26,17 @@ class OAuth2LoginError(HTTPException):
 class OAuth2Strategy(BaseStrategy):
     """Dummy strategy for using the `BaseOAuth2.user_data` method."""
 
-    def request_data(self, merge=True):
+    def request_data(self, merge=True) -> Dict[str, Any]:
         return {}
 
-    def absolute_uri(self, path=None):
+    def absolute_uri(self, path=None) -> str:
         return path
 
-    def get_setting(self, name):
-        return None
+    def get_setting(self, name) -> Any:
+        """Mocked setting method."""
 
     @staticmethod
-    def get_json(url, method='GET', *args, **kwargs):
+    def get_json(url, method='GET', *args, **kwargs) -> httpx.Response:
         return httpx.request(method, url, *args, **kwargs)
 
 

src/fastapi_oauth2/middleware.py

@@ -104,7 +104,7 @@ def __init__(
             Auth.register_client(client)
         self.callback = callback
 
-    async def authenticate(self, request: Request) -> Optional[Tuple["Auth", "User"]]:
+    async def authenticate(self, request: Request) -> Optional[Tuple[Auth, User]]:
         authorization = request.headers.get(
             "Authorization",
             request.cookies.get("Authorization"),

src/fastapi_oauth2/security.py

@@ -1,13 +1,19 @@
+from typing import Callable
+from typing import Optional
+from typing import Type
+
 from fastapi.security import OAuth2 as FastAPIOAuth2
 from fastapi.security import OAuth2AuthorizationCodeBearer as FastAPICodeBearer
 from fastapi.security import OAuth2PasswordBearer as FastAPIPasswordBearer
 from starlette.datastructures import Headers
 from starlette.requests import Request
 
 
-def use_cookie(cls: FastAPIOAuth2):
-    def _use_cookie(*args, **kwargs):
-        async def __call__(self, request: Request):
+def use_cookies(cls: Type[FastAPIOAuth2]) -> Callable[[...], FastAPIOAuth2]:
+    """OAuth2 classes wrapped with this decorator will use cookies for the Authorization header."""
+
+    def _use_cookies(*args, **kwargs) -> FastAPIOAuth2:
+        async def __call__(self: FastAPIOAuth2, request: Request) -> Optional[str]:
             authorization = request.headers.get("Authorization", request.cookies.get("Authorization"))
             if authorization:
                 request._headers = Headers({**request.headers, "Authorization": authorization})
@@ -16,19 +22,19 @@ async def __call__(self, request: Request):
         cls.__call__ = __call__
         return cls(*args, **kwargs)
 
-    return _use_cookie
+    return _use_cookies
 
 
-@use_cookie
+@use_cookies
 class OAuth2(FastAPIOAuth2):
-    ...
+    """Wrapper class of the `fastapi.security.OAuth2` class."""
 
 
-@use_cookie
+@use_cookies
 class OAuth2PasswordBearer(FastAPIPasswordBearer):
-    ...
+    """Wrapper class of the `fastapi.security.OAuth2PasswordBearer` class."""
 
 
-@use_cookie
+@use_cookies
 class OAuth2AuthorizationCodeBearer(FastAPICodeBearer):
-    ...
+    """Wrapper class of the `fastapi.security.OAuth2AuthorizationCodeBearer` class."""