a !f,@sdZddlmZddlmZddlmZddlmZddl m Z ddl Z ddl Z ddl Z ddlmZdd 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|tr(t|}d|d<t||d<|dur4d}|d|dg}|ddt|D|d||d |}t |d t dS) Nzapplication/jsonz Content-TypezContent-Length z HTTP/1.1cSsg|]\}}|d|qS)z: ).0keyvaluerr[/var/www/html/python-backend/venv/lib/python3.9/site-packages/google/cloud/storage/batch.py =z0MIMEApplicationHTTP.__init__..z http) isinstancedictjsondumpslenextendsorteditemsappendjoinsuper__init__r)selfmethoduriheadersbodylinespayload __class__rrr!5s      zMIMEApplicationHTTP.__init__)__name__ __module__ __qualname____doc__r! __classcell__rrr)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|d|ddS)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(z , default=z ) on a futureNKeyError)rdefaultrrrgetJs z_FutureDict.getcCstd|ddS)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. zCannot get item z from a futureNr1)r"rrrr __getitem__Ys z_FutureDict.__getitem__cCstd|d|ddS)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. z Cannot set z -> z on a futureNr1)r"rrrrr __setitem__ds z_FutureDict.__setitem__)N)r+r,r-r. staticmethodr4r5r6rrrrr0Ds   r0cs4eZdZdZfddZddZeddZZS)_FutureResponsezEReponse that returns a placeholder dictionary for a batched requests.cstt|||_d|_dS)N)r r8r! _future_dict status_code)r"Z future_dictr)rrr!vsz_FutureResponse.__init__cCs|jSNr:r"rrrr{sz_FutureResponse.jsoncCs|jSr<r=r>rrrcontent~sz_FutureResponse.content) r+r,r-r.r!rpropertyr?r/rrr)rr8ss  r8cs`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_URLZ _client_infor rAr! _requests_target_objects)r"clientrCrBr)rrr!s zBatch.__init__cCs\t|j|jkrtd|j|j|||||ft}|j||durT||_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) rrF_MAX_BATCH_SIZE ValueErrorrr0rG _propertiesr8)r"r#urlr%data target_objecttimeoutresultrrr _do_requests" zBatch._do_requestc Cst|jdkrtdt}t}|jD]*\}}}}}t||||}|||}q&t} 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 )rrFrJrr r attachioStringIOrflattengetvaluesplitr_headers) r"multirOr#r$r%r&_timeout subrequestbuf generatorr(_rrr_prepare_batch_requests   zBatch._prepare_batch_requestc Csd}t|jt|krtdt|j|D]\\}}d|jkrHdksTn|pP|}q*|dur*z||_Wq*ty|j|_Yq*0q*|durt |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.,) rrGrJzipr;rrKr?rfrom_http_response)r" responsesZexception_argsrN subresponserrr_finish_futuress   zBatch._finish_futurescCsn|\}}}|jd}|jjjd||||d}d|jkrHdksTnt|tt |}| ||S)zSubmit a single `multipart/mixed` request with deferred requests. :rtype: list of tuples :returns: one ``(headers, payload)`` tuple per deferred request. z/batch/storage/v1POST)rMr%rOrarb) r`rE_clientZ_base_connection _make_requestr;rrdlist_unpack_batch_responserg)r"r%r&rOrLresponsererrrfinishs     z Batch.finishcCs|jjS)z"Return the topmost batch, or None.)riZ current_batchr>rrrcurrentsz Batch.currentcCs|j||Sr<)riZ _push_batchr>rrr __enter__s zBatch.__enter__cCs0z|dur|W|jn |j0dSr<)rnriZ _pop_batch)r"exc_typeexc_valexc_tbrrr__exit__s zBatch.__exit__)r+r,r-r.rIr!r rQr`rgrnrorprtr/rrr)rrAs  -rAcCs8t|jdd}dd|d|jg}||dS)zdConvert response, content -> (multipart) email.message. Helper for _unpack_batch_response. z content-typer rsContent-Type: s MIME-Version: 1.0 utf-8)r _to_bytesr%r4rr?parsestrdecode)parserrm content_typeZ faux_messagerrr_generate_faux_mime_message#s  r{c cst}t||}t|jts$td|jD]}|jdd\}}|dd\}}}||}|j} t|j } | d} t } t j dd| d | _t|| _| j| | d | _| Vq*d 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 rRr z Content-IDZBATCHz contentid://)r#rLruN)rr{r_payloadrkrJrXrwrrYr4requestsResponseRequestpreparerequestintr;r%updateencode_content) rmrymessager\Z status_linerestr_statusZ sub_messager(Z msg_headersZ content_idrfrrrrl4s&           rl)r.email.encodersremail.generatorrZemail.mime.applicationrZemail.mime.multipartr email.parserrrTrrZ google.cloudrrZgoogle.cloud.storage._httpr Zgoogle.cloud.storage.constantsr r objectr0rr8rAr{rlrrrrs&         "/!