Add type stubs for google-auth-library-python

Author: Tesla2000

stubs/google/auth/__init__.pyi

@@ -0,0 +1,5 @@
+from google.auth._default import default as default, load_credentials_from_dict as load_credentials_from_dict, load_credentials_from_file as load_credentials_from_file
+
+__all__ = ['default', 'load_credentials_from_file', 'load_credentials_from_dict']
+
+class Python37DeprecationWarning(DeprecationWarning): ...

stubs/google/auth/_agent_identity_utils.pyi

@@ -0,0 +1,10 @@
+from typing import Any, Optional
+
+CRYPTOGRAPHY_NOT_FOUND_ERROR: str
+
+def get_agent_identity_certificate_path() -> Optional[str]: ...
+def get_and_parse_agent_identity_certificate() -> Optional[Any]: ...
+def parse_certificate(cert_bytes: bytes) -> Any: ...
+def calculate_certificate_fingerprint(cert: Any) -> str: ...
+def should_request_bound_token(cert: Any) -> bool: ...
+def get_cached_cert_fingerprint(cached_cert: Optional[bytes]) -> str: ...

stubs/google/auth/_cache.pyi

@@ -0,0 +1,15 @@
+from typing import Hashable, TypeVar
+
+_Key = TypeVar("_Key", bound=Hashable)
+_Value = TypeVar("_Value")
+
+class LRUCache(dict[_Key, _Value]):
+    maxsize: int
+
+    def __init__(self, maxsize: int) -> None: ...
+    def clear(self) -> None: ...
+    def __getitem__(self, key: _Key) -> _Value: ...
+    def __setitem__(self, key: _Key, value: _Value) -> None: ...
+    def __delitem__(self, key: _Key) -> None: ...
+    def popitem(self) -> tuple[_Key, _Value]: ...
+    def _update(self, key: _Key) -> None: ...

stubs/google/auth/_cloud_sdk.pyi

@@ -0,0 +1,8 @@
+from typing import Optional
+
+CLOUD_SDK_CLIENT_ID: str
+
+def get_config_path() -> str: ...
+def get_application_default_credentials_path() -> str: ...
+def get_project_id() -> Optional[str]: ...
+def get_auth_access_token(account: Optional[str] = None) -> str: ...

stubs/google/auth/_constants.pyi

@@ -0,0 +1,3 @@
+_SERVICE_ACCOUNT_TRUST_BOUNDARY_LOOKUP_ENDPOINT: str
+_WORKFORCE_POOL_TRUST_BOUNDARY_LOOKUP_ENDPOINT: str
+_WORKLOAD_IDENTITY_POOL_TRUST_BOUNDARY_LOOKUP_ENDPOINT: str

stubs/google/auth/_credentials_async.pyi

@@ -0,0 +1,21 @@
+import abc
+from typing import Mapping, Sequence
+
+from google.auth import credentials
+from google.auth.transport import Request as _Request
+
+class Credentials(credentials.Credentials, metaclass=abc.ABCMeta):
+    async def before_request(
+        self, request: _Request, method: str, url: str, headers: Mapping[str, str]
+    ) -> None: ...
+
+class CredentialsWithQuotaProject(
+    credentials.CredentialsWithQuotaProject, metaclass=abc.ABCMeta
+): ...
+class AnonymousCredentials(credentials.AnonymousCredentials, Credentials): ...
+class ReadOnlyScoped(credentials.ReadOnlyScoped, metaclass=abc.ABCMeta): ...
+class Scoped(credentials.Scoped, metaclass=abc.ABCMeta): ...
+
+def with_scopes_if_required(
+    credentials: "Credentials", scopes: Sequence[str]
+) -> "Credentials": ...

stubs/google/auth/_credentials_base.pyi

@@ -0,0 +1,11 @@
+import abc
+
+from _typeshed import Incomplete
+from google.auth.transport import Request as _TransportRequest
+
+class _BaseCredentials(metaclass=abc.ABCMeta):
+    token: Incomplete
+
+    def __init__(self) -> None: ...
+    @abc.abstractmethod
+    def refresh(self, request: _TransportRequest) -> None: ...

stubs/google/auth/_default.pyi

@@ -0,0 +1,29 @@
+from typing import TYPE_CHECKING, Any, Mapping, Optional, Sequence, Tuple
+
+from google.auth.credentials import Credentials as Credentials
+from google.auth.transport import Request as Request
+
+if TYPE_CHECKING:
+    from google.auth.api_key import Credentials as _ApiKeyCredentials
+
+def load_credentials_from_file(
+    filename: str,
+    scopes: Optional[Sequence[str]] = None,
+    default_scopes: Optional[Sequence[str]] = None,
+    quota_project_id: Optional[str] = None,
+    request: Optional[Request] = None,
+) -> Tuple[Credentials, Optional[str]]: ...
+def load_credentials_from_dict(
+    info: Mapping[str, Any],
+    scopes: Optional[Sequence[str]] = None,
+    default_scopes: Optional[Sequence[str]] = None,
+    quota_project_id: Optional[str] = None,
+    request: Optional[Request] = None,
+) -> Tuple[Credentials, Optional[str]]: ...
+def get_api_key_credentials(key: str) -> "_ApiKeyCredentials": ...
+def default(
+    scopes: Optional[Sequence[str]] = None,
+    request: Optional[Request] = None,
+    quota_project_id: Optional[str] = None,
+    default_scopes: Optional[Sequence[str]] = None,
+) -> Tuple[Credentials, Optional[str]]: ...

stubs/google/auth/_default_async.pyi

@@ -0,0 +1,15 @@
+from typing import Optional, Sequence, Tuple
+
+from google.auth.credentials import Credentials as Credentials
+from google.auth.transport import Request as _Request
+
+def load_credentials_from_file(
+    filename: str,
+    scopes: Optional[Sequence[str]] = None,
+    quota_project_id: Optional[str] = None,
+) -> Tuple[Credentials, Optional[str]]: ...
+def default_async(
+    scopes: Optional[Sequence[str]] = None,
+    request: Optional[_Request] = None,
+    quota_project_id: Optional[str] = None,
+) -> Tuple[Credentials, Optional[str]]: ...

stubs/google/auth/_exponential_backoff.pyi

@@ -0,0 +1,22 @@
+class _BaseExponentialBackoff:
+    def __init__(
+        self,
+        total_attempts: int = 3,
+        initial_wait_seconds: float = 1.0,
+        randomization_factor: float = 0.1,
+        multiplier: float = 2.0,
+    ) -> None: ...
+    @property
+    def total_attempts(self) -> int: ...
+    @property
+    def backoff_count(self) -> int: ...
+
+class ExponentialBackoff(_BaseExponentialBackoff):
+    def __init__(self, *args: object, **kwargs: object) -> None: ...
+    def __iter__(self) -> "ExponentialBackoff": ...
+    def __next__(self) -> int: ...
+
+class AsyncExponentialBackoff(_BaseExponentialBackoff):
+    def __init__(self, *args: object, **kwargs: object) -> None: ...
+    def __aiter__(self) -> "AsyncExponentialBackoff": ...
+    async def __anext__(self) -> int: ...