File: //opt/cppython/lib/python3.8/site-packages/google/auth/__pycache__/external_account.cpython-38.pyc
U
KD�gye � @ s� d Z ddlZddlZddlmZ ddlZddlZddlZddlZddl Z ddl
m Z ddl
m Z ddl
m
Z
ddl
mZ ddl
mZ dd lmZ dd
lmZ d Zd Zd
ZdZdZeG dd� d��ZG dd� de je je jej d�Z dS )a� External Account Credentials.
This module provides credentials that exchange workload identity pool external
credentials for Google access tokens. This facilitates accessing Google Cloud
Platform resources from on-prem and non-Google Cloud platforms (e.g. AWS,
Microsoft Azure, OIDC identity providers), using native credentials retrieved
from the current environment without the need to copy, save and manage
long-lived service account credentials.
Specifically, this is intended to use access tokens acquired using the GCP STS
token exchange endpoint following the `OAuth 2.0 Token Exchange`_ spec.
.. _OAuth 2.0 Token Exchange: https://tools.ietf.org/html/rfc8693
� N)� dataclass)�_helpers)� credentials)�
exceptions)�impersonated_credentials)�metrics)�sts)�utilsZexternal_accountz/urn:ietf:params:oauth:grant-type:token-exchangez-urn:ietf:params:oauth:token-type:access_tokenz8https://cloudresourcemanager.googleapis.com/v1/projects/z&https://sts.{universe_domain}/v1/tokenc @ s" e Zd ZU dZeed< eed< dS )�SupplierContextu� A context class that contains information about the requested third party credential that is passed
to AWS security credential and subject token suppliers.
Attributes:
subject_token_type (str): The requested subject token type based on the Oauth2.0 token exchange spec.
Expected values include::
“urn:ietf:params:oauth:token-type:jwt”
“urn:ietf:params:oauth:token-type:id-token”
“urn:ietf:params:oauth:token-type:saml2”
“urn:ietf:params:aws:token-type:aws4_request”
audience (str): The requested audience for the subject token.
�subject_token_type�audienceN)�__name__�
__module__� __qualname__�__doc__�str�__annotations__� r r �I/opt/cppython/lib/python3.8/site-packages/google/auth/external_account.pyr
; s
r
c sn e Zd ZdZdddddddddejdf � fdd� Zedd� �Zdd� Z ed d
� �Z
ed d � �Z ed
d� �Z edd� �Z
edd� �Zedd� �Ze�ej�dd� �Ze�ej�d5dd��Zejdd� �Zdd � Ze�ej�d d � �Zd d � Ze�ej �d!d"� �Z e�ej �d#d$� �Z e�ej �d%d&� �Z!d'd(� Z"d)d*� Z#d+d,� Z$d-d.� Z%d/d0� Z&e'd1d2� �Z(e'd3d4� �Z)� Z*S )6� CredentialsaL Base class for all external account credentials.
This is used to instantiate Credentials for exchanging external account
credentials for Google access token and authorizing requests to Google APIs.
The base class implements the common logic for exchanging external account
credentials for Google access tokens.
Nc s t t| ��� || _|| _|| _|| _| jtkrB| j�d| j�| _| | _ || _
|| _ |pZi | _ || _
|| _|
| _| | _| | _|
| _g dd�| _| j
r�t�tjj| j
| j�| _nd| _t�| j| j�| _| � � | _ d| _ d| _ t | j| j�| _!d| _"| j#�s | j�r t$�%d��dS )u6 Instantiates an external account credentials object.
Args:
audience (str): The STS audience field.
subject_token_type (str): The subject token type based on the Oauth2.0 token exchange spec.
Expected values include::
“urn:ietf:params:oauth:token-type:jwt”
“urn:ietf:params:oauth:token-type:id-token”
“urn:ietf:params:oauth:token-type:saml2”
“urn:ietf:params:aws:token-type:aws4_request”
token_url (str): The STS endpoint URL.
credential_source (Mapping): The credential source dictionary.
service_account_impersonation_url (Optional[str]): The optional service account
impersonation generateAccessToken URL.
client_id (Optional[str]): The optional client ID.
client_secret (Optional[str]): The optional client secret.
token_info_url (str): The optional STS endpoint URL for token introspection.
quota_project_id (Optional[str]): The optional quota project ID.
scopes (Optional[Sequence[str]]): Optional scopes to request during the
authorization grant.
default_scopes (Optional[Sequence[str]]): Default scopes passed by a
Google client library. Use 'scopes' for user-defined scopes.
workforce_pool_user_project (Optona[str]): The optional workforce pool user
project number when the credential corresponds to a workforce pool and not
a workload identity pool. The underlying principal must still have
serviceusage.services.use IAM permission to use the project for
billing/quota.
universe_domain (str): The universe domain. The default universe
domain is googleapis.com.
trust_boundary (str): String representation of trust boundary meta.
Raises:
google.auth.exceptions.RefreshError: If the generateAccessToken
endpoint returned an error.
z{universe_domain}Z0x0)� locationsZencoded_locationsNzPworkforce_pool_user_project should not be set for non-workforce pool credentials)&�superr �__init__� _audience�_subject_token_type�_universe_domain�
_token_url�_DEFAULT_TOKEN_URL�replace�_token_info_url�_credential_source�"_service_account_impersonation_url�&_service_account_impersonation_options�
_client_id�_client_secret�_quota_project_id�_scopes�_default_scopes� _workforce_pool_user_projectZ_trust_boundaryr ZClientAuthenticationZClientAuthType�basicZ _client_authr ZClient� _sts_client� _create_default_metrics_options�_metrics_options�_impersonated_credentials� _project_idr
Z_supplier_context�_cred_file_path�is_workforce_poolr Z InvalidValue)�selfr r � token_url�credential_source�!service_account_impersonation_url�%service_account_impersonation_options� client_id�
client_secret�token_info_url�quota_project_id�scopes�default_scopes�workforce_pool_user_project�universe_domainZtrust_boundary�� __class__r r r ^ sX 6
��� �
��zCredentials.__init__c C sH | � � }|jt|�dd�d� |�dd� |�dd� dd� |�� D �S )at Generates the dictionary representation of the current credentials.
Returns:
Mapping: The dictionary representation of the credentials. This is the
reverse of "from_info" defined on the subclasses of this class. It is
useful for serializing the current credentials so it can deserialized
later.
r5 N)�type� service_account_impersonationr: r; c S s i | ]\}}|d k r||�qS �Nr )�.0�key�valuer r r �
<dictcomp>� s z$Credentials.info.<locals>.<dictcomp>)�_constructor_args�update�_EXTERNAL_ACCOUNT_JSON_TYPE�pop�items)r1 Z config_infor r r �info� s
�� zCredentials.infoc C sb | j | j| j| j| jt�| j�p"d t�| j�| j | j
| j | j | j
| j| jd�}| js^|�d� |S )N)r r r2 r8 r4 r5 r3 r9 r6 r7 r< r: r; r= r< )r r r r r! �copy�deepcopyr" r r% r# r$ r( r& r' r r0 rJ )r1 �argsr r r rG � s* �
�
z Credentials._constructor_argsc C sP | j rL| j }|�d�}|�d�}|dkrL|dkrL||k rL|d }|||� S dS )z�Returns the service account email if service account impersonation is used.
Returns:
Optional[str]: The service account email if impersonation is used. Otherwise
None is returned.
�/z:generateAccessToken���� N)r! �rfind�find)r1 �url� start_indexZ end_indexr r r �service_account_email� s
z!Credentials.service_account_emailc C s | j r
dS | jS )a% Returns whether the credentials represent a user (True) or workload (False).
Workloads behave similarly to service accounts. Currently workloads will use
service account impersonation but will eventually not require impersonation.
As a result, this property is more reliable than the service account email
property in determining if the credentials represent a user or workload.
Returns:
bool: True if the credentials represent a user. False if they represent a
workload.
F)r! r0 �r1 r r r �is_user s zCredentials.is_userc C s t �d�}|�| jpd�dk S )ag Returns whether the credentials represent a workforce pool (True) or
workload (False) based on the credentials' audience.
This will also return True for impersonated workforce pool credentials.
Returns:
bool: True if the credentials represent a workforce pool. False if they
represent a workload.
z6//iam\.googleapis\.com/locations/[^/]+/workforcePools/� N)�re�compile�matchr )r1 �pr r r r0 s
z Credentials.is_workforce_poolc C s | j o| j S )z�Checks if the credentials requires scopes.
Returns:
bool: True if there are no scopes set otherwise False.
)r& r' rX r r r �requires_scopes( s zCredentials.requires_scopesc C sX | j �d�}z0|�d�}|d t|�k r:||d p6dW S W n tk
rR Y dS X dS )zNOptional[str]: The project number corresponding to the workload identity pool.rP ZprojectsrR N)r �split�index�len�
ValueError)r1 �
componentsZ
project_indexr r r �project_number1 s
zCredentials.project_numberc C s | j S )z4Optional[str]: The STS token introspection endpoint.)r rX r r r r8 ? s zCredentials.token_info_urlc C s* | j r&| j dd�}| jr"| j|d< |S d S )Nz external account credentials)r3 Zcredential_typeZ principal)r/ rW )r1 Zcred_info_jsonr r r �
get_cred_infoE s �
zCredentials.get_cred_infoc C s6 | � � }|j||d� | jf |�}| j|_| j|_|S )N)r: r; )rG rH r? r/ r, )r1 r: r; �kwargsZscopedr r r � with_scopesQ s zCredentials.with_scopesc C s t d��dS )a Retrieves the subject token using the credential_source object.
Args:
request (google.auth.transport.Request): A callable used to make
HTTP requests.
Returns:
str: The retrieved subject token.
z*retrieve_subject_token must be implementedN��NotImplementedError)r1 �requestr r r �retrieve_subject_tokenZ s z"Credentials.retrieve_subject_tokenc C s� | j r | j S | jdk r | jn| j}| jp,| j}|r�|r�i }t| }| �|d||� ||d|d�}t|jd�rx|j� d�n|j}t
� |�}|j dkr�|�
d�| _ | j S dS )a� Retrieves the project ID corresponding to the workload identity or workforce pool.
For workforce pool credentials, it returns the project ID corresponding to
the workforce_pool_user_project.
When not determinable, None is returned.
This is introduced to support the current pattern of using the Auth library:
credentials, project_id = google.auth.default()
The resource may not have permission (resourcemanager.projects.get) to
call this API or the required scopes may not be selected:
https://cloud.google.com/resource-manager/reference/rest/v1/projects/get#authorization-scopes
Args:
request (google.auth.transport.Request): A callable used to make
HTTP requests.
Returns:
Optional[str]: The project ID corresponding to the workload identity pool
or workforce pool if determinable.
N�GET)rU �method�headers�decode�utf-8�� Z projectId)r. r& r' re r( �_CLOUD_RESOURCE_MANAGERZbefore_request�hasattr�datarp �json�loads�status�get) r1 rk r: re ro rU �responseZ
response_body�
response_datar r r �get_project_idh s$
��
zCredentials.get_project_idc C s | j d k r| j n| j}| �� r0tj|| �� d�}| �� rB| �� | _| jrj| j� |� | jj
| _
| jj | _ n�t �
� }d }| jr�| js�d| ji}tjt�| j�i}| jj|t| �|�| j| j|t||d� }|�d�| _
|�d�}t |t �r�t |�}t j |d�}|| | _ d S )N)�certZ userProject) rk Z
grant_typeZ
subject_tokenr r r: Zrequested_token_type�additional_options�additional_headersZ access_token�
expires_in)�seconds)!r& r' �_mtls_required� functools�partial� _get_mtls_cert_and_key_paths�+_should_initialize_impersonated_credentials�$_initialize_impersonated_credentialsr- �refresh�tokenZexpiryr �utcnowr( r# r ZAPI_CLIENT_HEADERZbyoid_metrics_headerr, r* Zexchange_token�_STS_GRANT_TYPErl r r �_STS_REQUESTED_TOKEN_TYPEry �
isinstancer �int�datetime� timedelta) r1 rk r: �nowr~ r r{ r� �lifetimer r r r� � sL �
���
zCredentials.refreshc C s( | � � }| jf |�}| j|_| j|_|S rB )rG r? r/ r, )r1 rg Znew_credr r r �
_make_copy� s
zCredentials._make_copyc C s | � � }||_|S rB )r� r% )r1 r9 �credr r r �with_quota_project� s z Credentials.with_quota_projectc C s | � � }||_|S rB )r� r )r1 Z token_urir� r r r �with_token_uri� s zCredentials.with_token_uric C s | � � }||_|S rB )r� r )r1 r= r� r r r �with_universe_domain� s z Credentials.with_universe_domainc C s | j d k o| jd kS rB )r! r- rX r r r r� � s
�z7Credentials._should_initialize_impersonated_credentialsc C sv | � � }|jdi d� | jf |�}| j|_| j}|s>t�d��| jdk rN| jn| j}t j
|||| j | j | j
�d�d�S )a( Generates an impersonated credentials.
For more details, see `projects.serviceAccounts.generateAccessToken`_.
.. _projects.serviceAccounts.generateAccessToken: https://cloud.google.com/iam/docs/reference/credentials/rest/v1/projects.serviceAccounts/generateAccessToken
Returns:
impersonated_credentials.Credential: The impersonated credentials
object.
Raises:
google.auth.exceptions.RefreshError: If the generateAccessToken
endpoint returned an error.
N)r4 r5 zLUnable to determine target principal from service account impersonation URL.�token_lifetime_seconds)�source_credentials�target_principalZ
target_scopesr9 Ziam_endpoint_overrider� )rG rH r? r, rW r Z RefreshErrorr& r' r r r% r! r" ry )r1 rg r� r� r: r r r r� � s. � ���z0Credentials._initialize_impersonated_credentialsc C s> i }| j rd|d< nd|d< | j�d�r2d|d< nd|d< |S )N�truezsa-impersonation�falser� zconfig-lifetime)r! r" ry )r1 Zmetrics_optionsr r r r+ s
z+Credentials._create_default_metrics_optionsc C s dS )a> Returns a boolean representing whether the current credential is configured
for mTLS and should add a certificate to the outgoing calls to the sts and service
account impersonation endpoint.
Returns:
bool: True if the credential is configured for mTLS, False if it is not.
Fr rX r r r r� $ s zCredentials._mtls_requiredc C s t d��dS )a� Gets the file locations for a certificate and private key file
to be used for configuring mTLS for the sts and service account
impersonation calls. Currently only expected to return a value when using
X509 workload identity federation.
Returns:
Tuple[str, str]: The cert and key file locations as strings in a tuple.
Raises:
NotImplementedError: When the current credential is not configured for
mTLS.
z4_get_mtls_cert_and_key_location must be implemented.Nri rX r r r r� . s
�z(Credentials._get_mtls_cert_and_key_pathsc K sx | f |� d�|� d�|� d�|� d�|� d�|� d�p8i |� d�|� d�|� d �|� d
�|� d �|� d tj�d
� |��S )a� Creates a Credentials instance from parsed external account info.
Args:
info (Mapping[str, str]): The external account info in Google
format.
kwargs: Additional arguments to pass to the constructor.
Returns:
google.auth.identity_pool.Credentials: The constructed
credentials.
Raises:
InvalidValue: For invalid parameters.
r r r2 r8 r4 rA r6 r7 r3 r9 r< r= ) r r r2 r8 r4 r5 r6 r7 r3 r9 r< r= )ry r �DEFAULT_UNIVERSE_DOMAIN)�clsrL rg r r r � from_info? s0 �� ���zCredentials.from_infoc
K s@ t j|ddd��&}t�|�}| j|f|�W 5 Q R � S Q R X dS )aY Creates a Credentials instance from an external account json file.
Args:
filename (str): The path to the external account json file.
kwargs: Additional arguments to pass to the constructor.
Returns:
google.auth.identity_pool.Credentials: The constructed
credentials.
�rrq )�encodingN)�io�openrv �loadr� )r� �filenamerg Z json_fileru r r r � from_filef s
zCredentials.from_file)N)+r
r r r r r� r �propertyrL rG rW rY r0 r_ re r8 r Zcopy_docstringr rf �Scopedrh �abc�abstractmethodrl r| r� r� �CredentialsWithQuotaProjectr� �CredentialsWithTokenUrir� Z CredentialsWithUniverseDomainr� r� r� r+ r� r� � classmethodr� r� �
__classcell__r r r> r r P sj �h
0
2
,
&r )� metaclass) r r� rM � dataclassesr r� r� r� rv r[ Z google.authr r r r r Z
google.oauth2r r rI r� r� rs r r
r� r� r� �ABCMetar r r r r �<module> s8
