o
    g#E                     @   s   d Z ddlZddlZddlmZ ddlmZ ddlZddlm	Z	 ddlm
Z
 ddlmZ ddlmZ ddlmZ dd	lmZ dd
lmZ dZdZejdfddZG dd dejejejZG dd dejZdS )a  Google Cloud Impersonated credentials.

This module provides authentication for applications where local credentials
impersonates a remote service account using `IAM Credentials API`_.

This class can be used to impersonate a service account as long as the original
Credential object has the "Service Account Token Creator" role on the target
service account.

    .. _IAM Credentials API:
        https://cloud.google.com/iam/credentials/reference/rest/
    N)datetime)_exponential_backoff)_helpers)credentials)
exceptions)iam)jwt)metricsz*Unable to acquire impersonated credentialsi  c              
   C   s   |pt jtj||}t|d}| |d||d}t	|j
dr)|j
dn|j
}|jtjkr8tt|zt|}	|	d }
t|	d d}|
|fW S  ttfyh } ztdt|}||d	}~ww )
a  Makes a request to the Google Cloud IAM service for an access token.
    Args:
        request (Request): The Request object to use.
        principal (str): The principal to request an access token for.
        headers (Mapping[str, str]): Map of headers to transmit.
        body (Mapping[str, str]): JSON Payload body for the iamcredentials
            API call.
        iam_endpoint_override (Optiona[str]): The full IAM endpoint override
            with the target_principal embedded. This is useful when supporting
            impersonation with regional endpoints.

    Raises:
        google.auth.exceptions.TransportError: Raised if there is an underlying
            HTTP connection error
        google.auth.exceptions.RefreshError: Raised if the impersonated
            credentials are not available.  Common reasons are
            `iamcredentials.googleapis.com` is not enabled or the
            `Service Account Token Creator` is not assigned
    utf-8POST)urlmethodheadersbodydecodeaccessToken
expireTimez%Y-%m-%dT%H:%M:%SZz6{}: No access token or invalid expiration in response.N)r   _IAM_ENDPOINTreplacer   DEFAULT_UNIVERSE_DOMAINformatjsondumpsencodehasattrdatar   statushttp_clientOKr   RefreshError_REFRESH_ERRORloadsr   strptimeKeyError
ValueError)request	principalr   r   universe_domainiam_endpoint_overrideiam_endpointresponseresponse_bodytoken_responsetokenexpiry
caught_excnew_exc r1   p/var/www/html/Testing_prj/Navya-Bakers/venv/lib/python3.10/site-packages/google/auth/impersonated_credentials.py_make_iam_token_request0   s8   



r3   c                       s   e Zd ZdZdeddf fdd	Zdd Zee	j
dd Zd	d
 Zdd Zedd Zedd Zedd Zedd Zee	j
dd Zdd Zee	jdd Zee	jdddZ  ZS )Credentialsa  This module defines impersonated credentials which are essentially
    impersonated identities.

    Impersonated Credentials allows credentials issued to a user or
    service account to impersonate another. The target service account must
    grant the originating credential principal the
    `Service Account Token Creator`_ IAM role:

    For more information about Token Creator IAM role and
    IAMCredentials API, see
    `Creating Short-Lived Service Account Credentials`_.

    .. _Service Account Token Creator:
        https://cloud.google.com/iam/docs/service-accounts#the_service_account_token_creator_role

    .. _Creating Short-Lived Service Account Credentials:
        https://cloud.google.com/iam/docs/creating-short-lived-service-account-credentials

    Usage:

    First grant source_credentials the `Service Account Token Creator`
    role on the target account to impersonate.   In this example, the
    service account represented by svc_account.json has the
    token creator role on
    `impersonated-account@_project_.iam.gserviceaccount.com`.

    Enable the IAMCredentials API on the source project:
    `gcloud services enable iamcredentials.googleapis.com`.

    Initialize a source credential which does not have access to
    list bucket::

        from google.oauth2 import service_account

        target_scopes = [
            'https://www.googleapis.com/auth/devstorage.read_only']

        source_credentials = (
            service_account.Credentials.from_service_account_file(
                '/path/to/svc_account.json',
                scopes=target_scopes))

    Now use the source credentials to acquire credentials to impersonate
    another service account::

        from google.auth import impersonated_credentials

        target_credentials = impersonated_credentials.Credentials(
          source_credentials=source_credentials,
          target_principal='impersonated-account@_project_.iam.gserviceaccount.com',
          target_scopes = target_scopes,
          lifetime=500)

    Resource access is granted::

        client = storage.Client(credentials=target_credentials)
        buckets = client.list_buckets(project='your_project')
        for bucket in buckets:
          print(bucket.name)
    Nc                    s   t t|   t|| _t| jtjr,| jt	j
