# ------------------------------------
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
# ------------------------------------
from typing import Callable, Optional, Any
from azure.core.credentials import AccessToken
from .._internal import AadClient
from .._internal.get_token_mixin import GetTokenMixin
[docs]class ClientAssertionCredential(GetTokenMixin):
"""Authenticates a service principal with a JWT assertion.
This credential is for advanced scenarios. :class:`~azure.identity.CertificateCredential` has a more
convenient API for the most common assertion scenario, authenticating a service principal with a certificate.
:param str tenant_id: ID of the principal's tenant. Also called its "directory" ID.
:param str client_id: The principal's client ID
:param func: A callable that returns a string assertion. The credential will call this every time it
acquires a new token.
:paramtype func: Callable[[], str]
:keyword str authority: Authority of a Microsoft Entra endpoint, for example
"login.microsoftonline.com", the authority for Azure Public Cloud (which is the default).
:class:`~azure.identity.AzureAuthorityHosts` defines authorities for other clouds.
:keyword List[str] additionally_allowed_tenants: Specifies tenants in addition to the specified "tenant_id"
for which the credential may acquire tokens. Add the wildcard value "*" to allow the credential to
acquire tokens for any tenant the application can access.
.. admonition:: Example:
.. literalinclude:: ../samples/credential_creation_code_snippets.py
:start-after: [START create_client_assertion_credential]
:end-before: [END create_client_assertion_credential]
:language: python
:dedent: 4
:caption: Create a ClientAssertionCredential.
"""
def __init__(self, tenant_id: str, client_id: str, func: Callable[[], str], **kwargs: Any) -> None:
self._func = func
authority = kwargs.pop("authority", None)
cache = kwargs.pop("cache", None)
cae_cache = kwargs.pop("cae_cache", None)
additionally_allowed_tenants = kwargs.pop("additionally_allowed_tenants", None)
self._client = AadClient(
tenant_id,
client_id,
authority=authority,
cache=cache,
cae_cache=cae_cache,
additionally_allowed_tenants=additionally_allowed_tenants,
**kwargs
)
super(ClientAssertionCredential, self).__init__(**kwargs)
def __enter__(self) -> "ClientAssertionCredential":
self._client.__enter__()
return self
def __exit__(self, *args: Any) -> None:
self._client.__exit__(*args)
[docs] def close(self) -> None:
self.__exit__()
def _acquire_token_silently(self, *scopes: str, **kwargs: Any) -> Optional[AccessToken]:
return self._client.get_cached_access_token(scopes, **kwargs)
def _request_token(self, *scopes: str, **kwargs: Any) -> AccessToken:
assertion = self._func()
token = self._client.obtain_token_by_jwt_assertion(scopes, assertion, **kwargs)
return token