a !f+@sdZddlmZddlmZddlmZddlmZddl m Z ddl Z ddl Z ddl Z ddlZddlmZdd lmZGd d d eZGd d d eZGdddeZGdddeZddZddZdS)zzBatch 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)make_exception) Connectioncs 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 or None :param body: HTTP payload cst|tr(t|}d|d<t||d<|dur4d}d||fg}|ddt|D|d||d |}t j rt ||d tntt|j }||d tdS) Napplication/json Content-TypezContent-Lengthz%s %s HTTP/1.1cSsg|]\}}d||fqS)z%s: %s).0keyvaluer r U/var/www/html/python-backend/venv/lib/python3.9/site-packages/gcloud/storage/batch.py ;sz0MIMEApplicationHTTP.__init__..z http) isinstancedictjsondumpslenextendsorteditemsappendjoinsixPY2r__init__rsuperr )selfmethoduriheadersbodylinespayload super_init __class__r rr 3s"         zMIMEApplicationHTTP.__init__)__name__ __module__ __qualname____doc__r __classcell__r r r*rr !sr c@seZdZdZdZdS) NoContentz*Emulate an HTTP '204 No Content' response.N)r,r-r.r/statusr r r rr1Isr1c@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||fdS)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 futureNKeyError)rdefaultr r rgetTs z_FutureDict.getcCstd|fdS)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 futureNr5)r"rr r r __getitem__ds z_FutureDict.__getitem__cCstd||fdS)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 futureNr5)r"rrr r r __setitem__os z_FutureDict.__setitem__)N)r,r-r.r/ staticmethodr8r9r:r r r rr4Ns   r4cs\eZdZdZdZfddZddZddZd d Zd d Z d dZ ddZ ddZ Z S)BatchzProxy an underlying connection, batching up change operations. :type client: :class:`gcloud.storage.client.Client` :param client: The client to use for making connections. ics$tt|||_g|_g|_dSN)r!r<r _client _requests_target_objects)r"clientr*r rr szBatch.__init__cCs\t|j|jkrtd|j|j||||ft}|j||durR||_t|fS)aOverride 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 or :class:`NoneType` :param target_object: 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. :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) rr?_MAX_BATCH_SIZE ValueErrorrr4r@ _propertiesr1)r"r#urlr%data target_objectresultr r r _do_requests zBatch._do_requestc Cst|jdkrtdt}|jD]$\}}}}t||||}||q"tjrXt }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 )rr?rCrr attachrPY3ioStringIOBytesIOrflattengetvaluesplitr_headers) r"multir#r$r%r& subrequestbuf generatorr(_r r r_prepare_batch_requests    zBatch._prepare_batch_requestcCsd}t|jt|krtdt|j|D]D\}}|\}}d|jkrPdks`n|p\||f}q*|dur*||_q*|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.i,)rr@rCzipr3rDr)r" responsesZexception_argsrGZ sub_responseZ resp_headersZ sub_payloadr r r_finish_futuress  zBatch._finish_futurescCsL|\}}d|j}|jjjd|||d\}}tt||}|||S)zSubmit a single `multipart/mixed` request w/ deferred requests. :rtype: list of tuples :returns: one ``(headers, payload)`` tuple per deferred request. z%s/batchPOST)rFr%)rYZ API_BASE_URLr> _connection _make_requestlist_unpack_batch_responser])r"r%r&rEresponsecontentr\r r rfinishs    z Batch.finishcCs|jjS)z"Return the topmost batch, or None.)r>Z current_batchr"r r rcurrentsz Batch.currentcCs|j||Sr=)r>Z _push_batchrfr r r __enter__s zBatch.__enter__cCs0z|dur|W|jn |j0dSr=)rer>Z _pop_batch)r"exc_typeexc_valexc_tbr r r__exit__s zBatch.__exit__)r,r-r.r/rBr rIrYr]rergrhrlr0r r r*rr<~s %r<cCsjt|tjs|d}|d}t|tjs4|d}dd|d|g}tjrV||S||dSdS)zdConvert response, content -> (multipart) email.message. Helper for _unpack_batch_response. zutf-8z content-typesContent-Type: s MIME-Version: 1.0 N)rr binary_typeencoderrparsestrdecode)parserrcrd content_typeZ faux_messager r r_generate_faux_mime_message s     rtccst}t|||}t|jts&td|jD]~}|jdd\}}|dd\}}}||} | j} | d} t| j } || d<t | } | r| drt | } | | fVq,d S) a|Convert response, content -> [(headers, payload)]. Creates a generator of tuples of emulating the responses to :meth:`httplib2.Http.request` (a pair of headers and payload). :type response: :class:`httplib2.Response` :param response: HTTP response / headers from a request. :type content: str :param content: Response payload with a batch response. zBad response: not multi-part rJ r r3r N)rrtr_payloadrarCrRrprrShttplib2Response startswithrloads)rcrdrrmessagerUZ status_linerestrXr3Z sub_messager(ctypeZ msg_headersr%r r rrb's        rb)r/email.encodersremail.generatorrZemail.mime.applicationrZemail.mime.multipartr email.parserrrMrryrZgcloud.exceptionsrZgcloud.storage.connectionrr objectr1r4r<rtrbr r r rs$       (0