[Authlib] Add integrations dirs (#15147)

Author: donbarbos

stubs/Authlib/@tests/stubtest_allowlist.txt

@@ -47,14 +47,15 @@ authlib.oidc.core.OpenIDImplicitGrant.validate_consent_request
 authlib.oidc.core.grants.OpenIDImplicitGrant.validate_consent_request
 authlib.oidc.core.grants.implicit.OpenIDImplicitGrant.validate_consent_request
 
-# Exclude integrations dirs:
-authlib.integrations.django_client
-authlib.integrations.django_oauth1
-authlib.integrations.django_oauth2
-authlib.integrations.flask_client
-authlib.integrations.flask_oauth1
-authlib.integrations.flask_oauth2
-authlib.integrations.httpx_client
-authlib.integrations.requests_client
-authlib.integrations.sqla_oauth2
-authlib.integrations.starlette_client
+# Exclude integrations dirs
+# Failed to import, getting ModuleNotFoundError for third-party libs:
+authlib.integrations.django_client.*
+authlib.integrations.django_oauth1.*
+authlib.integrations.django_oauth2.*
+authlib.integrations.flask_client.*
+authlib.integrations.flask_oauth1.*
+authlib.integrations.flask_oauth2.*
+authlib.integrations.httpx_client.*
+authlib.integrations.requests_client.*
+authlib.integrations.sqla_oauth2.*
+authlib.integrations.starlette_client.*

stubs/Authlib/METADATA.toml

@@ -1,4 +1,3 @@
 version = "1.6.6"
 upstream_repository = "https://github.com/lepture/authlib"
 requires = ["cryptography"]
-partial_stub = true

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

@@ -10,4 +10,4 @@ class FrameworkIntegration:
     def clear_state_data(self, session, state): ...
     def update_token(self, token, refresh_token=None, access_token=None) -> None: ...
     @staticmethod
-    def load_config(oauth, name, params) -> None: ...
+    def load_config(oauth, name, params): ...

stubs/Authlib/authlib/integrations/django_client/__init__.pyi

@@ -0,0 +1,10 @@
+from ..base_client import BaseOAuth, OAuthError as OAuthError
+from .apps import DjangoOAuth1App as DjangoOAuth1App, DjangoOAuth2App as DjangoOAuth2App
+from .integration import DjangoIntegration as DjangoIntegration, token_update as token_update
+
+class OAuth(BaseOAuth):
+    oauth1_client_cls = DjangoOAuth1App
+    oauth2_client_cls = DjangoOAuth2App
+    framework_integration_cls = DjangoIntegration
+
+__all__ = ["OAuth", "DjangoOAuth1App", "DjangoOAuth2App", "DjangoIntegration", "token_update", "OAuthError"]

stubs/Authlib/authlib/integrations/django_client/apps.pyi

@@ -0,0 +1,14 @@
+from ..base_client import BaseApp, OAuth1Mixin, OAuth2Mixin, OpenIDMixin
+from ..requests_client import OAuth1Session, OAuth2Session
+
+class DjangoAppMixin:
+    def save_authorize_data(self, request, **kwargs) -> None: ...
+    def authorize_redirect(self, request, redirect_uri=None, **kwargs): ...
+
+class DjangoOAuth1App(DjangoAppMixin, OAuth1Mixin, BaseApp):
+    client_cls = OAuth1Session
+    def authorize_access_token(self, request, **kwargs): ...
+
+class DjangoOAuth2App(DjangoAppMixin, OAuth2Mixin, OpenIDMixin, BaseApp):
+    client_cls = OAuth2Session
+    def authorize_access_token(self, request, **kwargs): ...

stubs/Authlib/authlib/integrations/django_client/integration.pyi

@@ -0,0 +1,11 @@
+from _typeshed import Incomplete
+
+from ..base_client import FrameworkIntegration
+
+# actual type is django.dispatch.Signal
+token_update: Incomplete
+
+class DjangoIntegration(FrameworkIntegration):
+    def update_token(self, token, refresh_token=None, access_token=None) -> None: ...
+    @staticmethod
+    def load_config(oauth, name, params): ...

stubs/Authlib/authlib/integrations/django_oauth1/__init__.pyi

@@ -0,0 +1,4 @@
+from .authorization_server import BaseServer as BaseServer, CacheAuthorizationServer as CacheAuthorizationServer
+from .resource_protector import ResourceProtector as ResourceProtector
+
+__all__ = ["BaseServer", "CacheAuthorizationServer", "ResourceProtector"]

stubs/Authlib/authlib/integrations/django_oauth1/authorization_server.pyi

@@ -0,0 +1,26 @@
+import logging
+from _typeshed import Incomplete
+
+from authlib.oauth1 import AuthorizationServer as _AuthorizationServer, OAuth1Request, TemporaryCredential
+
+log: logging.Logger
+
+class BaseServer(_AuthorizationServer):
+    token_generator: Incomplete
+    client_model: Incomplete
+    token_model: Incomplete
+    SUPPORTED_SIGNATURE_METHODS: Incomplete
+    def __init__(self, client_model, token_model, token_generator=None) -> None: ...
+    def get_client_by_id(self, client_id): ...
+    def exists_nonce(self, nonce, request) -> bool: ...
+    def create_token_credential(self, request): ...
+    def check_authorization_request(self, request) -> OAuth1Request: ...
+    def create_oauth1_request(self, request) -> OAuth1Request: ...
+    def handle_response(self, status_code, payload, headers): ...
+
+class CacheAuthorizationServer(BaseServer):
+    def __init__(self, client_model, token_model, token_generator=None) -> None: ...
+    def create_temporary_credential(self, request) -> TemporaryCredential: ...
+    def get_temporary_credential(self, request) -> TemporaryCredential | None: ...
+    def delete_temporary_credential(self, request) -> None: ...
+    def create_authorization_verifier(self, request) -> str: ...

stubs/Authlib/authlib/integrations/django_oauth1/nonce.pyi

@@ -0,0 +1 @@
+def exists_nonce_in_cache(nonce, request, timeout) -> bool: ...

stubs/Authlib/authlib/integrations/django_oauth1/resource_protector.pyi

@@ -0,0 +1,14 @@
+from _typeshed import Incomplete
+
+from authlib.oauth1 import ResourceProtector as _ResourceProtector
+
+class ResourceProtector(_ResourceProtector):
+    client_model: Incomplete
+    token_model: Incomplete
+    SUPPORTED_SIGNATURE_METHODS: Incomplete
+    def __init__(self, client_model, token_model) -> None: ...
+    def get_client_by_id(self, client_id): ...
+    def get_token_credential(self, request): ...
+    def exists_nonce(self, nonce, request) -> bool: ...
+    def acquire_credential(self, request): ...
+    def __call__(self, realm=None): ...