SupportsGetToken -> TokenCredential

Author: chlowell

sdk/keyvault/azure-security-keyvault/azure/security/keyvault/_internal.py

@@ -8,7 +8,7 @@
 if TYPE_CHECKING:
     # pylint:disable=unused-import
     from typing import Any, Mapping, Optional
-    from azure.core.credentials import SupportsGetToken
+    from azure.core.credentials import TokenCredential
     from azure.core.pipeline.transport import HttpTransport
 
 try:
@@ -55,23 +55,23 @@ class _KeyVaultClientBase(object):
     """
     :param credential:  A credential or credential provider which can be used to authenticate to the vault,
         a ValueError will be raised if the entity is not provided
-    :type credential: azure.core.credentials.SupportsGetToken
+    :type credential: azure.core.credentials.TokenCredential
     :param str vault_url: The url of the vault to which the client will connect,
         a ValueError will be raised if the entity is not provided
     :param ~azure.core.configuration.Configuration config:  The configuration for the KeyClient
     """
 
     @staticmethod
     def create_config(credential, api_version=None, **kwargs):
-        # type: (SupportsGetToken, Optional[str], Mapping[str, Any]) -> Configuration
+        # type: (TokenCredential, Optional[str], Mapping[str, Any]) -> Configuration
         if api_version is None:
             api_version = KeyVaultClient.DEFAULT_API_VERSION
         config = KeyVaultClient.get_configuration_class(api_version, aio=False)(credential, **kwargs)
         config.authentication_policy = BearerTokenCredentialPolicy(credential, scopes=KEY_VAULT_SCOPES)
         return config
 
     def __init__(self, vault_url, credential, config=None, transport=None, api_version=None, **kwargs):
-        # type: (str, SupportsGetToken, Configuration, Optional[HttpTransport], Optional[str], **Any) -> None
+        # type: (str, TokenCredential, Configuration, Optional[HttpTransport], Optional[str], **Any) -> None
         if not credential:
             raise ValueError("credential should be a credential object from azure-identity")
         if not vault_url:

sdk/keyvault/azure-security-keyvault/azure/security/keyvault/aio/_internal.py

@@ -6,9 +6,9 @@
 from typing import Any, Callable, Mapping, Optional, TYPE_CHECKING
 if TYPE_CHECKING:
     try:
-        from azure.core.credentials import SupportsGetToken
+        from azure.core.credentials import TokenCredential
     except ImportError:
-        # SupportsGetToken is a typing_extensions.Protocol; we don't depend on that package
+        # TokenCredential is a typing_extensions.Protocol; we don't depend on that package
         pass
 
 from azure.core.async_paging import AsyncPagedMixin
@@ -52,7 +52,7 @@ class _AsyncKeyVaultClientBase:
 
     @staticmethod
     def create_config(
-        credential: "SupportsGetToken", api_version: str = None, **kwargs: Mapping[str, Any]
+        credential: "TokenCredential", api_version: str = None, **kwargs: Mapping[str, Any]
     ) -> Configuration:
         if api_version is None:
             api_version = KeyVaultClient.DEFAULT_API_VERSION
@@ -63,7 +63,7 @@ def create_config(
     def __init__(
         self,
         vault_url: str,
-        credential: "SupportsGetToken",
+        credential: "TokenCredential",
         config: Configuration = None,
         transport: HttpTransport = None,
         api_version: str = None,

sdk/keyvault/azure-security-keyvault/azure/security/keyvault/aio/vault_client.py

@@ -6,9 +6,9 @@
 from typing import Any, Optional, TYPE_CHECKING
 if TYPE_CHECKING:
     try:
-        from azure.core.credentials import SupportsGetToken
+        from azure.core.credentials import TokenCredential
     except ImportError:
-        # SupportsGetToken is a typing_extensions.Protocol; we don't depend on that package
+        # TokenCredential is a typing_extensions.Protocol; we don't depend on that package
         pass
 
 from azure.core import Configuration
@@ -23,7 +23,7 @@ class VaultClient(_AsyncKeyVaultClientBase):
     def __init__(
         self,
         vault_url: str,
-        credential: "SupportsGetToken",
+        credential: "TokenCredential",
         config: Configuration = None,
         transport: HttpTransport = None,
         api_version: str = None,

sdk/keyvault/azure-security-keyvault/azure/security/keyvault/vault_client.py

@@ -11,7 +11,7 @@
 if TYPE_CHECKING:
     # pylint:disable=unused-import
     from azure.core import Configuration
-    from azure.core.credentials import SupportsGetToken
+    from azure.core.credentials import TokenCredential
     from azure.core.pipeline.transport import HttpTransport
     from typing import Any, Mapping, Optional
 
@@ -22,7 +22,7 @@
 
 class VaultClient(_KeyVaultClientBase):
     def __init__(self, vault_url, credential, config=None, transport=None, api_version=None, **kwargs):
-        # type: (str, SupportsGetToken, Configuration, Optional[HttpTransport], Optional[str], **Any) -> None
+        # type: (str, TokenCredential, Configuration, Optional[HttpTransport], Optional[str], **Any) -> None
         super(VaultClient, self).__init__(
             vault_url, credential, config=config, transport=transport, api_version=api_version, **kwargs
         )