o Bvg(@sddlZddlmZddlZddlZddlmZmZddlm Z m Z ddl m Z ddl mZddlmZz ddlmZd ZWn eyId ZYnwed d ZGd ddZdS)N)asynccontextmanager)MappingOptional)_exponential_backoff exceptions) transport) Credentials) TimeoutError)RequestTFcsFt|fddfdd}z |VWdSw)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()) cs0t}|}|dkrtdd|S)Nrz3Context manager exceeded the configured timeout of s.)time monotonicr )elapsed remaining)start total_timeout_/var/www/html/portfolio/venv/lib/python3.10/site-packages/google/auth/aio/transport/sessions.py_remaining_time4s  z&timeout_guard.._remaining_timec sXz}t||IdH}|WStjtfy+}z td|dd|d}~ww)NzThe operation z$ exceeded the configured timeout of r )asynciowait_forr )cororresponsee)rrrr with_timeout=sz#timeout_guard..with_timeoutN)r r )timeoutrr)rrrr timeout_guard"s  rc@seZdZdZ ddedeejfddZddej ej fde de d ee d ee e e fd e d e d ejfddZeeddej ej fde d ee d ee e e fd e d e d ejf ddZeeddej ej fde d ee d ee e e fd e d e d ejf ddZeeddej ej fde d ee d ee e e fd e d e d ejf ddZeeddej ej fde d ee d ee e e fd e d e d ejf ddZeeddej ej fde d ee d ee e e fd e d e d ejf ddZdddZdS)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_requestcCsTt|tstdt|d||_|}|strt}|dur%td||_ dS)Nz#The configured credentials of type zJ are invalid and must be of type `google.auth.aio.credentials.Credentials`zv`auth_request` must either be configured or the external package `aiohttp` must be installed to use the default value.) isinstancerr InvalidTypetype _credentialsAIOHTTP_INSTALLEDAiohttpRequestTransportError _auth_request)selfrrr'rrr__init__ts  zAsyncAuthorizedSession.__init__methodurldataheadersmax_allowed_timerreturnc stjtjd}t|4IdH?} | |j|j|||IdH|2z 3dHW} | |j|||||fi|IdH} | jtj vrBnq"6WdIdH| S1IdHsUwY| S)a 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): 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. 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. 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`. )total_attemptsN) rAsyncExponentialBackoffrDEFAULT_MAX_RETRY_ATTEMPTSrr#before_requestr' status_codeDEFAULT_RETRYABLE_STATUS_CODES) r(r*r+r,r-r.rkwargsretriesr_rrrrrequests,)   zAsyncAuthorizedSession.requestc$|jd|||||fi|IdHS)NGETr9r(r+r,r-r.rr6rrrget   zAsyncAuthorizedSession.getcr:)NPOSTr<r=rrrpostr?zAsyncAuthorizedSession.postcr:)NPUTr<r=rrrputr?zAsyncAuthorizedSession.putcr:)NPATCHr<r=rrrpatchr?zAsyncAuthorizedSession.patchcr:)NDELETEr<r=rrrdeleter?zAsyncAuthorizedSession.deletecs|jIdHdS)z< Close the underlying auth request session. N)r'close)r(rrrrHszAsyncAuthorizedSession.close)N)r/N)__name__ __module__ __qualname____doc__rrrr r)_DEFAULT_TIMEOUT_SECONDSstrbytesrfloatResponser9 functoolswrapsr>rArCrErGrHrrrrrNs&  =          r)r contextlibrrRr typingrr google.authrrgoogle.auth.aiorgoogle.auth.aio.credentialsrgoogle.auth.exceptionsr !google.auth.aio.transport.aiohttpr r%r$ ImportErrorrrrrrrs$       +