| _t| jdr,| jjr,| jd |j| _|| _|| _|| _|p<t| _d| _t | _|| _|| _d| _dS )aL  
        Args:
            source_credentials (google.auth.Credentials): The source credential
                used as to acquire the impersonated credentials.
            target_principal (str): The service account to impersonate.
            target_scopes (Sequence[str]): Scopes to request during the
                authorization grant.
            delegates (Sequence[str]): The chained list of delegates required
                to grant the final access_token.  If set, the sequence of
                identities must have "Service Account Token Creator" capability
                granted to the prceeding identity.  For example, if set to
                [serviceAccountB, serviceAccountC], the source_credential
                must have the Token Creator role on serviceAccountB.
                serviceAccountB must have the Token Creator on
                serviceAccountC.
                Finally, C must have Token Creator on target_principal.
                If left unset, source_credential must have that role on
                target_principal.
            lifetime (int): Number of seconds the delegated credential should
                be valid for (upto 3600).
            quota_project_id (Optional[str]): The project ID used for quota and billing.
                This project may be different from the project used to
                create the credentials.
            iam_endpoint_override (Optiona[str]): The full IAM endpoint override
                with the target_principal embedded. This is useful when supporting
                impersonation with regional endpoints.
        _create_self_signed_jwtN)superr4   __init__copy_source_credentials
isinstancer   Scopedwith_scopesr   
_IAM_SCOPEr   _always_use_jwt_accessr5   r'   _universe_domain_target_principal_target_scopes
_delegates_DEFAULT_TOKEN_LIFETIME_SECS	_lifetimer-   r   utcnowr.   _quota_project_id_iam_endpoint_override_cred_file_path)selfsource_credentialstarget_principaltarget_scopes	delegateslifetimequota_project_idr(   	__class__r1   r2   r7      s*   &



zCredentials.__init__c                 C   s   t jS N)r	   CRED_TYPE_SA_IMPERSONATErI   r1   r1   r2   _metric_header_for_usage   s   z$Credentials._metric_header_for_usagec                 C   s   |  | d S rR   )_update_token)rI   r%   r1   r1   r2   refresh   s   zCredentials.refreshc                 C   s   | j jtjjks| j jtjjkr| j | | j| jt	| j
d d}ddtjt i}| j | t|| j||| j| jd\| _| _dS )zUpdates credentials with a new access_token representing
        the impersonated account.

        Args:
            request (google.auth.transport.requests.Request): Request object
                to use for refreshing credentials.
        s)rM   scoperN   Content-Typeapplication/json)r%   r&   r   r   r'   r(   N)r9   token_stater   
TokenStateSTALEINVALIDrW   rB   rA   strrD   r	   API_CLIENT_HEADER&token_request_access_token_impersonateapplyr3   r@   r'   rG   r-   r.   )rI   r%   r   r   r1   r1   r2   rV      s&   
zCredentials._update_tokenc           
      C   s   ddl m} tjtj| j| j	}t
|d| jd}ddi}|| j}z=t }|D ]0}|j|||d}	|	jtjv r@q/|	jtjkrPtd|	 t
|	 d	   W |  S W |  n|  w td
)Nr   AuthorizedSessionr
   )payloadrM   rZ   r[   )r   r   r   zError calling sign_bytes: {}
signedBlobz#exhausted signBlob endpoint retries)google.auth.transport.requestsre   r   _IAM_SIGN_ENDPOINTr   r   r   r'   r   r@   base64	b64encoder   rB   r9   r   ExponentialBackoffpoststatus_codeIAM_RETRY_CODESr   r   r   TransportErrorr   	b64decodeclose)
rI   messagere   iam_sign_endpointr   r   authed_sessionretries_r*   r1   r1   r2   
sign_bytes  s8   


