g9dZddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z dZ Gdd e jZGd d e jZGd d e j ZGddej"Zy)zTransport adapter for Async HTTP (aiohttp). NOTE: This async support is experimental and marked internal. This surface may change in minor releases. )absolute_importN) exceptions) transport)requestscXeZdZdZdZdZedZedZedZ dZ dZ y ) _CombinedResponsea@ In order to more closely resemble the `requests` interface, where a raw and deflated content could be accessed at once, this class lazily reads the stream in `transport.Response` so both return forms can be used. The gzip and deflate transfer-encodings are automatically decoded for you because the default parameter for autodecompress into the ClientSession is set to False, and therefore we add this class to act as a wrapper for a user to be able to access both the raw and decoded response bodies - mirroring the sync implementation. c ||_d|_yN) _response _raw_contentselfresponses \/var/www/openai/venv/lib/python3.12/site-packages/google/auth/transport/_aiohttp_requests.py__init__z_CombinedResponse.__init__3s! c^|jj}d|vxr|ddk(xs|ddk(S)NContent-Encodinggzipdeflater headers)rrs r_is_compressedz _CombinedResponse._is_compressed7sD..((!W, & '6 1 8)*i7 rc.|jjSr r statusrs rrz_CombinedResponse.status>~~$$$rc.|jjSr rrs rrz_CombinedResponse.headersB~~%%%rc.|jjSr r contentrs rdataz_CombinedResponse.dataFr!rcK|j1|jjjd{|_|jS7wr )r r r$readrs r raw_contentz_CombinedResponse.raw_contentJsB    $&*nn&<&<&A&A&C CD    !Ds4AA AcK|jd{|jrStjj |j j d}|j|j}|S|jS7sw)Nr) r(rurllib3r MultiDecoderr r decompressr )rdecoder decompresseds rr$z_CombinedResponse.contentOs|      &&33&&'9:G#--d.?.?@L     !sB B A4B N) __name__ __module__ __qualname____doc__rrpropertyrrr%r(r$rrr r &sW ! %%&&&&! !rr cFeZdZdZdZedZedZedZy) _Responsezz Requests transport response adapter. Args: response (requests.Response): The raw Requests response. c||_yr )r rs rrz_Response.__init__ds !rc.|jjSr rrs rrz_Response.statusgrrc.|jjSr rrs rrz_Response.headerskr!rc.|jjSr r#rs rr%z_Response.dataor!rN) r/r0r1r2rr3rrr%r4rrr6r6\sH"%%&&&&rr6c(eZdZdZddZdddefdZy)RequestaRequests request adapter. This class is used internally for making requests using asyncio transports in a consistent way. If you use :class:`AuthorizedSession` you do not need to construct or use this class directly. This class can be useful if you want to manually refresh a :class:`~google.auth.credentials.Credentials` instance:: import google.auth.transport.aiohttp_requests request = google.auth.transport.aiohttp_requests.Request() credentials.refresh(request) Args: session (aiohttp.ClientSession): An instance :class:`aiohttp.ClientSession` used to make HTTP requests. If not specified, a session will be created. .. automethod:: __call__ NcX|!|jrtjd||_y)Nz) rr[refresh_status_codesmax_refresh_attemptsrefresh_timeout auth_requestrCrR __class__s rrzAuthorizedSession.__init__su /9&9&%9"%9" / )%)"++- $\\^ /rc K|rA|jD].} t|| tus|| jd|| <0t j |j |jdd4d{|_t|j} | |_ |jdd} ||jni} | |jn tj|j|} |} tj | t"j$5}|j&j)| ||| d{dddtj | t"j$5}t+t,|^||f|| |d |d{}dddj0} j2|j4vrE| |j6kr5tj8j;d |j2| d z|j6| |jn tj|j|} tj | t"j$5}|j<4d{|j>jAd|j&jB| d{dddd{ddd|j0} |j.||f||| || d zd |d{cdddd{Sdddd{|S77#1swYxYw7#1swYxYw777#1d{7swYxYw#1swYxYw7q7c7T#1d{7swYSxYww) aImplementation of Authorized Session aiohttp request. Args: method (str): The http request method used (e.g. GET, PUT, DELETE) url (str): The url at which the http request is sent. data (Optional[dict]): Dictionary, list of tuples, bytes, or file-like object to send in the body of the Request. headers (Optional[dict]): Dictionary of HTTP Headers to send with the Request. timeout (Optional[Union[float, aiohttp.ClientTimeout]]): The amount of time in seconds to wait for the server response with each individual request. Can also be passed as an ``aiohttp.ClientTimeout`` object. max_allowed_time (Optional[float]): If the method runs longer than this, a ``Timeout`` exception is automatically raised. Unlike the ``timeout`` parameter, this value applies to the total method execution time, even if multiple requests are made under the hood. Mind that it is not guaranteed that the timeout error is raised at ``max_allowed_time``. It might take longer, for example, if an underlying request takes a lot of time, but the request itself does not timeout, e.g. if a large file is being transmitted. The timout error will be raised after such request completes. zutf-8 trust_envF)rCrlN_credential_refresh_attemptr)rErDz;Refreshing credentials due to a %s response. Attempt %s/%s.)r%rmax_allowed_timerErm)"keystypebytesdecoderFrGr>getrar<r`popcopy functoolspartialr TimeoutGuardrMrNr[before_requestrZrXrJremaining_timeoutrr\r]rHinforercrun_in_executorrefresh)rrPrOr%rrorErCrRkeyrirmrequest_headersremaining_timeguardrrjs rrJzAuthorizedSession.requestsfV ||~ %.#*3<#6#6w#?GCL&(( 11jje4   '"4#=#=>L!-D +1**5RTU*V '180CgllnO ?""&&t'9'97K  .N&&~w7K7KLPU&&55 &#M &&~w7K7KLPU!&'8$!G"+# "  "M#44N4#=#==/$2L2LL  %%QOO/!3.. &&"**4+=+=wO **"G$8$8#111"jj88 $"2"2":":L 21"'!8!8)T\\ #%3#0Ka0O   O     de 2ML MLF 2 2111 O     ds3*M<AM<<L=M<BM&"L?LL,M&0L, L)L,C M&M,L9 -M05L?%L; &L?* M5L=6M:4M&.M /M&2 M<>M"?M<M$M<LL& !M&)L,,L6 1M&9M;L?=M?M MM MM M&"M<$M<&M9,M/ -M94M<) r/r0r1r2rDEFAULT_REFRESH_STATUS_CODESDEFAULT_MAX_REFRESH_ATTEMPTSrrVrJ __classcell__)rjs@rrXrXsH$R'CC&CC04 BBrrX)r2 __future__rrMrwrFr* google.authrrgoogle.auth.transportrrVResponser r6r<rGrXr4rrrsz '"!*3! **3!l& ""&0Q*iQ*h--r