U /j @s dZddlZddlZddlmZmZmZmZmZz ddl Z Wn,e k rhZ ze de W5dZ [ XYnXddl m Z ddl mZddlm ZddlmZerddl mZn.zddl mZWne efk rdZYnXeeZGd 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@seZdZdZejdddZee e j e dddZ ee e j eeefddd Ze e j de eed fd d dZe e j edddZe 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 rE/tmp/pip-unpacked-wheel-l8yytoim/google/auth/aio/transport/aiohttp.py__init__9szResponse.__init__returncCs|jjSr )rstatusrrrr status_code<szResponse.status_codecCsdd|jjDS)NcSsi|]\}}||qSrr).0keyvaluerrr Dsz$Response.headers..)rheadersitemsrrrrrAszResponse.headersN) chunk_sizerc Cs^z(|jj|2z3dHW}|Vq6Wn0tjk rX}ztd|W5d}~XYnXdS)Nz'Failed to read from the payload stream.)rcontentZ iter_chunkedaiohttpZClientPayloadErrorr ResponseError)rrchunkexcrrrr FszResponse.contentc sHz|jIdHWStjk rB}ztd|W5d}~XYnXdS)Nz!Failed to read the response body.)rreadr!ZClientResponseErrorrr")rr$rrrr%Rsz Response.readcs|jdSr )rcloserrrrr&YszResponse.close)r)__name__ __module__ __qualname____doc__r!ZClientResponserpropertyrZcopy_docstringr r intrrstrrrbytesr r%r&rrrrr -s      r c@sreZdZdZd eejdddZdddej fe e ee ee e e fe eefejddd Zdd d d ZdS)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__ N)sessioncCs||_d|_dS)NF)_session_closed)rr0rrrr|szRequest.__init__GET)urlmethodbodyrtimeoutrc s&z|jrtd|js"t|_t|tjr4|}n tj|d}t t |||||jj ||f|||d|IdH}t t |IdHt|WStjk r} ztd|d} | | W5d} ~ XYnVtjk r } z4t|tjr|j} n|} td| d} | | W5d} ~ XYnXdS) 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)datarr7NzFailed to send request to .zRequest timed out after z seconds.)r2rZTransportErrorr1r! ClientSession isinstancer rZ request_log_LOGGERrequest_helpers_asyncZresponse_log_asyncr Z ClientErrorasyncio TimeoutErrorr8) rr4r5r6rr7kwargsZclient_timeoutr caught_excZ client_excZtimeout_secondsZ timeout_excrrr__call__s@"        zRequest.__call__rcs&|js|jr|jIdHd|_dS)zY Close the underlying aiohttp session to release the acquired resources. NT)r2r1r&rrrrr&s z Request.close)N)r'r(r)r*rr!r;rr Z_DEFAULT_TIMEOUT_SECONDSr-r.rrfloatr r rDr&rrrrr/^s  Gr/)r*r@loggingtypingrrrrrr! ImportErrorrCZ google.authrrZgoogle.auth.aior?r r AttributeError getLoggerr'r=r r/rrrrs0       1