vjIHdZddlmZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddl m Z ddlmZej eZdZGd d e j(ZGd d e j(ZGd de j.ZGddej0Zy)zTransport adapter for Async HTTP (aiohttp). NOTE: This async support is experimental and marked internal. This surface may change in minor releases. )absolute_importN)_helpers) 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 Z/var/www/html/venv/lib/python3.12/site-packages/google/auth/transport/_aiohttp_requests.py__init__z_CombinedResponse.__init__:s! c^|jj}d|vxr|ddk(xs|ddk(S)NContent-Encodinggzipdeflater headers)rrs r_is_compressedz _CombinedResponse._is_compressed>sD..((!W, & '6 1 8)*i7 rc.|jjSr r statusrs rrz_CombinedResponse.statusE~~$$$rc.|jjSr rrs rrz_CombinedResponse.headersI~~%%%rc.|jjSr r contentrs rdataz_CombinedResponse.dataMr"rcK|j1|jjjd{|_|jS7wr )rr r%readrs r raw_contentz_CombinedResponse.raw_contentQsB    $&*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.contentVs|      &&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__ks !rc.|jjSr rrs rrz_Response.statusnr rc.|jjSr rrs rrz_Response.headersrr"rc.|jjSr r$rs rr&z_Response.datavr"rN) r0r1r2r3rr4rrr&r5rrr7r7csH"%%&&&&rr7c4eZdZdZddZdddefdZdZdZy) 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__ Ncl|$t|dddurtjd||_d|_y)Nauto_decompressTz*>?/6.B.B=!.(C@#*>;La#P ' N-==nM /{+/48Z A>~w=292G2G!%nhD$+NNE$R3N;/ !//>tN?S>TU   .>E .2=.AN? +& 1 !Iz$,, 48C-0z* 1w44F~FGGrcKt|dds.|jr"|jjd{d|_y7 w)z?Cleanly release the underlying aiohttp ClientSession resources.rCFNT)r@rBcloserCrs rrz Request.close"s:tY.4<<,,$$& & &  's7AA Ar ) r0r1r2r3r_DEFAULT_TIMEOUTrYrrr5rrr=r={s-,  6*pNH`rr=cleZdZdZej ej dddffd Zdddedffd Z xZ S)AuthorizedSessionaNThis is an async implementation of the Authorized Session class. We utilize an aiohttp transport instance, and the interface mirrors the google.auth.transport.requests Authorized Session class, except for the change in the transport used in the async use case. A Requests Session class with credentials. This class is used to perform requests to API endpoints that require authorization:: from google.auth.transport import aiohttp_requests async with aiohttp_requests.AuthorizedSession(credentials) as authed_session: response = await authed_session.request( 'GET', 'https://www.googleapis.com/storage/v1/b') The underlying :meth:`request` implementation handles adding the credentials' headers to the request and refreshing credentials as needed. Args: credentials (google.auth._credentials_async.Credentials): The credentials to add to the request. refresh_status_codes (Sequence[int]): Which HTTP status codes indicate that credentials should be refreshed and the request should be retried. max_refresh_attempts (int): The maximum number of times to attempt to refresh the credentials and retry the request. refresh_timeout (Optional[int]): The timeout value in seconds for credential refresh HTTP requests. auth_request (google.auth.transport.aiohttp_requests.Request): (Optional) An instance of :class:`~google.auth.transport.aiohttp_requests.Request` used when refreshing credentials. If not passed, an instance of :class:`~google.auth.transport.aiohttp_requests.Request` is created. kwargs: Additional arguments passed through to the underlying ClientSession :meth:`aiohttp.ClientSession` object. NFc tt| di|||_||_||_||_d|_||_d|_ tj|_ tj|_||_y)NFr5)superrr credentials_refresh_status_codes_max_refresh_attempts_refresh_timeout_is_mtls _auth_request_auth_request_sessionrQget_event_loop_loopLock _refresh_lock_auto_decompress) rrrefresh_status_codesmax_refresh_attemptsrefresh_timeout auth_requestr?rV __class__s rrzAuthorizedSession.__init__Psu /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 timeout error will be raised after such request completes. zutf-8r\Fr[N_credential_refresh_attemptr)rGrFz;Refreshing credentials due to a %s response. Attempt %s/%s.)r&rmax_allowed_timerGr)"keysr{bytesdecoderIrJrgetrr=rpopcopy functoolspartialr TimeoutGuardrQrRrbefore_requestrrrMremaining_timeoutrrrrLinforrrun_in_executorrefresh)rrTrSr&rrrGr?rVkeyrrrequest_headersremaining_timeguardrrs rrMzAuthorizedSession.requestfsT ||~ @ %.#*3<#6#6w#?GCL @(( 11jje4 O O  '"4#=#=>L!-D +1**5RTU*V '180CgllnO ?""&&t'9'97K  .N&&~w7K7KL PU&&55 &#  &&~w7K7KL PU!&'8$!G"+# "  " #44N4#=#==/$2L2LL  %%QOO/!3.. &&"**4+=+=wO **"G$8$8#11"jj88 $"2"2":":L"'!8!8)T\\ #%3#0Ka0O   MO O O O O bcO 2     D MO O O O O bs3*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<) r0r1r2r3rDEFAULT_REFRESH_STATUS_CODESDEFAULT_MAX_REFRESH_ATTEMPTSrrrM __classcell__)rs@rrr)sH$R'CC&CC04 @@rr)r3 __future__rrQrloggingrIr+ google.authrrrgoogle.auth.aiorNgoogle.auth.transportr getLoggerr0rLrResponser r7r=rJrr5rrrs ' "!6* '  H % 3! **3!l& ""&0kik\}--}r