o tBhJ@sddlZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZddlmZmZmZddlmZddlmZdd lmZmZmZdd lmZeeZGd d d eZ dS) N)suppress) SSLContext)AnyDictListOptionalTupleUnioncast) DocumentNodeExecutionResult print_ast) HeadersLike) Subprotocol)TransportProtocolErrorTransportQueryErrorTransportServerError)WebsocketsTransportBasecsxeZdZdZeedZeedZddidddddddidf ded e e d e e e fd eeefd e e eefd e e eefde e eefde e eefde e eefde e eefde deeefde eeddffdd ZdHddZdHddZddZdIde eddfdd ZdIde eddfd!d"Zd#eddfd$d%Zd#eddfd&d'Zd#efd(d)ZdHd*d+Z  dJd,ed-e eeefd.e edefd/d0Zd1d2Zd3eeefde ee ee e!ffd4d5Z"d3eeefde ee ee e!ffd6d7Z#d8ede ee ee e!ffd9d:Z$dHd;d<Z%d=ed>e ed?e e!ddffd@dA Z&dBdCZ'dDdEZ(dFdGZ)Z*S)KWebsocketsTransportz:ref:`Async Transport ` used to execute GraphQL queries on remote servers with websocket connection. This transport uses asyncio and the websockets library in order to send requests on a websocket connection. z graphql-wszgraphql-transport-wsNF Turlheadersssl init_payloadconnect_timeout close_timeout ack_timeoutkeep_alive_timeout ping_interval pong_timeout answer_pings connect_args subprotocolsreturnc st||||||||| | |_|| |_| dur'| dur$| d|_n| |_d|_t|_ t|_ | durC|j |j g|_ dS| |_ dS)aInitialize the transport with the given parameters. :param url: The GraphQL server URL. Example: 'wss://server.com:PORT/graphql'. :param headers: Dict of HTTP Headers. :param ssl: ssl_context of the connection. Use ssl=False to disable encryption :param init_payload: Dict of the payload sent in the connection_init message. :param connect_timeout: Timeout in seconds for the establishment of the websocket connection. If None is provided this will wait forever. :param close_timeout: Timeout in seconds for the close. If None is provided this will wait forever. :param ack_timeout: Timeout in seconds to wait for the connection_ack message from the server. If None is provided this will wait forever. :param keep_alive_timeout: Optional Timeout in seconds to receive a sign of liveness from the server. :param ping_interval: Delay in seconds between pings sent by the client to the backend for the graphql-ws protocol. None (by default) means that we don't send pings. Note: there are also pings sent by the underlying websockets protocol. See the :ref:`keepalive documentation ` for more information about this. :param pong_timeout: Delay in seconds to receive a pong from the backend after we sent a ping (only for the graphql-ws protocol). By default equal to half of the ping_interval. :param answer_pings: Whether the client answers the pings from the backend (for the graphql-ws protocol). By default: True :param connect_args: Other parameters forwarded to `websockets.connect `_ :param subprotocols: list of subprotocols sent to the backend in the 'subprotocols' http header. By default: both apollo and graphql-ws subprotocols. N) super__init__rr!r send_ping_taskasyncioEvent ping_received pong_receivedAPOLLO_SUBPROTOCOLGRAPHQLWS_SUBPROTOCOLsupported_subprotocols)selfrrrrrrrrrr r!r"r# __class__o/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/gql/transport/websockets.pyr'#s82     zWebsocketsTransport.__init__cs@ |IdH}||\}}}|dkrdS|dkrtdq)zDWait for the connection_ack message. Keep alive messages are ignoredTNconnection_ackkaz0Websocket server did not return a connection ack)_receive _parse_answerr)r0 init_answer answer_type answer_idexecution_resultr3r3r4 _wait_ack}szWebsocketsTransport._wait_ackcs@td|jd}||IdHt||jIdHdS)zSend init message to the provided websocket and wait for the connection ACK. If the answer is not a connection_ack message, we will return an Exception. connection_init)typepayloadN)jsondumpsr_sendr)wait_forr=r)r0 init_messager3r3r4_send_init_message_and_wait_acks  z3WebsocketsTransport._send_init_message_and_wait_ackcs|IdHdSN)rFr0r3r3r4 _initializeszWebsocketsTransport._initializer@c4ddi}|dur ||d<|t|IdHdS)z/Send a ping message for the graphql-ws protocolr?pingNr@rCrArB)r0r@ ping_messager3r3r4 send_ping zWebsocketsTransport.send_pingcrJ)z/Send a pong message for the graphql-ws protocolr?pongNr@rL)r0r@ pong_messager3r3r4 send_pongrOzWebsocketsTransport.send_pongquery_idc*tt|dd}||IdHdS)zSend stop message to the provided websocket connection and query_id. The server should afterwards return a 'complete' message. stopidr?NrArBstrrC)r0rS stop_messager3r3r4_send_stop_messagez&WebsocketsTransport._send_stop_messagecrT)znSend a complete message for the provided query_id. This is only for the graphql-ws protocol. completerVNrX)r0rScomplete_messager3r3r4_send_complete_messager\z*WebsocketsTransport._send_complete_messagecs\td||j|jkr$||IdH|j|dIdHdS||IdHdS)aoStop the listener corresponding to the query_id depending on the detected backend protocol. For apollo: send a "stop" message (a "complete" message will be sent from the backend) For graphql-ws: send a "complete" message and simulate the reception of a "complete" message from the backend zstop listener N)r]N)logdebug subprotocolr.r_ listenersputr[)r0rSr3r3r4_stop_listeners  z"WebsocketsTransport._stop_listenercs$tddi}||IdHdS)zSend a connection_terminate message to the provided websocket connection. This message indicates that the connection will disconnect. r?connection_terminateN)rArBrC)r0connection_terminate_messager3r3r4"_send_connection_terminate_messagesz6WebsocketsTransport._send_connection_terminate_messagedocumentvariable_valuesoperation_namecsx|j}|jd7_dt|i}|r||d<|r||d<d}|j|jkr'd}tt|||d}||IdH|S) zSend a query to the provided websocket connection. We use an incremented id to reference the query. Returns the used id for this query. rquery variables operationNamestart subscribe)rWr?r@N) next_query_idr rbr.rArBrYrC)r0rirjrkrSr@ query_type query_strr3r3r4 _send_querys    zWebsocketsTransport._send_querycs$|j|jkr|IdHdSdSrG)rbr-rhrHr3r3r4_connection_terminates z)WebsocketsTransport._connection_terminate json_answerc CsDd}d}d}zt|d}|dvrmtt|d}|dks#|dkrl|d}|dkrTt|ts5td d |vrAd |vrAtd t|d |d |d d}d }n)|dkrlt|tsatdtt|d||dn|dvr{|dd|j |<nt|j dur|j Wnty}zt d||d}~ww|||fS)aParse the answer received from the server if the server supports the graphql-ws protocol. Returns a list consisting of: - the answer_type (between: 'connection_ack', 'ping', 'pong', 'data', 'error', 'complete') - the answer id (Integer) if received or None - an execution Result if the answer_type is 'data' or None Differences with the apollo websockets protocol (superclass): - the "data" message is now called "next" - the "stop" message is now called "complete" - there is no connection_terminate or connection_error messages - instead of a unidirectional keep-alive (ka) message from server to client, there is now the possibility to send bidirectional ping/pong messages - connection_ack has an optional payload - the 'error' answer type returns a list of errors instead of a single error Nr?)nexterrorr]rWrxryr@payload is not a dicterrorsdata2payload does not contain 'data' or 'errors' fields extensionsr{r|r~zpayload is not a listrrSr{)rKrPr5(Server did not return a GraphQL result: )rYgetint isinstancedict ValueErrorr listrpayloadscheck_keep_alive_task_next_keep_alive_messagesetr)r0rvr:r;r<r@er3r3r4_parse_answer_graphqlws sX      z+WebsocketsTransport._parse_answer_graphqlwsc CsJd}d}d}zt|d}|dvratt|d}|dks#|dkr`|d}t|ts1td |dkrRd |vrAd|vrAtd t|d |d|d d }n8|dkr`tt|||gdn)|dkrp|jduro|j n|dkrun|dkr|d}t dt |dtWnty}zt d||d}~ww|||fS)aParse the answer received from the server if the server supports the apollo websockets protocol. Returns a list consisting of: - the answer_type (between: 'connection_ack', 'ka', 'connection_error', 'data', 'error', 'complete') - the answer id (Integer) if received or None - an execution Result if the answer_type is 'data' or None rwNr?)r|ryr]rWr|ryr@rzr{r}r~rrr6r5connection_errorzServer error: ''r)rYrrrrrr rrrrrreprr)r0rvr:r;r<r@ error_payloadrr3r3r4_parse_answer_apolloZs\        z(WebsocketsTransport._parse_answer_apolloanswercCsLzt|}Wntytd|w|j|jkr!||S||S)zaParse the answer received from the server depending on the detected subprotocol. r)rAloadsrrrbr.rr)r0rrvr3r3r4r8s    z!WebsocketsTransport._parse_answercs|jdusJz$ t|jIdH|IdHt|j|jIdH|jq tj yP|j durM|j t d|jdddIdHYdSYdSw)aCoroutine to periodically send a ping from the client to the backend. Only used for the graphql-ws protocol. Send a ping every ping_interval seconds. Close the connection if a pong is not received within pong_timeout seconds. NTzNo pong received after z secondsF) clean_close) rr)sleeprNrDr,waitr clear TimeoutError close_task_failrrHr3r3r4_send_ping_coros&    z#WebsocketsTransport._send_ping_coror:r;r<cs`t|||IdH|dkr#|j|jr!|IdHdSdS|dkr.|jdSdS)NrKrP)r&_handle_answerr+rr!rRr,)r0r:r;r<r1r3r4rs z"WebsocketsTransport._handle_answercsJ|jj}z|d|_Wn ty|j|_Ynwtd|jdS)NzSec-WebSocket-Protocolzbackend subprotocol returned: ) websocketresponse_headersrbKeyErrorr-r`ra)r0rr3r3r4_after_connects  z"WebsocketsTransport._after_connectcs4|j|jkr|jdurt||_dSdSdSrG)rbr.rr) ensure_futurerr(rHr3r3r4_after_initializes   z%WebsocketsTransport._after_initializecsZ|jdur+|jttj|jIdHWdn1s!wYd|_dSdSrG)r(cancelrr)CancelledErrorrHr3r3r4 _close_hooks    zWebsocketsTransport._close_hook)r$NrG)NN)+__name__ __module__ __qualname____doc__r rr-r.rYrrr rboolrrrfloatrr'r=rFrIrNrRr[r_rerhr rtrurr rrr8rrrrr __classcell__r3r3r1r4rs          Z       #  P  D !  r)!r)rAlogging contextlibrrrtypingrrrrrr r graphqlr r r websockets.datastructuresrwebsockets.typingr exceptionsrrrwebsockets_baser getLoggerrr`rr3r3r3r4s  $