o bi @sdZddlZddlZddlmZmZmZmZmZzddl Z Wne y/Z ze de dZ [ wwddl m Z ddl mZddlm ZddlmZerQddl mZnzddl mZWn e efyfdZYnweeZGd d d ejZGd d d ejZdS) zBTransport adapter for Asynchronous HTTP Requests based on aiohttp.N)AsyncGeneratorMappingOptional TYPE_CHECKINGUnionzjThe aiohttp library is not installed from please install the aiohttp package to use the aiohttp transport.)_helpers) exceptions) transport) ClientTimeoutc@seZdZdZdejfddZee e j de fddZ ee e j deeeffdd Ze e j dd e deed ffd dZe e j defddZe e j ddZd S)Responseaz Represents an HTTP response and its data. It is returned by ``google.auth.aio.transport.sessions.AsyncAuthorizedSession``. Args: response (aiohttp.ClientResponse): An instance of aiohttp.ClientResponse. Attributes: status_code (int): The HTTP status code of the response. headers (Mapping[str, str]): The HTTP headers of the response. responsecCs ||_dSN) _response)selfr r[/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/auth/aio/transport/aiohttp.py__init__9s zResponse.__init__returncCs|jjSr )rstatusrrrr status_code<szResponse.status_codecCsdd|jjDS)NcSsi|]\}}||qSrr).0keyvaluerrr Dsz$Response.headers..)rheadersitemsrrrrrAszResponse.headers chunk_sizeNc CsTz|jj|2z 3dHW}|Vq 6WdStjy)}ztd|d}~ww)Nz'Failed to read from the payload stream.)rcontentZ iter_chunkedaiohttpZClientPayloadErrorr ResponseError)rrchunkexcrrrrFszResponse.contentc s<z |jIdHWStjy}ztd|d}~ww)Nz!Failed to read the response body.)rreadr ZClientResponseErrorrr!)rr#rrrr$Rs z Response.readcs|jdSr )rcloserrrrr%YszResponse.close)r)__name__ __module__ __qualname____doc__r ZClientResponserpropertyrZcopy_docstringr r intrrstrrrbytesrr$r%rrrrr -s      r c @sxeZdZdZddeejfddZdddej fde de d ee d ee e e fd e eefd ejf d dZdddZdS)RequestaAsynchronous Requests request adapter. This class is used internally for making requests using aiohttp in a consistent way. If you use :class:`google.auth.aio.transport.sessions.AsyncAuthorizedSession` you do not need to construct or use this class directly. This class can be useful if you want to configure a Request callable with a custom ``aiohttp.ClientSession`` in :class:`AuthorizedSession` or if you want to manually refresh a :class:`~google.auth.aio.credentials.Credentials` instance:: import aiohttp import google.auth.aio.transport.aiohttp # Default example: request = google.auth.aio.transport.aiohttp.Request() await credentials.refresh(request) # Custom aiohttp Session Example: session = session=aiohttp.ClientSession(auto_decompress=False) request = google.auth.aio.transport.aiohttp.Request(session=session) auth_session = google.auth.aio.transport.sessions.AsyncAuthorizedSession(auth_request=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__ NsessioncCs||_d|_dS)NF)_session_closed)rr/rrrr|s zRequest.__init__GETurlmethodbodyrtimeoutrc szH|jr td|jst|_t|tjr|}ntj|d}t t |||||jj ||f|||d|IdH}t t |IdHt|WStjyb} z td|d} | | d} ~ wtjy} zt|tjrt|j} n|} td| d} | | d} ~ ww) a Make an HTTP request using aiohttp. Args: url (str): The URL to be requested. method (Optional[str]): The HTTP method to use for the request. Defaults to 'GET'. body (Optional[bytes]): The payload or body in HTTP request. headers (Optional[Mapping[str, str]]): Request headers. timeout (float): The number of seconds to wait for a response from the server. If not specified or if None, the requests default timeout will be used. kwargs: Additional arguments passed through to the underlying aiohttp :meth:`aiohttp.Session.request` method. Returns: google.auth.aio.transport.Response: The HTTP response. Raises: - google.auth.exceptions.TransportError: If the request fails or if the session is closed. - google.auth.exceptions.TimeoutError: If the request times out. zsession is closed.)total)datarr6NzFailed to send request to .zRequest timed out after z seconds.)r1rZTransportErrorr0r ClientSession isinstancer rZ request_log_LOGGERrequest_helpers_asyncZresponse_log_asyncr Z ClientErrorasyncio TimeoutErrorr7) rr3r4r5rr6kwargsZclient_timeoutr caught_excZ client_excZtimeout_secondsZ timeout_excrrr__call__sH"        zRequest.__call__cs(|js|jr|jIdHd|_dS)zY Close the underlying aiohttp session to release the acquired resources. NT)r1r0r%rrrrr%s  z Request.closer )rN)r&r'r(r)rr r:rr Z_DEFAULT_TIMEOUT_SECONDSr,r-rrfloatr r rCr%rrrrr.^s*  Gr.)r)r?loggingtypingrrrrrr ImportErrorrBZ google.authrrZgoogle.auth.aior>r r AttributeError getLoggerr&r<r r.rrrrs8      1