[Authlib] Update return types (#15153)

* Remove return types for not implemented methods
* Add some obvious return types
* Fix some incorrect types found

Author: donbarbos

stubs/Authlib/authlib/common/urls.pyi

@@ -13,7 +13,7 @@ def url_decode(query: str) -> _ExplodedQueryString: ...
 def add_params_to_qs(query: str, params: _ExplodedQueryString) -> str: ...
 def add_params_to_uri(uri: str, params: _ExplodedQueryString, fragment: bool = False): ...
 def quote(s: str, safe: bytes = b"/") -> str: ...
-def unquote(s: str) -> str: ...
+def unquote(s: str | bytes) -> str: ...
 def quote_url(s: str) -> str: ...
 def extract_params(raw: dict[str, str] | _ExplodedQueryString) -> _ExplodedQueryString: ...
 def is_valid_url(url: str, fragments_allowed: bool = True) -> bool: ...

stubs/Authlib/authlib/integrations/base_client/async_app.pyi

@@ -1,3 +1,4 @@
+from _typeshed import Incomplete
 from logging import Logger
 
 from authlib.integrations.base_client.sync_app import OAuth1Base, OAuth2Base
@@ -8,11 +9,11 @@ __all__ = ["AsyncOAuth1Mixin", "AsyncOAuth2Mixin"]
 
 class AsyncOAuth1Mixin(OAuth1Base):
     async def request(self, method, url, token=None, **kwargs): ...
-    async def create_authorization_url(self, redirect_uri=None, **kwargs): ...
+    async def create_authorization_url(self, redirect_uri=None, **kwargs) -> dict[Incomplete, Incomplete]: ...
     async def fetch_access_token(self, request_token=None, **kwargs): ...
 
 class AsyncOAuth2Mixin(OAuth2Base):
-    async def load_server_metadata(self): ...
+    async def load_server_metadata(self) -> dict[Incomplete, Incomplete]: ...
     async def request(self, method, url, token=None, **kwargs): ...
-    async def create_authorization_url(self, redirect_uri=None, **kwargs): ...
+    async def create_authorization_url(self, redirect_uri=None, **kwargs) -> dict[Incomplete, Incomplete]: ...
     async def fetch_access_token(self, redirect_uri=None, **kwargs): ...

stubs/Authlib/authlib/integrations/base_client/async_openid.pyi

@@ -1,6 +1,8 @@
+from authlib.oidc.core.claims import UserInfo
+
 __all__ = ["AsyncOpenIDMixin"]
 
 class AsyncOpenIDMixin:
     async def fetch_jwk_set(self, force: bool = False): ...
-    async def userinfo(self, **kwargs): ...
-    async def parse_id_token(self, token, nonce, claims_options=None, claims_cls=None, leeway: int = 120): ...
+    async def userinfo(self, **kwargs) -> UserInfo: ...
+    async def parse_id_token(self, token, nonce, claims_options=None, claims_cls=None, leeway: int = 120) -> UserInfo: ...

stubs/Authlib/authlib/integrations/base_client/framework_integration.pyi

@@ -8,6 +8,6 @@ class FrameworkIntegration:
     def get_state_data(self, session, state): ...
     def set_state_data(self, session, state, data): ...
     def clear_state_data(self, session, state): ...
-    def update_token(self, token, refresh_token=None, access_token=None) -> None: ...
+    def update_token(self, token, refresh_token=None, access_token=None): ...
     @staticmethod
     def load_config(oauth, name, params): ...

stubs/Authlib/authlib/integrations/base_client/registry.pyi

@@ -14,6 +14,6 @@ class BaseOAuth:
     def __init__(self, cache=None, fetch_token=None, update_token=None) -> None: ...
     def create_client(self, name): ...
     def register(self, name, overwrite: bool = False, **kwargs): ...
-    def generate_client_kwargs(self, name, overwrite, **kwargs): ...
+    def generate_client_kwargs(self, name, overwrite, **kwargs) -> dict[Incomplete, Incomplete]: ...
     def load_config(self, name, params): ...
     def __getattr__(self, key): ...

stubs/Authlib/authlib/integrations/base_client/sync_app.pyi

@@ -50,7 +50,7 @@ class OAuth1Base:
 
 class OAuth1Mixin(_RequestMixin, OAuth1Base):
     def request(self, method, url, token=None, **kwargs): ...
-    def create_authorization_url(self, redirect_uri=None, **kwargs): ...
+    def create_authorization_url(self, redirect_uri=None, **kwargs) -> dict[Incomplete, Incomplete]: ...
     def fetch_access_token(self, request_token=None, **kwargs): ...
 
 class OAuth2Base:
@@ -91,6 +91,6 @@ class OAuth2Base:
 
 class OAuth2Mixin(_RequestMixin, OAuth2Base):
     def request(self, method, url, token=None, **kwargs): ...
-    def load_server_metadata(self): ...
-    def create_authorization_url(self, redirect_uri=None, **kwargs): ...
+    def load_server_metadata(self) -> dict[Incomplete, Incomplete]: ...
+    def create_authorization_url(self, redirect_uri=None, **kwargs) -> dict[Incomplete, Incomplete]: ...
     def fetch_access_token(self, redirect_uri=None, **kwargs): ...

stubs/Authlib/authlib/integrations/base_client/sync_openid.pyi

@@ -1,5 +1,7 @@
+from authlib.oidc.core.claims import UserInfo
+
 class OpenIDMixin:
     def fetch_jwk_set(self, force: bool = False): ...
-    def userinfo(self, **kwargs): ...
-    def parse_id_token(self, token, nonce, claims_options=None, claims_cls=None, leeway: int = 120): ...
+    def userinfo(self, **kwargs) -> UserInfo: ...
+    def parse_id_token(self, token, nonce, claims_options=None, claims_cls=None, leeway: int = 120) -> UserInfo | None: ...
     def create_load_key(self): ...

stubs/Authlib/authlib/jose/drafts/_jwe_algorithms.pyi

@@ -1,6 +1,6 @@
 from _typeshed import Incomplete
 from collections.abc import Iterable
-from typing import ClassVar
+from typing import ClassVar, Final
 
 from authlib.jose.rfc7516 import JWEAlgorithmWithTagAwareKeyAgreement
 
@@ -13,17 +13,19 @@ class ECDH1PUAlgorithm(JWEAlgorithmWithTagAwareKeyAgreement):
     aeskw: Incomplete
     def __init__(self, key_size=None) -> None: ...
     def prepare_key(self, raw_data): ...
-    def generate_preset(self, enc_alg, key): ...
+    def generate_preset(self, enc_alg, key) -> dict[str, Incomplete]: ...
     def compute_shared_key(self, shared_key_e, shared_key_s): ...
-    def compute_fixed_info(self, headers, bit_size, tag): ...
-    def compute_derived_key(self, shared_key, fixed_info, bit_size): ...
-    def deliver_at_sender(self, sender_static_key, sender_ephemeral_key, recipient_pubkey, headers, bit_size, tag): ...
-    def deliver_at_recipient(self, recipient_key, sender_static_pubkey, sender_ephemeral_pubkey, headers, bit_size, tag): ...
-    def generate_keys_and_prepare_headers(self, enc_alg, key, sender_key, preset=None): ...
+    def compute_fixed_info(self, headers, bit_size, tag) -> bytes: ...
+    def compute_derived_key(self, shared_key, fixed_info, bit_size) -> bytes: ...
+    def deliver_at_sender(self, sender_static_key, sender_ephemeral_key, recipient_pubkey, headers, bit_size, tag) -> bytes: ...
+    def deliver_at_recipient(
+        self, recipient_key, sender_static_pubkey, sender_ephemeral_pubkey, headers, bit_size, tag
+    ) -> bytes: ...
+    def generate_keys_and_prepare_headers(self, enc_alg, key, sender_key, preset=None) -> dict[str, Incomplete]: ...
     def agree_upon_key_and_wrap_cek(self, enc_alg, headers, key, sender_key, epk, cek, tag): ...
-    def wrap(self, enc_alg, headers, key, sender_key, preset=None): ...
-    def unwrap(self, enc_alg, ek, headers, key, sender_key, tag=None): ...
+    def wrap(self, enc_alg, headers, key, sender_key, preset=None) -> dict[str, Incomplete]: ...
+    def unwrap(self, enc_alg, ek, headers, key, sender_key, tag=None) -> bytes: ...
 
-JWE_DRAFT_ALG_ALGORITHMS: Incomplete
+JWE_DRAFT_ALG_ALGORITHMS: Final[list[ECDH1PUAlgorithm]]
 
 def register_jwe_alg_draft(cls) -> None: ...

stubs/Authlib/authlib/jose/drafts/_jwe_enc_cryptography.pyi

@@ -9,5 +9,5 @@ class C20PEncAlgorithm(JWEEncAlgorithm):
     key_size: Incomplete
     CEK_SIZE: Incomplete
     def __init__(self, key_size) -> None: ...
-    def encrypt(self, msg, aad, iv, key): ...
-    def decrypt(self, ciphertext, aad, iv, tag, key): ...
+    def encrypt(self, msg, aad, iv, key) -> tuple[bytes, bytes]: ...
+    def decrypt(self, ciphertext, aad, iv, tag, key) -> bytes: ...

stubs/Authlib/authlib/jose/jwk.pyi

@@ -1,2 +1,7 @@
+from _typeshed import Incomplete
+from typing_extensions import deprecated
+
+@deprecated("Please use `JsonWebKey` directly.")
 def loads(obj, kid=None): ...
-def dumps(key, kty=None, **params): ...
+@deprecated("Please use `JsonWebKey` directly.")
+def dumps(key, kty=None, **params) -> dict[Incomplete, Incomplete]: ...