Add typing

oauthlib · auvipy · Oct 17, 2024 · Oct 2, 2024 · Oct 14, 2024 · Oct 14, 2024
commit ecdbcd3031f0b61ae7a045c56887630ec5da24ae
diff --git a/oauthlib/oauth2/rfc8628/endpoints/device_authorization.py b/oauthlib/oauth2/rfc8628/endpoints/device_authorization.py
@@ -7,6 +7,7 @@
 """
 import json
 import logging
+from typing import Callable
 
 from oauthlib.common import Request, generate_token
 from oauthlib.oauth2.rfc6749 import errors
@@ -39,7 +40,7 @@ def __init__(
         expires_in=1800,
         interval=None,
         verification_uri_complete=None,
-        user_code_generator = None
+        user_code_generator: Callable[[None], str] = None
     ):
         """
         :param request_validator: An instance of RequestValidator.

diff --git a/oauthlib/oauth2/rfc8628/pre_configured.py b/oauthlib/oauth2/rfc8628/pre_configured.py
@@ -1,13 +1,14 @@
 from oauthlib.oauth2.rfc8628.endpoints.device_authorization import (
     DeviceAuthorizationEndpoint,
 )
+from typing import Callable
 
 
 class DeviceApplicationServer(DeviceAuthorizationEndpoint):
 
     """An all-in-one endpoint featuring Authorization code grant and Bearer tokens."""
 
-    def __init__(self, request_validator, verification_uri, user_code_generator = None, **kwargs):
+    def __init__(self, request_validator, verification_uri, user_code_generator: Callable[[None], str] = None, **kwargs):
         """Construct a new web application server.
 
         :param request_validator: An implementation of