@wja.ddlZddlmZddlZddlZddlmZmZmZm Z ddl Z ddl m Z m Z ddlmZddlmZddlmZddlmZddlZer ddlZdd lmZn ddlZdd lmZ dd lmZd Zed Z GddZ!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 `/root/dashboard-youtube/.venv/lib/python3.12/site-packages/google/auth/aio/transport/sessions.py_remaining_timez&timeout_guard.._remaining_timeDsG.."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_timeoutMsq ')I$--dI>>HO?$$l3  &J=/Y[\  s1A" .,.A".AAAA"N)rr)timeoutr"rrrs @@@r timeout_guardr$2s= 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. This method configures mTLS if client certificates are explicitly enabled (via GOOGLE_API_USE_CLIENT_CERTIFICATE=true) or auto-enabled (when the env variable is unset and workload certificates are discovered). In these cases, 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. Nc8Ktjtjjj j d{}|sy tj d{\}}}|rtrt jtrtjtj||d{}tj|}tj|} j}t| _ |j!d{nd}t%j&dt(| _|r| _yd _y7777A#t"$rY-wxYw#t"$r}t/j0|} | |d}~wwxYww)N)ssl) connector)sessionFa9Attempted to establish mTLS, but a custom async transport was provided. google-auth cannot automatically configure custom transports for mTLS. Falling back to standard TLS. If your custom transport is not manually configured for mTLS, you may encounter 401 Unauthorized errors when using Certificate-Bound Tokens.)r _run_in_executorgoogleauthr _mtls_helpercheck_use_client_certget_client_cert_and_keyr.r*r4r/make_client_cert_ssl_contextaiohttp TCPConnector ClientSessionclose Exceptionwarningswarn UserWarningr0r2r MutualTLSChannelError) use_client_certis_mtlscertkey ssl_contextr: new_sessionold_auth_request caught_excnew_excclient_cert_callbackr5s r _do_configurezDAsyncAuthorizedSession.configure_mtls_channel.._do_configures(,(=(=KK))66LL)#'-2 #::;OPP , ..2150E0E $ A A41+K)0(<(<(MI*1*?*?)*TK/3/A/A,1? 1TD.%&6&<&<&> > > ',G$MM!B !, %,DM,0),0)_#Q+!?#,% $%(!2(>>zJG!z12sAFE FE1)E*AE1<E=A E1 E"E E"#-E1FE1FE1E1 E"" E.+E1-E..E11 F:FFF)r1r create_task)r5rUrVs`` rconfigure_mtls_channelz-AsyncAuthorizedSession.configure_mtls_channelsD2    '5 2n$+#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)r1rGrAsyncExponentialBackoffr$r-before_requestr4rr*totalintfloat status_coder DEFAULT_RETRYABLE_STATUS_CODES)r5rYrZr[r\r]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`. GETNrlr5rZr[r\r]r#r^rhs rgetzAsyncAuthorizedSession.get=ET"T\\             &$&c PK|jd||||||fi|d{S7w)rnPOSTNrprqs rpostzAsyncAuthorizedSession.postrsET"T\\             rtc PK|jd||||||fi|d{S7w)rnPUTNrprqs rputzAsyncAuthorizedSession.putrsrtc PK|jd||||||fi|d{S7w)rnPATCHNrprqs rpatchzAsyncAuthorizedSession.patchsET"T\\             rtc PK|jd||||||fi|d{S7w)rnDELETENrprqs rdeletezAsyncAuthorizedSession.deletesET"T\\             rtc|jS)z#Indicates if mutual TLS is enabled.)r0r5s rrMzAsyncAuthorizedSession.is_mtlsFs}}rc4K|jrI|jjs/|jj |jd{|j j d{y7'#tj$rY:wxYw7 w)z< Close the underlying auth request session. N)r1donecancelrCancelledErrorr4rFrs rrFzAsyncAuthorizedSession.closeKs   (<(<(A(A(C  ' ' ) ****  &&(((+))  (sHABA=A;A=B5B6B;A==BBBB)N)r_N)__name__ __module__ __qualname____doc__r rr rr6rX_DEFAULT_TIMEOUT_SECONDSDEFAULT_MAX_RETRY_ATTEMPTSstrbytesrrerrrdResponserl functoolswrapsrrrwrzr}rpropertyrMrFrrr&r&^sV#LUY+&+6>y?P?P6Q+(T*t!%/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 contextlibrrrtypingrrrrrH google.authrr google.auth.aior google.auth.aio.credentialsr google.auth.aio.transportr google.auth.exceptionsr "google.auth.transport._mtls_helperr=rCr ImportErrorAttributeError!google.auth.aio.transport.aiohttprr/r.r$r&rrrrs* ::8%3*/)%) K ((Vw)w)s  ( s$ A; B ; BB BB