p[i]&ddlZddlmZddlZddlZddlmZmZmZm Z ddl m Z m Z ddl mZddlmZddlmZddlmZddlZer ddlZdd lmZn ddlZdd lmZ dd lmZd Zed ZGddZ y#eef$rdZY'wxYw#e$rd ZY+wxYw)N)asynccontextmanager)MappingOptional TYPE_CHECKINGUnion)_exponential_backoff exceptions) transport) Credentials)mtls) TimeoutError) ClientTimeout)RequestTFcKtj|fdfd} |y#wxYww)a timeout_guard is an asynchronous context manager to apply a timeout to an asynchronous block of code. Args: timeout (float): The time in seconds before the context manager times out. Raises: google.auth.exceptions.TimeoutError: If the code within the context exceeds the provided timeout. Usage: async with timeout_guard(10) as with_timeout: await with_timeout(async_function()) chtjz }|z }|dkrtdd|S)Nrz3Context manager exceeded the configured timeout of s.)time monotonicr )elapsed remainingstart total_timeouts [/mnt/data/GridWatch/venv/lib/python3.12/site-packages/google/auth/aio/transport/sessions.py_remaining_timez&timeout_guard.._remaining_timeBsG.."U*!G+ >Em_TVW cK }tj||d{}|S7#tjtf$r}td|dd|d}~wwxYww)NzThe operation z$ exceeded the configured timeout of r)asynciowait_forr )cororresponseerrs r with_timeoutz#timeout_guard..with_timeoutKsq ')I$--dI>>HO?$$l3  &J=/Y[\  s1A" .,.A".AAAA"N)rr)timeoutr"rrrs @@@r timeout_guardr$0s= NN EM s#A6A ?AceZdZdZ ddedeejfdZddZ ddejejejfde de d ee d eee e fd ed eeefd eedej&fdZej,eddejejejfde d ee d eee e fd ed eeefd eedej&fdZej,eddejejejfde d ee d eee e fd ed eeefd eedej&fdZej,eddejejejfde d ee d eee e fd ed eeefd eedej&fdZej,eddejejejfde d ee d eee e fd ed eeefd eedej&fdZej,eddejejejfde d ee d eee e fd ed eeefd eedej&fdZedZddZy)AsyncAuthorizedSessionaThis is an asynchronous implementation of :class:`google.auth.requests.AuthorizedSession` class. We utilize an instance of a class that implements :class:`google.auth.aio.transport.Request` configured by the caller or otherwise default to `google.auth.aio.transport.aiohttp.Request` if the external aiohttp package is installed. A Requests Session class with credentials. This class is used to perform asynchronous requests to API endpoints that require authorization:: import aiohttp from google.auth.aio.transport import sessions async with sessions.AsyncAuthorizedSession(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.aio.credentials.Credentials): The credentials to add to the request. auth_request (Optional[google.auth.aio.transport.Request]): An instance of a class that implements :class:`~google.auth.aio.transport.Request` used to make requests and refresh credentials. If not passed, an instance of :class:`~google.auth.aio.transport.aiohttp.Request` is created. Raises: - google.auth.exceptions.TransportError: If `auth_request` is `None` and the external package `aiohttp` is not installed. - google.auth.exceptions.InvalidType: If the provided credentials are not of type `google.auth.aio.credentials.Credentials`. N credentials auth_requestct|ts"tjdt |d||_|}|st r t}d|_d|_ d|_ |tjd||_ y)Nz#The configured credentials of type zJ are invalid and must be of type `google.auth.aio.credentials.Credentials`Fzv`auth_request` must either be configured or the external package `aiohttp` must be installed to use the default value.) isinstancer r InvalidTypetype _credentialsAIOHTTP_INSTALLEDAiohttpRequest_is_mtls_mtls_init_task _cached_certTransportError _auth_request)selfr'r(r4s r__init__zAsyncAuthorizedSession.__init__s+{3((5d;6G5HIST ($ !2*,M #   ++I +rcKj%fd}tj|_jd{S7w)aConfigure the client certificate and key for SSL connection. The function does nothing unless `GOOGLE_API_USE_CLIENT_CERTIFICATE` is explicitly set to `true`. In this case if client certificate and key are successfully obtained (from the given client_cert_callback or from application default SSL credentials), the underlying transport will be reconfigured to use mTLS. Note: This function does nothing if the `aiohttp` library is not installed. Important: Calling this method will close any ongoing API requests associated with the current session. To ensure a smooth transition, it is recommended to call this during session initialization. Args: client_cert_callback (Optional[Callable[[], (bytes, bytes)]]): The optional callback returns the client certificate and private key bytes both in PEM format. If the callback is None, application default SSL credentials will be used. Raises: google.auth.exceptions.MutualTLSChannelError: If mutual TLS channel creation failed for any reason. NcKtjtjjj j d{}|sd _y tj d{\ _}} jr| _ tjtj||d{}tr}t jtrbtj |}tj"|} j}t| _ |j%d{yyyy7777#t&j(t*t,f$r}t'j.|}||d}~wwxYww)NF)ssl) connector)session)r _run_in_executorgoogleauthr _mtls_helpercheck_use_client_certr0get_client_cert_and_keyr2make_client_cert_ssl_contextr.r*r4r/aiohttp TCPConnector ClientSessioncloser ClientCertError ImportErrorOSErrorMutualTLSChannelError) use_client_certcertkey ssl_contextr: new_sessionold_auth_request caught_excnew_excclient_cert_callbackr5s r _do_configurezDAsyncAuthorizedSession.configure_mtls_channel.._do_configurese(,(=(=KK))66LL)#'$)DM2 #::;OPP  }},0),0,A,A ==tS-' - ..2)0(<(<(MI*1*?*?)*TK/3/A/A,1? 1TD."2"8"8":::2,%#Q';..2 )>>zJG!z1 2snAF E F E 0E1AE 9E:BE :E ;E ?F E E E F)FFF )r1r create_task)r5rSrTs`` rconfigure_mtls_channelz-AsyncAuthorizedSession.configure_mtls_channelsD2    '( 2T$+#6#6}#GD )))))sAA A A methodurldataheadersmax_allowed_timer#total_attemptsreturnc K|jr |jd{tj|} |i}t |4d{} | |j j |j|||d{d} t+t|tr|j |jnd} n!t|ttfr t|} | 23d{} | |j||||| fi|d{} | jtjvsKdddd{ S7#t$rY'wxYw777q7N607$#1d{7swY SxYww)aY Args: method (str): The http method used to make the request. url (str): The URI to be requested. data (Optional[bytes]): The payload or body in HTTP request. headers (Optional[Mapping[str, str]]): Request headers. timeout (float, aiohttp.ClientTimeout): The amount of time in seconds to wait for the server response with each individual request. max_allowed_time (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. total_attempts (int): The total number of retry attempts. 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. Returns: google.auth.aio.transport.Response: The HTTP response. Raises: google.auth.exceptions.TimeoutError: If the method does not complete within the configured `max_allowed_time` or the request exceeds the configured `timeout`. N)r\g)r1 ExceptionrAsyncExponentialBackoffr$r-before_requestr4rr*totalintfloat status_coder DEFAULT_RETRYABLE_STATUS_CODES)r5rWrXrYrZr[r#r\kwargsretriesr"actual_timeout_r s rrequestzAsyncAuthorizedSession.requestsV    **** '>>)  ?G !12  l!!00&&W   %(N(Z-O29--2KQTGc5\2!&w#  a!-&D&&VT7NFL" ''y/W/WW+  ,A+     #    ,s E2D?D<D?)E2 EE22EEAEE"E#E&!EE E)E* E25E6E2<D?? E E2 E  E2EEEEE2E/"E% #E/*E2c PK|jd||||||fi|d{S7w) Args: url (str): The URI to be requested. data (Optional[bytes]): The payload or body in HTTP request. headers (Optional[Mapping[str, str]]): Request headers. max_allowed_time (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. timeout (float, aiohttp.ClientTimeout): The amount of time in seconds to wait for the server response with each individual request. total_attempts (int): The total number of retry attempts. 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. Returns: google.auth.aio.transport.Response: The HTTP response. Raises: google.auth.exceptions.TimeoutError: If the method does not complete within the configured `max_allowed_time` or the request exceeds the configured `timeout`. GETNrkr5rXrYrZr[r#r\rgs rgetzAsyncAuthorizedSession.get.ET"T\\             &$&c PK|jd||||||fi|d{S7w)rmPOSTNrorps rpostzAsyncAuthorizedSession.postcsET"T\\             rsc PK|jd||||||fi|d{S7w)rmPUTNrorps rputzAsyncAuthorizedSession.putrrrsc PK|jd||||||fi|d{S7w)rmPATCHNrorps rpatchzAsyncAuthorizedSession.patchsET"T\\             rsc PK|jd||||||fi|d{S7w)rmDELETENrorps rdeletezAsyncAuthorizedSession.deletesET"T\\             rsc|jS)z#Indicates if mutual TLS is enabled.)r0r5s ris_mtlszAsyncAuthorizedSession.is_mtls7s}}rcTK|jjd{y7w)z< Close the underlying auth request session. N)r4rFrs rrFzAsyncAuthorizedSession.close<s   &&(((s (&()N)r]N)__name__ __module__ __qualname____doc__r rr rr6rV_DEFAULT_TIMEOUT_SECONDSDEFAULT_MAX_RETRY_ATTEMPTSstrbytesrrdrrrcResponserk functoolswrapsrqrvryr|rpropertyrrFrrr&r&\sV#LUY+&+6>y?P?P6Q+(G*Z!%/3"+"D"D/8/Q/Q(1(L(LMMMuo M '#s(+, M  Mum+,M! M   M^Y__W!%/3"+"D"D/8/Q/Q(1(L(L2 2 uo2 '#s(+, 2  2 um+, 2 ! 2    2 2 hY__W!%/3"+"D"D/8/Q/Q(1(L(L2 2 uo2 '#s(+, 2  2 um+, 2 ! 2    2 2 hY__W!%/3"+"D"D/8/Q/Q(1(L(L2 2 uo2 '#s(+, 2  2 um+, 2 ! 2    2 2 hY__W!%/3"+"D"D/8/Q/Q(1(L(L2 2 uo2 '#s(+, 2  2 um+, 2 ! 2    2 2 hY__W!%/3"+"D"D/8/Q/Q(1(L(L2 2 uo2 '#s(+, 2  2 um+, 2 ! 2    2 2 h)rr&)!r contextlibrrrtypingrrrr google.authrr google.auth.aior google.auth.aio.credentialsr google.auth.aio.transportr google.auth.exceptionsr "google.auth.transport._mtls_helperr=rCrrHAttributeError!google.auth.aio.transport.aiohttprr/r.r$r&rrrrs* ::8%3*/)%) K ((Vd)d)q  ( s$ A7B7 BBBB