gj]8ddlZddlmZddlZddlZddlmZmZmZm Z ddl m Z m Z ddl mZddlmZddlmZddlmZddlZer ddlZdd lmZn ddlZdd lmZn#eef$rdZYnwxYw dd lmZd Zn #e$rd ZYnwxYwed ZGddZ dS)N)asynccontextmanager)MappingOptional TYPE_CHECKINGUnion)_exponential_backoff exceptions) transport) Credentials)mtls) TimeoutError) ClientTimeout)RequestTFcKtj|fdfd} |WVdS#wxYw)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()) cptjz }|z }|dkrtdd|S)Nrz3Context manager exceeded the configured timeout of s.)time monotonicr )elapsed remainingstart total_timeouts f/var/www/app.qaans.com/backend/venv/lib/python3.11/site-packages/google/auth/aio/transport/sessions.py_remaining_timez&timeout_guard.._remaining_timeBsP.""U*!G+ >>WmWWW cK }tj||d{V}|S#tjtf$r}td|dd|d}~wwxYw)NzThe operation z$ exceeded the configured timeout of r)asynciowait_forr )cororresponseerrs r with_timeoutz#timeout_guard..with_timeoutKs '))I$-dI>>>>>>>>HO$l3   \\\=\\\  s&,AAAN)rr)timeoutr"rrrs @@@r timeout_guardr$0s N  EM s 9 AceZdZdZ ddedeejfdZddZ ddej ej ej fde de d ee d eee e fd ed eeefd eedejfdZejeddej ej ej fde d ee d eee e fd ed eeefd eedejfdZejeddej ej ej fde d ee d eee e fd ed eeefd eedejfdZejeddej ej ej fde d ee d eee e fd ed eeefd eedejfdZejeddej ej ej fde d ee d eee e fd ed eeefd eedejfdZejeddej ej ej fde d ee d eee e fd ed eeefd eedejfdZedZddZdS)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 rt}d|_d|_ d|_ |tj d||_ dS)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+{33 (Td;6G6GTTT ($  -!2 -*,,M #   +I +rcKj'fd}tj|_jd{VS)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. NcKtjtjjjjd{V}|s d _dS tj d{V\ _}} jr| _ tjtj ||d{V}tr~t j trftj|}tj|} j }t| _ |d{VdSdSdSdS#t&jt*t,f$r}t'j|}||d}~wwxYw)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_configures(,(=K)6L))######'$)DMF2 #:;OPPPPPPPP  };,0),0,A =tS--'''''' - ; .22 ;)0(<(M(M(MI*1*?)*T*T*TK/3/A,1? 1T1T1TD."2"8"8":"::::::::::!;; ; ; ; ;.222 )>zJJG!z1 2sCD##E?EE)r1r create_task)r5rSrTs`` rconfigure_mtls_channelz-AsyncAuthorizedSession.configure_mtls_channelsh2   '( 2( 2( 2( 2( 2( 2T$+#6}}#G#GD ))))))))rmethodurldataheadersmax_allowed_timer#total_attemptsreturnc K|jr |jd{Vn#t$rYnwxYwtj|} |i}t |4d{V} | |j|j|||d{Vd} t&t|tr|j |j nd} n+t|ttfrt|} | 23d{V} | |j||||| fi|d{V} | j tjvrn>6 dddd{Vn#1d{VswxYwY| S)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 !122       l,!00&W        %(N(Z-O-O(29-2KQTGc5\22 0!&w#       a!-&D&VT7NFL"" 'y/WWWEX #7                           ,s+  &&BD7%D#+9D77 EEc :K|jd||||||fi|d{VS) 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.aT"T\                   rc :K|jd||||||fi|d{VS)rmPOSTNrorps rpostzAsyncAuthorizedSession.postcsaT"T\                   rc :K|jd||||||fi|d{VS)rmPUTNrorps rputzAsyncAuthorizedSession.putrrrc :K|jd||||||fi|d{VS)rmPATCHNrorps rpatchzAsyncAuthorizedSession.patchsaT"T\                   rc :K|jd||||||fi|d{VS)rmDELETENrorps rdeletezAsyncAuthorizedSession.deletesaT"T\                   rc|jS)z#Indicates if mutual TLS is enabled.)r0r5s ris_mtlszAsyncAuthorizedSession.is_mtls7s }rcHK|jd{VdS)z< Close the underlying auth request session. N)r4rFrs rrFzAsyncAuthorizedSession.close<s5 &&(((((((((((r)N)r]N)__name__ __module__ __qualname____doc__r rr rr6rV_DEFAULT_TIMEOUT_SECONDSDEFAULT_MAX_RETRY_ATTEMPTSstrbytesrrdrrrcResponserk functoolswrapsrqrurxr{r~propertyrrFrrr&r&\s##LUY++&+6>y?P6Q++++(G*G*G*G*Z!%/3"+"D/8/Q(1(LMMMMuo M '#s(+, M  Mum+,M! M  MMMM^Y_W!%/3"+"D/8/Q(1(L2 2 2 uo2 '#s(+, 2  2 um+, 2 ! 2   2 2 2 2 hY_W!%/3"+"D/8/Q(1(L2 2 2 uo2 '#s(+, 2  2 um+, 2 ! 2   2 2 2 2 hY_W!%/3"+"D/8/Q(1(L2 2 2 uo2 '#s(+, 2  2 um+, 2 ! 2   2 2 2 2 hY_W!%/3"+"D/8/Q(1(L2 2 2 uo2 '#s(+, 2  2 um+, 2 ! 2   2 2 2 2 hY_W!%/3"+"D/8/Q(1(L2 2 2 uo2 '#s(+, 2  2 um+, 2 ! 2   2 2 2 2 hX))))))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****** ::::::::::::88888888%%%%%%333333******//////)))) NNN%%%%%%%)))))))  ( KKKKKK(((Vd)d)d)d)d)d)d)d)d)d)s$ A A('A(,A55A?>A?