o tBh@sddlZddlZddlmZddlmZmZmZmZm Z m Z ddl m Z ddl mZmZmZddlmZmZddlmZmZdd lmZmZed ZzddlZWn eyZYnwGd d d eZdS) N) SSLContext)AnyDictOptionalTupleUnioncast)urlparse) DocumentNodeExecutionResult print_ast)AppSyncAuthenticationAppSyncIAMAuthentication)TransportProtocolErrorTransportServerError)WebsocketsTransportWebsocketsTransportBasezgql.transport.appsynccs2eZdZUdZeeed<dddddddifdedeededd ee e fd e d e d e d eee e fde eefddffdd Zdedeeee eeffddZ  ddedee eefdeede fddZejZ  ddedee eefdeedefddZejZejZejZejZZS)AppSyncWebsocketsTransportz:ref:`Async Transport ` used to execute GraphQL subscription on AWS appsync realtime endpoint. This transport uses asyncio and the websockets library in order to send requests on a websocket connection. authNF urlsessionzbotocore.session.Sessionsslconnect_timeout close_timeout ack_timeoutkeep_alive_timeout connect_argsreturnc s`|stt|j} t| |d}||_|j|}tj||||||| dtj g|_ tj |_ dS)aInitialize the transport with the given parameters. :param url: The GraphQL endpoint URL. Example: https://XXXXXXXXXXXXXXXXXXXXXXXXXX.appsync-api.REGION.amazonaws.com/graphql :param auth: Optional AWS authentication class which will provide the necessary headers to be correctly authenticated. If this argument is not provided, then we will try to authenticate using IAM. :param ssl: ssl_context of the connection. :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 connect_args: Other parameters forwarded to websockets.connect )hostr)rrrrrrN) strr netlocrr get_auth_urlsuper__init__rAPOLLO_SUBPROTOCOLsupported_subprotocols subprotocol) selfrrrrrrrrrr  __class__w/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/gql/transport/appsync_websockets.pyr% s"     z#AppSyncWebsocketsTransport.__init__answercCsd}z2t|}t|d}|dkrWdS|dkr+d|vr+|d}td|d ttt||WStyBt d |w) akParse the answer received from the server. Difference between apollo protocol and aws protocol: - aws protocol can return an error without an id - aws protocol will send start_ack messages Returns a list consisting of: - the answer_type: - 'connection_ack', - 'connection_error', - 'start_ack', - 'ka', - 'data', - 'error', - 'complete' - the answer id (Integer) if received or None - an execution Result if the answer_type is 'data' or None type start_ack)r1NNerroridpayloadzServer error: ''z(Server did not return a GraphQL result: ) jsonloadsr!getrr_parse_answer_apollor ValueErrorr)r)r. answer_type json_answer error_payloadr,r,r- _parse_answer\s"    z(AppSyncWebsocketsTransport._parse_answerdocumentvariable_valuesoperation_namec s|j}|jd7_dt|i}|r||d<|r||d<tj|dd}d|i}t|d|d }|jdus7Jd |j|i|d d <|tj|ddIdH|S) Nr query variables operationName),:) separatorsdatastart)r3r0r4 authorizationr4 extensions) next_query_idr r6dumpsr!r get_headers_send) r)r?r@rAquery_idrHserialized_datar4messager,r,r- _send_querys0    z&AppSyncWebsocketsTransport._send_querycs td)zThis method is not available. Only subscriptions are supported on the AWS realtime endpoint. :raise: AssertionErrorz}execute method is not allowed for AppSyncWebsocketsTransport because only subscriptions are allowed on the realtime endpoint.)AssertionError)r)r?r@rAr,r,r-executes z"AppSyncWebsocketsTransport.execute)NN)__name__ __module__ __qualname____doc__rr__annotations__r!rrboolintfloatrrr%rr r>r rSr subscriberUr _initialize_send_stop_message_stop_listener_send_init_message_and_wait_ack _wait_ack __classcell__r,r,r*r-rs      < 3 ,  r)r6loggingrrtypingrrrrrr urllib.parser graphqlr r r appsync_authrr exceptionsrr websocketsrr getLoggerlogbotocore ImportErrorrr,r,r,r-s