o ~j6hQJ@sdZddlZddlmZddlmZddlmZddlmZddlmZddlmZdd l m Z dd l m Z Gd d d ej ejejZd ZdZGdddejejejZdS)zGoogle Compute Engine credentials. This module provides authentication for an application running on Google Compute Engine using the Compute Engine metadata server. N)_helpers) credentials) exceptions)iam)jwt)metrics) _metadata)_clientcseZdZdZ     dfdd ZddZdd Zd d Zed d Z eddZ eddZ e ejddZe ejddZe ejdddZe ejddZZS) CredentialsaCompute Engine Credentials. These credentials use the Google Compute Engine metadata server to obtain OAuth 2.0 access tokens associated with the instance's service account, and are also used for Cloud Run, Flex and App Engine (except for the Python 2.7 runtime, which is supported only on older versions of this library). For more information about Compute Engine authentication, including how to configure scopes, see the `Compute Engine authentication documentation`_. .. note:: On Compute Engine the metadata server ignores requested scopes. On Cloud Run, Flex and App Engine the server honours requested scopes. .. _Compute Engine authentication documentation: https://cloud.google.com/compute/docs/authentication#using defaultNcsDtt|||_||_||_||_d|_|r ||_d|_dSdS)a_ Args: service_account_email (str): The service account email to use, or 'default'. A Compute Engine instance may have multiple service accounts. quota_project_id (Optional[str]): The project ID used for quota and billing. scopes (Optional[Sequence[str]]): The list of scopes for the credentials. default_scopes (Optional[Sequence[str]]): Default scopes passed by a Google client library. Use 'scopes' for user-defined scopes. universe_domain (Optional[str]): The universe domain. If not provided or None, credential will attempt to fetch the value from metadata server. If metadata server doesn't have universe domain endpoint, then the default googleapis.com will be used. FTN) superr __init___service_account_email_quota_project_id_scopes_default_scopes_universe_domain_cached_universe_domain)selfservice_account_emailquota_project_idscopesdefault_scopesuniverse_domain __class__u/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/site-packages/google/auth/compute_engine/credentials.pyr 9s zCredentials.__init__cCs6tj||jd}|d|_|jdur|d|_dSdS)zRetrieve information about the service account. Updates the scopes and retrieves the full service account email. Args: request (google.auth.transport.Request): The object used to make HTTP requests. )service_accountemailNr)rget_service_account_inforr)rrequestinforrr_retrieve_infoZs   zCredentials._retrieve_infocCstjSN)rCRED_TYPE_SA_MDSrrrr_metric_header_for_usagemsz$Credentials._metric_header_for_usagec Csl|jdur|jn|j}z||tj||j|d\|_|_WdStj y5}zt |}||d}~ww)ajRefresh the access token and scopes. Args: request (google.auth.transport.Request): The object used to make HTTP requests. Raises: google.auth.exceptions.RefreshError: If the Compute Engine metadata service can't be reached if if the instance has not credentials. N)rr) rrr#rget_service_account_tokenrtokenexpiryrTransportError RefreshError)rr!r caught_excnew_excrrrrefreshps   zCredentials.refreshcC|jS)zThe service account email. .. note:: This is not guaranteed to be set until :meth:`refresh` has been called. rr&rrrrsz!Credentials.service_account_emailcCs|j Sr$)rr&rrrrequires_scopesszCredentials.requires_scopescCs4|jr|jSddlm}t||_d|_|jS)Nr)requestsT)rrgoogle.auth.transportr3rget_universe_domainRequest)rgoogle_auth_requestsrrrrs zCredentials.universe_domaincCsdd|jdS)Nzmetadata serverzVM credentials)credential_sourcecredential_type principal)rr&rrr get_cred_infoszCredentials.get_cred_infocCs,|j|j||j|jd}|j|_|j|_|S)N)rrrr)rrrrrr)rrcredsrrrwith_quota_projectszCredentials.with_quota_projectcCs*|j|||j|jd}|j|_|j|_|S)N)rrrr)rrrrr)rrrr<rrr with_scopesszCredentials.with_scopescCs|j|j|j|j|j|dS)N)rrrrr)rrrrr)rrrrrwith_universe_domainsz Credentials.with_universe_domain)r NNNNr$)__name__ __module__ __qualname____doc__r r#r'r/propertyrr2rrcopy_docstringrr r;CredentialsWithQuotaProjectr=Scopedr>CredentialsWithUniverseDomainr? __classcell__rrrrr "s2!        r iz*https://www.googleapis.com/oauth2/v4/tokencseZdZdZ      dfdd ZddZeej dd Z eej d d Z d d Z ddZddZeeejddZddZeddZeddZZS)IDTokenCredentialsa>Open ID Connect ID Token-based service account credentials. These credentials relies on the default service account of a GCE instance. ID token can be requested from `GCE metadata server identity endpoint`_, IAM token endpoint or other token endpoints you specify. If metadata server identity endpoint is not used, the GCE instance must have been started with a service account that has access to the IAM Cloud API. .. _GCE metadata server identity endpoint: https://cloud.google.com/compute/docs/instances/verifying-instance-identity NFc stt|||_||_||_|r(|s|s|s|rtdd|_d|_ d|_ |dur7t |} | d|_ n||_ |s`|durJtj|t|j d}||_ |pPt|_|dur[||_ dSi|_ dSdS)a\ Args: request (google.auth.transport.Request): The object used to make HTTP requests. target_audience (str): The intended audience for these credentials, used when requesting the ID Token. The ID Token's ``aud`` claim will be set to this string. token_uri (str): The OAuth 2.0 Token URI. additional_claims (Mapping[str, str]): Any additional claims for the JWT assertion used in the authorization grant. service_account_email (str): Optional explicit service account to use to sign JWT tokens. By default, this is the default GCE service account. signer (google.auth.crypt.Signer): The signer used to sign JWTs. In case the signer is specified, the request argument will be ignored. use_metadata_identity_endpoint (bool): Whether to use GCE metadata identity endpoint. For backward compatibility the default value is False. If set to True, ``token_uri``, ``additional_claims``, ``service_account_email``, ``signer`` argument should not be set; otherwise ValueError will be raised. quota_project_id (Optional[str]): The project ID used for quota and billing. Raises: ValueError: If ``use_metadata_identity_endpoint`` is set to True, and one of ``token_uri``, ``additional_claims``, ``service_account_email``, ``signer`` arguments is set. zIf use_metadata_identity_endpoint is set, token_uri, additional_claims, service_account_email, signer arguments must not be setNr)r!rr)r rJr r_use_metadata_identity_endpoint_target_audiencerMalformedError _token_uri_additional_claims_signerrr rrSignerr _DEFAULT_TOKEN_URI) rr!target_audience token_uriadditional_claimsrsigneruse_metadata_identity_endpointrsa_inforrrr s:)     zIDTokenCredentials.__init__c CsB|jr |jd|d|jdS|jd|j|j||j|jd|jdS)a]Create a copy of these credentials with the specified target audience. Args: target_audience (str): The intended audience for these credentials, used when requesting the ID Token. Returns: google.auth.service_account.IDTokenCredentials: A new credentials instance. NTrSrWrFrrTrSrUrVrWr)rKrrrrNrOcopyrV)rrSrrrwith_target_audience3s" z'IDTokenCredentials.with_target_audiencec CsB|jr |jd|jd|dS|jd|j|j|j|j|jd|dS)NTrYFrZ)rKrrLrrNrOr[rV)rrrrrr=Rs"z%IDTokenCredentials.with_quota_projectc Cs8|jrtd|jd|j||j|j|jd|j dS)NzCIf use_metadata_identity_endpoint is set, token_uri must not be setFrZ) rKrrMrrrLrOr[rVr)rrTrrrwith_token_urijsz!IDTokenCredentials.with_token_uricCs\t}tjtd}||}t|t||j|j|jd}| |j t |j |}|S)zCreate the OAuth 2.0 assertion. This assertion is used during the OAuth 2.0 grant to acquire an ID token. Returns: bytes: The authorization grant assertion. )seconds)iatexpissaudrS)rutcnowdatetime timedelta_DEFAULT_TOKEN_LIFETIME_SECSdatetime_to_secsrrNrLupdaterOrencoderP)rnowlifetimer*payloadr)rrr#_make_authorization_grant_assertions  z6IDTokenCredentials._make_authorization_grant_assertionc Cszd}|jdd}tjti}tj||||d}Wntjy/}zt|}||d}~wwt |\}} }}|t j | dfS)aRequest ID token from metadata identity endpoint. Args: request (google.auth.transport.Request): The object used to make HTTP requests. Returns: Tuple[str, datetime.datetime]: The ID token and the expiry of the ID token. Raises: google.auth.exceptions.RefreshError: If the Compute Engine metadata service can't be reached or if the instance has no credentials. ValueError: If extracting expiry from the obtained ID token fails. z*instance/service-accounts/default/identityfull)audienceformat)paramsheadersNr`) rLrAPI_CLIENT_HEADERtoken_request_id_token_mdsrgetrr+r,r_unverified_decoderdutcfromtimestamp) rr!pathrqmetrics_headerid_tokenr-r._rlrrr _call_metadata_identity_endpoints    z3IDTokenCredentials._call_metadata_identity_endpointcCsJ|jr||\|_|_dS|}t||j|\}}}||_||_dS)agRefreshes the ID token. Args: request (google.auth.transport.Request): The object used to make HTTP requests. Raises: google.auth.exceptions.RefreshError: If the credentials could not be refreshed. ValueError: If extracting expiry from the obtained ID token fails. N)rKr|r)r*rmr id_token_jwt_grantrN)rr! assertion access_tokenr*r{rrrr/s   zIDTokenCredentials.refreshcCr0r$)rPr&rrrrVzIDTokenCredentials.signercCs|jrtd|j|S)a&Signs the given message. Args: message (bytes): The message to sign. Returns: bytes: The message's cryptographic signature. Raises: ValueError: Signer is not available if metadata identity endpoint is used. z=Signer is not available if metadata identity endpoint is used)rKrInvalidOperationrPsign)rmessagerrr sign_bytess   zIDTokenCredentials.sign_bytescCr0)zThe service account email.r1r&rrrrrz(IDTokenCredentials.service_account_emailcCr0r$r1r&rrr signer_emailszIDTokenCredentials.signer_email)NNNNFN)r@rArBrCr r\rrErrFr=CredentialsWithTokenUrir]rmr|r/rDSigningrVrrrrIrrrrrJs2O       rJ)rCrd google.authrrrrrrgoogle.auth.compute_enginer google.oauth2r rGrFrHr rfrRrrrJrrrrs.         -