o tBhv.@sdZddlmZddlmZddlmZddlmZddl m Z ddl Z ddl Z ddl Z ddlZddlmZdd lmZdd lmZdd lmZGd d d eZGdddeZGddde jZGdddeZddZddZdS)zyBatch updates / deletes of storage buckets / blobs. See https://cloud.google.com/storage/docs/json_api/v1/how-tos/batch ) encode_noop) Generator)MIMEApplication) MIMEMultipart)ParserN)_helpers) exceptions) Connection)_DEFAULT_TIMEOUTcs eZdZdZfddZZS)MIMEApplicationHTTPa?MIME type for ``application/http``. Constructs payload from headers and body :type method: str :param method: HTTP method :type uri: str :param uri: URI for HTTP request :type headers: dict :param headers: HTTP headers :type body: str :param body: (Optional) HTTP payload cst|trt|}d|d<t||d<|durd}d||fg}|ddt|D|d||d |}t j rKt ||d tdStt|j }||d tdS) Nzapplication/jsonz Content-TypezContent-Lengthz%s %s HTTP/1.1cSsg|] \}}d||fqS)z%s: %s).0keyvaluer r q/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/google/cloud/storage/batch.py ?sz0MIMEApplicationHTTP.__init__..z http) isinstancedictjsondumpslenextendsorteditemsappendjoinsixPY2r__init__rsuperr )selfmethoduriheadersbodylinespayload super_init __class__r rr 6s"       zMIMEApplicationHTTP.__init__)__name__ __module__ __qualname____doc__r __classcell__r r r*rr #sr c@s.eZdZdZed ddZddZddZdS) _FutureDictzvClass to hold a future value for a deferred request. Used by for requests that get sent in a :class:`Batch`. NcCstd||f)a,Stand-in for dict.get. :type key: object :param key: Hashable dictionary key. :type default: object :param default: Fallback value to dict.get. :raises: :class:`KeyError` always since the future is intended to fail as a dictionary. z&Cannot get(%r, default=%r) on a futureKeyError)rdefaultr r rgetSs z_FutureDict.getcCstd|f)zStand-in for dict[key]. :type key: object :param key: Hashable dictionary key. :raises: :class:`KeyError` always since the future is intended to fail as a dictionary. z Cannot get item %r from a futurer2)r"rr r r __getitem__bs z_FutureDict.__getitem__cCstd||f)a'Stand-in for dict[key] = value. :type key: object :param key: Hashable dictionary key. :type value: object :param value: Dictionary value. :raises: :class:`KeyError` always since the future is intended to fail as a dictionary. zCannot set %r -> %r on a futurer2)r"rrr r r __setitem__ms z_FutureDict.__setitem__N)r,r-r.r/ staticmethodr5r6r7r r r rr1Ms   r1cs4eZdZdZfddZddZeddZZS)_FutureResponsezEReponse that returns a placeholder dictionary for a batched requests.cstt|||_d|_dS)N)r!r:r _future_dict status_code)r" future_dictr*r rr s z_FutureResponse.__init__cC|jSr8r<r"r r rrsz_FutureResponse.jsoncCr?r8r@rAr r rcontentsz_FutureResponse.content) r,r-r.r/r rpropertyrBr0r r r*rr:|s  r:cs`eZdZdZdZfddZefddZddZd d Z d d Z d dZ ddZ ddZ ZS)BatchzProxy an underlying connection, batching up change operations. :type client: :class:`google.cloud.storage.client.Client` :param client: The client to use for making connections. ics6|jj}|jj}tt|j|||dg|_g|_dS)N) client_info api_endpoint) _connection API_BASE_URL _client_infor!rDr _requests_target_objects)r"clientrFrEr*r rr s  zBatch.__init__cCs\t|j|jkrtd|j|j|||||ft}|j||dur*||_t|S)auOverride Connection: defer actual HTTP request. Only allow up to ``_MAX_BATCH_SIZE`` requests to be deferred. :type method: str :param method: The HTTP method to use in the request. :type url: str :param url: The URL to send the request to. :type headers: dict :param headers: A dictionary of HTTP headers to send with the request. :type data: str :param data: The data to send as the body of the request. :type target_object: object :param target_object: (Optional) This allows us to enable custom behavior in our batch connection. Here we defer an HTTP request and complete initialization of the object at a later time. :type timeout: float or tuple :param timeout: (Optional) The amount of time, in seconds, to wait for the server response. See: :ref:`configuring_timeouts` :rtype: tuple of ``response`` (a dictionary of sorts) and ``content`` (a string). :returns: The HTTP response object and the content of the response. z#Too many deferred requests (max %d)N) rrJ_MAX_BATCH_SIZE ValueErrorrr1rK _propertiesr:)r"r#urlr%data target_objecttimeoutresultr r r _do_requests" zBatch._do_requestc Cst|jdkr tdt}t}|jD]\}}}}}t||||}|||}qtjr1t } nt } t | dd} | || } | dd\} }t|j||fS)zPrepares headers and body for a batch request. :rtype: tuple (dict, str) :returns: The pair of headers and body of the batch request to be sent. :raises: :class:`ValueError` if no requests have been deferred. rzNo deferred requestsFz )rrJrNrr r attachrrioBytesIOStringIOrflattengetvaluesplitr_headers) r"multirSr#r$r%r&_timeout subrequestbuf generatorr(_r r r_prepare_batch_requests     zBatch._prepare_batch_requestc Csd}t|jt|krtdt|j|D].\}}d|jkr$dks*n|p(|}q|durCz||_WqtyB|j|_Yqwq|durMt |dS)a6Apply all the batch responses to the futures created. :type responses: list of (headers, payload) tuples. :param responses: List of headers and payloads from each response in the batch. :raises: :class:`ValueError` if no requests have been deferred. Nz&Expected a response for every request.i,) rrKrNzipr=rrOrBrfrom_http_response)r" responsesexception_argsrR subresponser r r_finish_futuress      zBatch._finish_futurescCsJ|\}}}d|j}|jjjd||||d}tt|}|||S)zSubmit a single `multipart/mixed` request with deferred requests. :rtype: list of tuples :returns: one ``(headers, payload)`` tuple per deferred request. z%s/batch/storage/v1POST)rQr%rS)rerH_client_base_connection _make_requestlist_unpack_batch_responserl)r"r%r&rSrPresponserir r rfinishs    z Batch.finishcCs|jjS)z"Return the topmost batch, or None.)rn current_batchrAr r rcurrentsz Batch.currentcCs|j||Sr8)rn _push_batchrAr r r __enter__s zBatch.__enter__cCs>z|dur|W|jdSW|jdS|jwr8)rtrn _pop_batch)r"exc_typeexc_valexc_tbr r r__exit__#s  zBatch.__exit__)r,r-r.r/rMr r rUrerlrtrvrxr}r0r r r*rrDs  - rDcCsHt|jdd}dd|d|jg}tjr||S|| dS)zdConvert response, content -> (multipart) email.message. Helper for _unpack_batch_response. z content-typer sContent-Type: s MIME-Version: 1.0 utf-8) r _to_bytesr%r5rrBrrparsestrdecode)parserrs content_type faux_messager r r_generate_faux_mime_message+s  rc cst}t||}t|jtstd|jD]K}|jdd\}}|dd\}}}||}|j} t|j } | d} t } t j dd| d | _t|| _| j| | d | _| Vqd S) a Convert requests.Response -> [(headers, payload)]. Creates a generator of tuples of emulating the responses to :meth:`requests.Session.request`. :type response: :class:`requests.Response` :param response: HTTP response / headers from a request. zBad response: not multi-part rV z Content-IDBATCHzcontentid://{})r#rPrN)rrr_payloadrqrNr]rrr^r5requestsResponseRequestformatpreparerequestintr=r%updateencode_content) rsrmessagera status_linerestrdstatus sub_messager( msg_headers content_idrkr r rrr?s.          rr)r/email.encodersremail.generatorremail.mime.applicationremail.mime.multipartr email.parserrrXrrr google.cloudrrgoogle.cloud.storage._httpr google.cloud.storage.constantsr r objectr1rr:rDrrrr r r rs*         */