o tBh4@sddlZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZmZddlZddlmZddlmZddlmZddlmZmZddlmZmZmZdd lmZd d l m!Z!d d l"m#Z#d dl$m%Z%d dl&m'Z'm(Z(m)Z)m*Z*e+e,Z-Gddde%Z.dS)N) SSLContext)AnyAsyncGeneratorCallableDictOptionalTupleTypeUnion)ClientResponseError) Fingerprint) BasicAuth) LooseCookies LooseHeaders) DocumentNodeExecutionResult print_ast)CIMultiDictProxy) extract_files)AppSyncAuthentication)AsyncTransport)TransportAlreadyConnectedTransportClosedTransportProtocolErrorTransportServerErrorc@s\eZdZUdZejejefZ e e e dfe d<ddddddejdfdedeed eed eeed fd eeeefd eedeeeefdedeeee fddfddZd#ddZede j!fddZ"d#ddZ#    d$de$deeee fdeedeeee fdede%f dd Z&  d%de$deeee fdeedee%dffd!d"Z'dS)&AIOHTTPTransportz:ref:`Async Transport ` to execute GraphQL queries on remote servers with an HTTP connection. This transport use the aiohttp library with asyncio. . file_classesNF urlheaderscookiesauthrssltimeoutssl_close_timeoutjson_serializeclient_session_argsreturnc CsD||_||_||_||_||_||_||_| |_d|_|||_ dS)aInitialize the transport with the given aiohttp parameters. :param url: The GraphQL server URL. Example: 'https://server.com:PORT/path'. :param headers: Dict of HTTP Headers. :param cookies: Dict of HTTP cookies. :param auth: BasicAuth object to enable Basic HTTP auth if needed Or Appsync Authentication class :param ssl: ssl_context of the connection. Use ssl=False to disable encryption :param ssl_close_timeout: Timeout in seconds to wait for the ssl connection to close properly :param json_serialize: Json serializer callable. By default json.dumps() function :param client_session_args: Dict of extra args passed to `aiohttp.ClientSession`_ .. _aiohttp.ClientSession: https://docs.aiohttp.org/en/stable/client_reference.html#aiohttp.ClientSession N) r r!r"r#r$r%r&r(sessionr') selfr r!r"r#r$r%r&r'r(r,l/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/gql/transport/aiohttp.py__init__+s zAIOHTTPTransport.__init__cs|jdur@|j|jt|jtrdn|j|jd}|jdur'tj |jd|d<|j r0| |j t dtjdi||_dStd)aECoroutine which will create an aiohttp ClientSession() as self.session. Don't call this coroutine directly on the transport, instead use :code:`async with` on the client and this coroutine will be executed to create the session. Should be cleaned with a call to the close coroutine. N)r"r!r#r')totalr%zConnecting transportzTransport is already connectedr,)r*r"r! isinstancer#rr'r%aiohttp ClientTimeoutr(updatelogdebug ClientSessionr)r+r(r,r,r-connectUs$      zAIOHTTPTransport.connectc sdtfdd}dd}|jjD]/}|D]*\}}t|jdd}|dur+qd7|j}|j}t j ||d |_t j ||d |_qqdkrO S) a Work around aiohttp issue that doesn't properly close transports on exit. See https://github.com/aio-libs/aiohttp/issues/1925#issuecomment-639080209 Returns: An event that will be set once all transports have been properly closed. rcsHz||Wd8dkrdSdSd8dkr#ww)Nrr)set)exc orig_lost all_is_lostssl_transportsr,r-connection_losts   zEAIOHTTPTransport.create_aiohttp_closed_event..connection_lostcSs"z|WdStyYdSw)N)AttributeErrororig_eof_receivedr,r,r- eof_receiveds   zBAIOHTTPTransport.create_aiohttp_closed_event..eof_received _ssl_protocolNr)r:r@) asyncioEvent connector_connsvaluesgetattr transportr>rB functoolspartialr8) r*r>rBconnhandler_protor:rAr,r;r-create_aiohttp_closed_eventzs,    z,AIOHTTPTransport.create_aiohttp_closed_eventcs|jdurCtd|jr|jddurtdn'||j}|jIdHzt| |j IdHWn tj yBYnwd|_dS)aCoroutine which will close the aiohttp session. Don't call this coroutine directly on the transport, instead use :code:`async with` on the client and this coroutine will be executed when you exit the async context manager. NzClosing transportconnector_ownerFz1connector_owner is False -> not closing connector) r*r4r5r(getrQcloserDwait_forwaitr& TimeoutError)r+ closed_eventr,r,r-rTs     zAIOHTTPTransport.closedocumentvariable_valuesoperation_name extra_args upload_filesc st|}d|i}|r||d<|r|dusJt||jd\}||d<t} ddtD} fddtD} ||} td | | j d | d d || } td | | j d| d d | D]\}}t |d|}t |dd}| j ||||dqfd| i}n|r||d<t t jrtd||d|i}|r||t|jtr|j||dd i|d<|jdurtd|jj|jfd|ji|4IdHz}|j|_dtjdtfdd}z|jdd IdH}t t jr|IdH}td|Wnt y||dIdHYnw|dur"||dIdHd|vr4d|vr4||d IdHt!|"d|"d|"d!d"WdIdHS1IdHsUwYdS)#aExecute the provided document AST against the configured remote server using the current session. This uses the aiohttp library to perform a HTTP POST request asynchronously to the remote server. Don't call this coroutine directly on the transport, instead use :code:`execute` on a client or a session. :param document: the parsed GraphQL request :param variable_values: An optional Dict of variable values :param operation_name: An optional Operation name for the request :param extra_args: additional arguments to send to the aiohttp post method :param upload_files: Set to True if you want to put files in the variable values :returns: an ExecutionResult object. query operationNameN) variablesrr`cSsi|] \}}t||gqSr,str.0ipathr,r,r- sz,AIOHTTPTransport.execute..csi|] \}}t||qSr,rarcfilesr,r-rgsz operations %s operationszapplication/json) content_typez file_map %smapnamerk)filenamerkdataz>>> %sjsonz content-typer!zTransport is not connectedr$respreasonc s\z|Wnty}z tt||j|d}~ww|IdH}td|d|)Nz(Server did not return a GraphQL result: z: )raise_for_statusr rrbstatustextr)rqrre result_textr,r,r-raise_response_error9s z6AIOHTTPTransport.execute..raise_response_errorz<<< %szNot a JSON answererrorsz$No "data" or "errors" keys in answer extensions)ryrorz)#rrrr1FormData enumerater'r4r5 add_fielditemsrI isEnabledForloggingINFOinfor3r0r#r get_headersr*rpostr r$r!response_headersClientResponserbrpru ExceptionrrS)r+rYrZr[r\r] query_strpayloadnulled_variable_valuesrofile_map file_streamsoperations_str file_map_strkfrmrk post_argsrqrxresultrwr,rhr-executes             &  2zAIOHTTPTransport.executecCstd)zDSubscribe is not supported on HTTP. :meta private: z2 The HTTP transport does not support subscriptions)NotImplementedError)r+rYrZr[r,r,r- subscribe`s zAIOHTTPTransport.subscribe)r)N)NNNF)NN)(__name__ __module__ __qualname____doc__ioIOBaser1 StreamReaderrrrr r__annotations__rpdumpsrbrrrr r rboolr intfloatrrr.r7 staticmethodrDrErQrTrrrrr,r,r,r-rs      *% 4  r)/rDrKrrprr$rtypingrrrrrrr r r1aiohttp.client_exceptionsr aiohttp.client_reqrepr aiohttp.helpersr aiohttp.typedefsrrgraphqlrrr multidictrutilsr appsync_authrasync_transportr exceptionsrrrr getLoggerrr4rr,r,r,r-s( (