o tBhd @s0ddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl mZejjZdZddZdd Zd d ZejejejfZd d ZddZedgdZddZ           d$ddZdZdZedddddddddddf ddZ ddZ!ddZ"d d!Z#d"d#Z$dS)%N) exceptions)requests)_helpersz[https://googleapis.dev/python/google-api-core/latest/auth.html#setting-up-a-service-accountcCs(t|tjjjstdt|tdS)aiRaise AttributeError if the credentials are unsigned. :type credentials: :class:`google.auth.credentials.Signing` :param credentials: The credentials used to create a private key for signing text. :raises: :exc:`AttributeError` if credentials is not an instance of :class:`google.auth.credentials.Signing`. zyou need a private key to sign credentials.the credentials you are currently using {} just contains a token. see {} for more details.N) isinstancegoogleauth credentialsSigningAttributeErrorformattypeSERVICE_ACCOUNT_URL)rrt/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/google/cloud/storage/_signing.pyensure_signed_credentials(s   rcCs4t|||d}t|}|j}|||dS)aGets query parameters for creating a signed URL. :type credentials: :class:`google.auth.credentials.Signing` :param credentials: The credentials used to create a private key for signing text. :type expiration: int or long :param expiration: When the signed URL should expire. :type string_to_sign: str :param string_to_sign: The string to be signed by the credentials. :raises: :exc:`AttributeError` if credentials is not an instance of :class:`google.auth.credentials.Signing`. :rtype: dict :returns: Query parameters matching the signing credentials with a signed payload. asciiGoogleAccessIdExpires Signature)r sign_bytesencodebase64 b64encode signer_email)r expirationstring_to_signsignature_bytes signatureservice_account_namerrrget_signed_query_params_v2;s r cCsbt|tjrtjtjd}||}t|tjr!t|}|d}t|tj s/t dt ||S)a Convert 'expiration' to a number of seconds in the future. :type expiration: Union[Integer, datetime.datetime, datetime.timedelta] :param expiration: Point in time when the signed URL should expire. If a ``datetime`` instance is passed without an explicit ``tzinfo`` set, it will be assumed to be ``UTC``. :raises: :exc:`TypeError` when expiration is not a valid type. :rtype: int :returns: a timestamp as an absolute number of seconds since epoch. tzinfoi@B=Expected an integer timestamp, datetime, or timedelta. Got %s) rdatetime timedeltaNOWreplacerUTC_microseconds_from_datetimesix integer_types TypeErrorr )rnowmicrosrrrget_expiration_seconds_v2Zs    r/cCst|ts tdt|tjtjd}t|tj r|}t|t j r3|j dur/|jtjd}||}t|t j r?t |}|tkrJtdt|S)aVConvert 'expiration' to a number of seconds offset from the current time. :type expiration: Union[Integer, datetime.datetime, datetime.timedelta] :param expiration: Point in time when the signed URL should expire. If a ``datetime`` instance is passed without an explicit ``tzinfo`` set, it will be assumed to be ``UTC``. :raises: :exc:`TypeError` when expiration is not a valid type. :raises: :exc:`ValueError` when expiration is too large. :rtype: Integer :returns: seconds in the future when the signed URL will expire r#r!Nz0Max allowed expiration interval is seven days {})r_EXPIRATION_TYPESr,r r&r'rr(r*r+r$r"r%int total_seconds SEVEN_DAYS ValueErrorr )rr-secondsrrrget_expiration_seconds_v4|s(      r6cCs|durg}n t|trt|}|sggfStt}|D]\}}|}d| }|| |qt dd|D}dd|D}||fS)amCanonicalize headers for signing. See: https://cloud.google.com/storage/docs/access-control/signed-urls#about-canonical-extension-headers :type headers: Union[dict|List(Tuple(str,str))] :param headers: (Optional) Additional HTTP headers to be included as part of the signed URLs. See: https://cloud.google.com/storage/docs/xml-api/reference-headers Requests using the signed URL *must* pass the specified header (name and value) with each request for the URL. :rtype: str :returns: List of headers, normalized / sortted per the URL refernced above. N css"|] \}}|d|fVqdS),N)join).0keyvalrrr s z(get_canonical_headers..cSsg|]}dj|qS)z{}:{})r )r:itemrrr sz)get_canonical_headers..) rdictlistitems collections defaultdictlowerstripr9splitappendsorted)headers normalizedr;r<ordered_headerscanonical_headersrrrget_canonical_headerss     rN _Canonical)methodresourcequery_parametersrJcCsxt|\}}|dkrd}|d|durt||g|Stdd|D}tjjj |}d ||}t||||S)ahCanonicalize method, resource per the V2 spec. :type method: str :param method: The HTTP verb that will be used when requesting the URL. Defaults to ``'GET'``. If method is ``'RESUMABLE'`` then the signature will additionally contain the `x-goog-resumable` header, and the method changed to POST. See the signed URL docs regarding this flow: https://cloud.google.com/storage/docs/access-control/signed-urls :type resource: str :param resource: A pointer to a specific resource (typically, ``/bucket-name/path/to/blob.txt``). :type query_parameters: dict :param query_parameters: (Optional) Additional query parameters to be included as part of the signed URLs. See: https://cloud.google.com/storage/docs/xml-api/reference-headers#query :type headers: Union[dict|List(Tuple(str,str))] :param headers: (Optional) Additional HTTP headers to be included as part of the signed URLs. See: https://cloud.google.com/storage/docs/xml-api/reference-headers Requests using the signed URL *must* pass the specified header (name and value) with each request for the URL. :rtype: :class:_Canonical :returns: Canonical method, resource, query_parameters, and headers. RESUMABLEPOSTzx-goog-resumable:startNcss,|]\}}||r|pdfVqdS)N)rErFr:r;valuerrrr=s  z"canonicalize_v2..z{}?{}) rNrHrOrIrBr*movesurllibparse urlencoder )rPrQrRrJ_ normalized_qp encoded_qpcanonical_resourcerrrcanonicalize_v2s   r`rUGETcCst|}t||| | }|j|pd|pdt|g}||j||jd|}| r:| r:t || | }| ||d}nt |||}|durH||d<|durP||d<| durX| |d<| |j t |}dj||tjjj|d S) a6Generate a V2 signed URL to provide query-string auth'n to a resource. .. note:: Assumes ``credentials`` implements the :class:`google.auth.credentials.Signing` interface. Also assumes ``credentials`` has a ``signer_email`` property which identifies the credentials. .. note:: If you are on Google Compute Engine, you can't generate a signed URL. Follow `Issue 922`_ for updates on this. If you'd like to be able to generate a signed URL from GCE, you can use a standard service account from a JSON file rather than a GCE service account. See headers `reference`_ for more details on optional arguments. .. _Issue 922: https://github.com/GoogleCloudPlatform/ google-cloud-python/issues/922 .. _reference: https://cloud.google.com/storage/docs/reference-headers :type credentials: :class:`google.auth.credentials.Signing` :param credentials: Credentials object with an associated private key to sign text. :type resource: str :param resource: A pointer to a specific resource (typically, ``/bucket-name/path/to/blob.txt``). Caller should have already URL-encoded the value. :type expiration: Union[Integer, datetime.datetime, datetime.timedelta] :param expiration: Point in time when the signed URL should expire. If a ``datetime`` instance is passed without an explicit ``tzinfo`` set, it will be assumed to be ``UTC``. :type api_access_endpoint: str :param api_access_endpoint: (Optional) URI base. Defaults to empty string. :type method: str :param method: The HTTP verb that will be used when requesting the URL. Defaults to ``'GET'``. If method is ``'RESUMABLE'`` then the signature will additionally contain the `x-goog-resumable` header, and the method changed to POST. See the signed URL docs regarding this flow: https://cloud.google.com/storage/docs/access-control/signed-urls :type content_md5: str :param content_md5: (Optional) The MD5 hash of the object referenced by ``resource``. :type content_type: str :param content_type: (Optional) The content type of the object referenced by ``resource``. :type response_type: str :param response_type: (Optional) Content type of responses to requests for the signed URL. Ignored if content_type is set on object/blob metadata. :type response_disposition: str :param response_disposition: (Optional) Content disposition of responses to requests for the signed URL. :type generation: str :param generation: (Optional) A value that indicates which generation of the resource to fetch. :type headers: Union[dict|List(Tuple(str,str))] :param headers: (Optional) Additional HTTP headers to be included as part of the signed URLs. See: https://cloud.google.com/storage/docs/xml-api/reference-headers Requests using the signed URL *must* pass the specified header (name and value) with each request for the URL. :type service_account_email: str :param service_account_email: (Optional) E-mail address of the service account. :type access_token: str :param access_token: (Optional) Access token for a service account. :type query_parameters: dict :param query_parameters: (Optional) Additional query parameters to be included as part of the signed URLs. See: https://cloud.google.com/storage/docs/xml-api/reference-headers#query :raises: :exc:`TypeError` when expiration is not a valid type. :raises: :exc:`AttributeError` if credentials is not an instance of :class:`google.auth.credentials.Signing`. :rtype: str :returns: A signed URL you can use to access the resource until expiration. rU rNresponse-content-typeresponse-content-disposition generationz"{endpoint}{resource}?{querystring})endpointrQ querystring)r/r`rPstrextendrJrHrQr9 _sign_messager updaterRrIrBr r*rXrYrZr[)rrQrapi_access_endpointrP content_md5 content_type response_typeresponse_dispositionrerJrRservice_account_email access_tokenexpiration_stamp canonicalelements_to_signrrsigned_query_paramssorted_signed_query_paramsrrrgenerate_signed_url_v2s@q      rxi: zhttps://storage.googleapis.comc$Cs@t|}|durt\}}n|}|dd}| }| r| s#t||j}d|}d||}| dur4i} |dur<|| d<|durD|| d<dd| D}d |vrZtjjj |j | d <| d krfd }d | d<t | \}}d |d}d dd|D}| duri} n dd| D} d| d<|| d<|| d<|| d<|| d<|dur|| d<|dur|| d<| dur| | d<t| }t|}d|vr|d}nd}||||||g}d |}t|d}d|||g} d | }!| r| rt|!| | }"t|"}#t|#d}"n||!d}#t|#d}"d ||||"S)!aGenerate a V4 signed URL to provide query-string auth'n to a resource. .. note:: Assumes ``credentials`` implements the :class:`google.auth.credentials.Signing` interface. Also assumes ``credentials`` has a ``signer_email`` property which identifies the credentials. .. note:: If you are on Google Compute Engine, you can't generate a signed URL. Follow `Issue 922`_ for updates on this. If you'd like to be able to generate a signed URL from GCE, you can use a standard service account from a JSON file rather than a GCE service account. See headers `reference`_ for more details on optional arguments. .. _Issue 922: https://github.com/GoogleCloudPlatform/ google-cloud-python/issues/922 .. _reference: https://cloud.google.com/storage/docs/reference-headers :type credentials: :class:`google.auth.credentials.Signing` :param credentials: Credentials object with an associated private key to sign text. That credentials must provide signer_email only if service_account_email and access_token are not passed. :type resource: str :param resource: A pointer to a specific resource (typically, ``/bucket-name/path/to/blob.txt``). Caller should have already URL-encoded the value. :type expiration: Union[Integer, datetime.datetime, datetime.timedelta] :param expiration: Point in time when the signed URL should expire. If a ``datetime`` instance is passed without an explicit ``tzinfo`` set, it will be assumed to be ``UTC``. :type api_access_endpoint: str :param api_access_endpoint: (Optional) URI base. Defaults to "https://storage.googleapis.com/" :type method: str :param method: The HTTP verb that will be used when requesting the URL. Defaults to ``'GET'``. If method is ``'RESUMABLE'`` then the signature will additionally contain the `x-goog-resumable` header, and the method changed to POST. See the signed URL docs regarding this flow: https://cloud.google.com/storage/docs/access-control/signed-urls :type content_md5: str :param content_md5: (Optional) The MD5 hash of the object referenced by ``resource``. :type content_type: str :param content_type: (Optional) The content type of the object referenced by ``resource``. :type response_type: str :param response_type: (Optional) Content type of responses to requests for the signed URL. Ignored if content_type is set on object/blob metadata. :type response_disposition: str :param response_disposition: (Optional) Content disposition of responses to requests for the signed URL. :type generation: str :param generation: (Optional) A value that indicates which generation of the resource to fetch. :type headers: dict :param headers: (Optional) Additional HTTP headers to be included as part of the signed URLs. See: https://cloud.google.com/storage/docs/xml-api/reference-headers Requests using the signed URL *must* pass the specified header (name and value) with each request for the URL. :type query_parameters: dict :param query_parameters: (Optional) Additional query parameters to be included as part of the signed URLs. See: https://cloud.google.com/storage/docs/xml-api/reference-headers#query :type service_account_email: str :param service_account_email: (Optional) E-mail address of the service account. :type access_token: str :param access_token: (Optional) Access token for a service account. :raises: :exc:`TypeError` when expiration is not a valid type. :raises: :exc:`AttributeError` if credentials is not an instance of :class:`google.auth.credentials.Signing`. :rtype: str :returns: A signed URL you can use to access the resource until expiration. Nz{}/auto/storage/goog4_requestz{}/{}z Content-Typez Content-MD5cSsg|]}|qSr)rE)r:r;rrrr?4z*generate_signed_url_v4..hostHostrSrTstartzx-goog-resumablerb;cSsg|]\}}|qSrr)r:r;r\rrrr?@rzcSsi|] \}}||p dqS)rUrrVrrr Esz*generate_signed_url_v4..zGOOG4-RSA-SHA256zX-Goog-AlgorithmzX-Goog-Credentialz X-Goog-DatezX-Goog-ExpireszX-Goog-SignedHeadersrcrdrezx-goog-content-sha256zUNSIGNED-PAYLOADrz{}{}?{}&X-Goog-Signature={})r6get_v4_now_dtstampsrrr r*rXrYrZurlparsenetlocupperrNr9rB _url_encoder@hashlibsha256r hexdigestrjr b64decodebinasciihexlifydecoder)$rrQrrlrPrmrnrorprerJrRrqrr_request_timestampexpiration_secondsrequest_timestamp datestamp client_emailcredential_scope credential header_namesrMrLcanonical_header_stringsigned_headerscanonical_query_stringlowercased_headerspayloadcanonical_elementscanonical_requestcanonical_request_hashstring_elementsrrrrrrgenerate_signed_url_v4sv             rcCs&t}|d}|d}||fS)z~Get current timestamp and datestamp in V4 valid format. :rtype: str, str :returns: Current timestamp, datestamp. z%Y%m%dT%H%M%SZz%Y%m%d)r&strftimedate)r- timestamprrrrrs rc Cst|}d}d|}d|dd}tdt|di}t }|||||d}|j t j j jkr=td |jt|jd} | d S) aSigns a message. :type message: str :param message: The message to be signed. :type access_token: str :param access_token: Access token for a service account. :type service_account_email: str :param service_account_email: E-mail address of the service account. :raises: :exc:`TransportError` if an `access_token` is unauthorized. :rtype: str :returns: The signature of the message. rTzXhttps://iamcredentials.googleapis.com/v1/projects/-/serviceAccounts/{}:signBlob?alt=jsonzBearer zapplication/json) Authorizationz Content-typerzutf-8)urlrPbodyrJz'Error calling the IAM signBytes API: {} signedBlob)r _to_bytesr jsondumpsrrrrRequeststatusr*rX http_clientOKrTransportErrordataloads) messagerrrqrPrrJrrequestresponserrrrrjs"  rjcCs dd|D}dt|S)zEncode query params into URL. :type query_params: dict :param query_params: Query params to be encoded. :rtype: str :returns: URL encoded query params. cSs$g|]\}}dt|t|qS)z{}={})r _quote_param)r:namerWrrrr?sz_url_encode..&)rBr9rI) query_paramsparamsrrrrs rcCs&t|ts t|}tjjjj|ddS)zQuote query param. :type param: Any :param param: Query param to be encoded. :rtype: str :returns: URL encoded query param. ~)safe)rbytesrhr*rXrYrZquote)paramrrrrs r) rUraNNNNNNNNN)%rrrCr$rrr*google.auth.credentialsr google.authrgoogle.auth.transportr google.cloudrutcnowr&r rr r/r+r%r0r6rN namedtuplerOr`rxr3DEFAULT_ENDPOINTrrrjrrrrrrsr   *%6   ^ ,