zCredentials.sign_bytesc                 C      | j S rR   r@   rT   r1   r1   r2   signer_emailA     zCredentials.signer_emailc                 C   ry   rR   rz   rT   r1   r1   r2   service_account_emailE  r|   z!Credentials.service_account_emailc                 C   s   | S rR   r1   rT   r1   r1   r2   signerI  s   zCredentials.signerc                 C   s   | j  S rR   )rA   rT   r1   r1   r2   requires_scopesM  s   zCredentials.requires_scopesc                 C   s   | j r| j d| jdS d S )Nzimpersonated credentials)credential_sourcecredential_typer&   )rH   r@   rT   r1   r1   r2   get_cred_infoQ  s   zCredentials.get_cred_infoc              	   C   s2   | j | j| j| j| j| j| j| jd}| j|_|S )N)rK   rL   rM   rN   rO   r(   )	rQ   r9   r@   rA   rB   rD   rF   rG   rH   )rI   credr1   r1   r2   
_make_copy[  s   	zCredentials._make_copyc                 C   s   |   }||_|S rR   )r   rF   )rI   rO   r   r1   r1   r2   with_quota_projecth  s   zCredentials.with_quota_projectc                 C   s   |   }|p||_|S rR   )r   rA   )rI   scopesdefault_scopesr   r1   r1   r2   r<   n  s   
zCredentials.with_scopesrR   )__name__
__module____qualname____doc__rC   r7   rU   r   copy_docstringr   r4   rW   rV   rx   propertyr{   r}   r~   r   r   r   CredentialsWithQuotaProjectr   r;   r<   __classcell__r1   r1   rP   r2   r4   n   s6    BC

'"





	


r4   c                       sj   e Zd ZdZ			d fdd	ZdddZdd	 Zd
d Ze	e
jdd Ze	e
jdd Z  ZS )IDTokenCredentialszAOpen ID Connect ID Token-based service account credentials.

    NFc                    s>   t t|   t|tstd|| _|| _|| _	|| _
dS )a  
        Args:
            target_credentials (google.auth.Credentials): The target
                credential used as to acquire the id tokens for.
            target_audience (string): Audience to issue the token for.
            include_email (bool): Include email in IdToken
            quota_project_id (Optional[str]):  The project ID used for
                quota and billing.
        z4Provided Credential must be impersonated_credentialsN)r6   r   r7   r:   r4   r   GoogleAuthError_target_credentials_target_audience_include_emailrF   )rI   target_credentialstarget_audienceinclude_emailrO   rP   r1   r2   r7   z  s   

zIDTokenCredentials.__init__c                 C   s   | j ||| j| jdS N)r   r   r   rO   )rQ   r   rF   )rI   r   r   r1   r1   r2   from_credentials  s   z#IDTokenCredentials.from_credentialsc                 C   s   | j | j|| j| jdS r   )rQ   r   r   rF   )rI   r   r1   r1   r2   with_target_audience  s   z'IDTokenCredentials.with_target_audiencec                 C   s   | j | j| j|| jdS r   )rQ   r   r   rF   )rI   r   r1   r1   r2   with_include_email  s   z%IDTokenCredentials.with_include_emailc                 C   s   | j | j| j| j|dS r   )rQ   r   r   r   )rI   rO   r1   r1   r2   r     s   z%IDTokenCredentials.with_quota_projectc           	      C   s   ddl m} tjtj| jj	| jj
}| j| jj| jd}ddtjt i}|| jj|d}z|j||t|dd}W |  n|  w |jtjkrZtd		| | d
 }|| _ttj |ddd | _!d S )Nr   rd   )audiencerM   includeEmailrZ   r[   )auth_requestr
   )r   r   r   zError getting ID token: {}r-   F)verifyexp)"rh   re   r   _IAM_IDTOKEN_ENDPOINTr   r   r   r   r'   r   r{   r   rB   r   r	   ra   "token_request_id_token_impersonater9   rm   r   r   r   rr   rn   r   r   r   r   r-   r   utcfromtimestampr   r   r.   )	rI   r%   re   rt   r   r   ru   r*   id_tokenr1   r1   r2   rW     sB   


zIDTokenCredentials.refresh)NFNrR   )r   r   r   r   r7   r   r   r   r   r   r   r   r   r4   rW   r   r1   r1   rP   r2   r   u  s    



r   )r   rj   r8   r   http.clientclientr   r   google.authr   r   r   r   r   r   r	   r    rC   r   r3   r;   r   Signingr4   r   r1   r1   r1   r2   <module>   s0   

>  	