o ˷ec @s4ddlZddlZddlZddlZddlZddlZddlZddlZddlZ ddl m Z ddl m Z ddlmZejjZdZddZdd Zd d ZeejejfZd d ZddZedgdZddZ           d$ddZdZdZ e dddddddddddf 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/ideatree/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 cCs`t|tjrtjtjd}||}t|tjr!t|}|d}t|ts.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_datetimeint TypeErrorr )rnowmicrosrrrget_expiration_seconds_v2[s    r.cCst|ts tdt|tjtjd}t|tr|}t|t j r2|j dur.|jtjd}||}t|t j r>t| }|t krItdt |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!Nz.Max allowed expiration interval is seven days )r_EXPIRATION_TYPESr+r r&r'rr(r*r$r"r% total_seconds SEVEN_DAYS ValueError)rr,secondsrrrget_expiration_seconds_v4}s$      r4cCs|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 )r8itemrrr sz)get_canonical_headers..) rdictlistitems collections defaultdictlowerstripr7splitappendsorted)headers normalizedr9r:ordered_headerscanonical_headersrrrget_canonical_headerss     rL _Canonical)methodresourcequery_parametersrHcCsvt|\}}|dkrd}|d|durt||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)rCrDr8r9valuerrrr;s  z"canonicalize_v2..?)rLrFrMrGr@urllibparse urlencode)rNrOrPrH_ normalized_qp encoded_qpcanonical_resourcerrrcanonicalize_v2s   r^rSGETcCst|}t||| | }|j|pd|pdt|g}||j||jd|}| r:| r:t || | }| ||d}nt |||}|durH||d<|durP||d<| durX| |d<| |j t |}dj||tj|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. rS rNresponse-content-typeresponse-content-disposition generationz"{endpoint}{resource}?{querystring})endpointrO querystring)r.r^rNstrextendrHrFrOr7 _sign_messager updaterPrGr@r rWrXrY)rrOrapi_access_endpointrN content_md5 content_type response_typeresponse_dispositionrcrHrPservice_account_email access_tokenexpiration_stamp canonicalelements_to_signrrsigned_query_paramssorted_signed_query_paramsrrrgenerate_signed_url_v2s@q       rvi: zhttps://storage.googleapis.comc$Cs>t|}|durt\}}n|}|dd}| }| r| s#t||j}|d}|d|}| dur5i} |dur=|| d<|durE|| d<dd| D}d |vrYtj|j| d <|d kred }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_request/z Content-Typez Content-MD5cSsg|]}|qSr)rC)r8r9rrrr=3z*generate_signed_url_v4..hostHostrQrRstartzx-goog-resumabler`;cSsg|]\}}|qSrr)r8r9rZrrrr=?rycSsi|] \}}||p dqS)rSrrTrrr Dsz*generate_signed_url_v4..zGOOG4-RSA-SHA256zX-Goog-AlgorithmzX-Goog-Credentialz X-Goog-DatezX-Goog-ExpireszX-Goog-SignedHeadersrarbrczx-goog-content-sha256zUNSIGNED-PAYLOADrz{}{}?{}&X-Goog-Signature={})r4get_v4_now_dtstampsrrrWrXurlparsenetlocupperrLr7r@ _url_encoder>hashlibsha256r hexdigestrhr b64decodebinasciihexlifydecoderr )$rrOrrjrNrkrlrmrnrcrHrProrp_request_timestampexpiration_secondsrequest_timestamp datestamp client_emailcredential_scope credential header_namesrKrJcanonical_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 kr.&)r@r7rG) query_paramsparamsrrrrs rcCs"t|ts t|}tjj|ddS)zQuote query param. :type param: Any :param param: Query param to be encoded. :rtype: str :returns: URL encoded query param. ~)safe)rbytesrfrWrXquote)paramrrrrs r) rSr_NNNNNNNNN)&rrrAr$rrrrWgoogle.auth.credentialsr google.authrgoogle.auth.transportr google.cloudrutcnowr&r rr r.r*r%r/r4rL namedtuplerMr^rvr1DEFAULT_ENDPOINTrrrhrrrrrrst   (%6   ^ ,