B uf3d @s:ddlZddlZddlZddlZddlZddlZddlZddlZddlZ ddl m Z ddl m Z ddlmZddlmZddlmZejjZdZdd Zd d Zd d ZeejejfZddZddZedddddgZddZ d)ddZ!dZ"dZ#e#dddddddddddf dd Z$d!d"Z%d#d$Z&d%d&Z'd'd(Z(dS)*N) exceptions)requests)_helpers)_NOW)_UTCz[https://googleapis.dev/python/google-api-core/latest/auth.html#setting-up-a-service-accountcCs(t|tjjjs$tdt|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 credentialsZSigningAttributeErrorformattypeSERVICE_ACCOUNT_URL)r r_/home/ankuromar296_gmail_com/myenv/lib/python3.7/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. ascii)GoogleAccessIdExpires Signature)r sign_bytesencodebase64 b64encode signer_email)r expirationstring_to_signsignature_bytes signatureZservice_account_namerrrget_signed_query_params_v2@s rcCsXt|tjrtt}||}t|tjr:t|}|d}t|tsTtdt ||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. i@Bz=Expected an integer timestamp, datetime, or timedelta. Got %s) rdatetime timedeltarrrZ_microseconds_from_datetimeint TypeErrorr )rnowmicrosrrrget_expiration_seconds_v2_s     r&cCst|tstdt|tt}t|tr0|}t|tjr\|jdkrT|j t j d}||}t|tj rtt| }|tkrtdt|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 z=Expected an integer timestamp, datetime, or timedelta. Got %sN)tzinfoz.Max allowed expiration interval is seven days )r_EXPIRATION_TYPESr#r rrr"r r'replacerUTCr! total_seconds SEVEN_DAYS ValueError)rr$secondsrrrget_expiration_seconds_v4s       r/cCs|dkrg}nt|tr$t|}|s0ggfStt}x8|D]0\}}|}d| }|| |q@Wt 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 sz(get_canonical_headers..cSsg|]}dj|qS)z{}:{})r )r3itemrrr sz)get_canonical_headers..) rdictlistitems collections defaultdictlowerstripr2splitappendsorted)headers normalizedr4r5ordered_headerscanonical_headersrrrget_canonical_headerss    rG _Canonicalmethodresourcequery_parametersrCcCsvt|\}}|dkr"d}|d|dkr8t||g|Stdd|D}tj|}|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)r>r?)r3r4valuerrrr6sz"canonicalize_v2..?)rGrArHrBr;urllibparse urlencode)rIrJrKrC_Z normalized_qpZ encoded_qpZcanonical_resourcerrrcanonicalize_v2s   rUrNGETcCst|}t||| | }|j|p d|p&dt|g}||j||jd|}| rt| rtt || | }| ||d}n t |||}|dk r||d<|dk r||d<| dk r| |d<| |j t |}dj||tj|d S) aGenerate 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. 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](https://cloud.google.com/storage/docs/reference-headers) for more details on optional arguments. :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. rN )rrrNzresponse-content-typezresponse-content-disposition generationz"{endpoint}{resource}?{querystring})ZendpointrJZ querystring)r&rUrIstrextendrCrArJr2 _sign_messagerupdaterKrBr;r rQrRrS)r rJrapi_access_endpointrI content_md5 content_type response_typeresponse_dispositionrXrCrKservice_account_email access_tokenZexpiration_stamp canonicalZelements_to_signrrZsigned_query_paramsZsorted_signed_query_paramsrrrgenerate_signed_url_v2s8m         rei: zhttps://storage.googleapis.comc$CsHt|}|dkrt\}}n|}|dd}| }| r8| sFt||j}|d}|d|}| dkrji} |dk rz|| d<|dk r|| d<dd| D}d |krtj|j| d <|d krd }d | d<t | \}}d |d}d dd|D}| dkri} ndd| D} d| d<|| d<|| d<|| d<|| d<|dk rT|| d<|dk rf|| d<| dk rx| | d<t | }t |}d|kr|d}nd}||||||g}d |}t|d}d|||g} d | }!| r| rt|!| | }"t|"}#t|#d}"n ||!d}#t|#d}"d ||||"S)!a/Generate 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. 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](https://cloud.google.com/storage/docs/reference-headers) for more details on optional arguments. :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: 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_request/z Content-Typez Content-MD5cSsg|] }|qSr)r>)r3r4rrrr80sz*generate_signed_url_v4..hostHostrLrMstartzx-goog-resumablerW;cSsg|] \}}|qSrr)r3r4rTrrrr8<scSsi|]\}}|pd|qS)rNr)r3r4rOrrr Asz*generate_signed_url_v4..zGOOG4-RSA-SHA256zX-Goog-AlgorithmzX-Goog-Credentialz X-Goog-DatezX-Goog-ExpireszX-Goog-SignedHeaderszresponse-content-typezresponse-content-dispositionrXzx-goog-content-sha256zUNSIGNED-PAYLOADrz{}{}?{}&X-Goog-Signature={})r/get_v4_now_dtstampsrrrQrRurlparsenetlocupperrGr2r; _url_encoder9hashlibsha256r hexdigestr[r b64decodebinasciihexlifydecoderr )$r rJrr]rIr^r_r`rarXrCrKrbrcZ_request_timestampZexpiration_secondsZrequest_timestamp datestampZ client_emailZcredential_scopeZ credential header_namesrFrEZcanonical_header_stringZsigned_headersZcanonical_query_stringZlowercased_headerspayloadZcanonical_elementsZcanonical_requestZcanonical_request_hashZstring_elementsrrrrrrgenerate_signed_url_v4sq                r|cCs0ttjdd}|d}|d}||fS)z~Get current timestamp and datestamp in V4 valid format. :rtype: str, str :returns: Current timestamp, datestamp. N)r'z%Y%m%dT%H%M%SZz%Y%m%d)rrr)strftimedate)r$ timestampryrrrrm~s rmc Cst|}d}d|}d|dd}tdt|di}t }|||||d}|j t j j krxtd |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. rMzXhttps://iamcredentials.googleapis.com/v1/projects/-/serviceAccounts/{}:signBlob?alt=jsonzBearer zapplication/json) Authorizationz Content-typer{zutf-8)urlrIbodyrCz%Error calling the IAM signBytes API: Z signedBlob)rZ _to_bytesr jsondumpsrrrxrRequeststatushttpclientOKrZTransportErrordataloads) messagercrbrIrrCrrequestresponserrrrr[s r[cCs 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|]\}}t|dt|qS)=) _quote_param)r3namerOrrrr8sz_url_encode..&)r;r2rB)Z query_paramsparamsrrrrqs  rqcCs"t|tst|}tjj|ddS)zQuote query param. :type param: Any :param param: Query param to be encoded. :rtype: str :returns: URL encoded query param. ~)safe)rbytesrYrQrRquote)paramrrrrs r) rNrVNNNNNNNNN))rrvr<r rrrrrQZgoogle.auth.credentialsrZ google.authrZgoogle.auth.transportrZ google.cloudrZgoogle.cloud.storage._helpersrrutcnowZNOWrrrr&r"r!r(r/rG namedtuplerHrUrer,ZDEFAULT_ENDPOINTr|rmr[rqrrrrrsn     &%6  L +