5iI_4dZddlZddlmZmZddlmZddlZddlm Z ddl m Z m Z m Z ddlmZddlmZdd lmZdd lmZdd lmZmZmZgd ZGd deZGddZGddeZdZdZdZ GddeZ!GddeZ"GddeZ#y)zI event-stream RPC (remote procedure call) protocol library for `awscrt`. N)ABCabstractmethod)NativeResource)Header)ClientBootstrap SocketOptionsTlsConnectionOptions)Callable)Future)IntEnum)partial)OptionalSequenceUnion) MessageType MessageFlagClientConnectionHandlerClientConnectionClientContinuationClientContinuationHandlercFeZdZdZdZ dZ dZ dZ dZ dZ dZ d Z d Z y ) raTypes of messages in the event-stream RPC protocol. The :attr:`~MessageType.APPLICATION_MESSAGE` and :attr:`~MessageType.APPLICATION_ERROR` types may only be sent on streams, and will never arrive as a protocol message (stream-id 0). For all other message types, they may only be sent as protocol messages (stream-id 0), and will never arrive as a stream message. Different message types expect specific headers and flags, consult documentation.rct|SNstrself format_specs x/home/marpiech/ifpan-abm-pgxpred/analysis/marpiech-gwas-test/venv/lib/python3.12/site-packages/awscrt/eventstream/rpc.py __format__zMessageType.__format__C 4yN) __name__ __module__ __qualname____doc__APPLICATION_MESSAGEAPPLICATION_ERRORPING PING_RESPONSECONNECT CONNECT_ACKPROTOCOL_ERRORINTERNAL_ERRORr'r)r&rrsVY DMGKmNNr)rc(eZdZdZdZ dZ dZ dZy)rzFlags for messages in the event-stream RPC protocol. Flags may be XORed together. Not all flags can be used with all message types, consult documentation. rrrct|Sr r!r#s r&r'zMessageFlag.__format__ar(r)N)r*r+r,r-NONECONNECTION_ACCEPTEDTERMINATE_STREAMr'r6r)r&rrHs0 DhZ r)rc neZdZdZed dZedeeddfdZede e de d e d e ddf d Zy) raBase class for handling connection events. Inherit from this class and override methods to handle connection events. All callbacks for this connection will be invoked on the same thread, and :meth:`on_connection_setup()` will always be the first callback invoked. returnNc y)a^Invoked upon completion of the setup attempt. If setup was successful, the connection is provided to the user. Note that the network connection stays alive until it is closed, even if no local references to the connection object remain. The user should store a reference to this connection, and call `connection.close()` when they are done with it to avoid leaking resources. Setup will always be the first callback invoked on the handler. If setup failed, no further callbacks will be invoked on this handler. Args: connection: The connection, if setup was successful, or None if setup failed. error: None, if setup was successful, or an Exception if setup failed. `**kwargs`: Forward compatibility kwargs. Nr6)r$ connectionerrorkwargss r&on_connection_setupz+ClientConnectionHandler.on_connection_setupn0 r)reasonc y)aKInvoked when the connection finishes shutting down. This event will not be invoked if connection setup failed. Args: reason: Reason will be None if the user initiated the shutdown, otherwise the reason will be an Exception. **kwargs: Forward compatibility kwargs. Nr6)r$rDrAs r&on_connection_shutdownz.ClientConnectionHandler.on_connection_shutdowns r)headerspayload message_typeflagsc y)aInvoked when a message for the connection (stream-id 0) is received. Args: headers: Message headers. payload: Binary message payload. message_type: Message type. flags: Message flags. Values from :class:`MessageFlag` may be XORed together. Not all flags can be used with all message types, consult documentation. **kwargs: Forward compatibility kwargs. Nr6r$rGrHrIrJrAs r&on_protocol_messagez+ClientConnectionHandler.on_protocol_messagerCr)r=N)r*r+r,r-rrBr ExceptionrFrrbytesrintrMr6r)r&rrfs  2  Xi-@  t     f%  &       r)rc|g}n|Dcgc]}|j}}|d}|tj}||||fScc}w)z~ Transform args that a python send-msg function would take, into args that a native send-msg function would take. r))_as_binding_tuplerr9rGrHrIrJis r&_to_binding_msg_argsrVsX 29:Q1&&(:: }   WlE 22 ;s?cz|Dcgc]}tj|}}|d}t|}||||fScc}w)zr Transform msg-received args that came from native, into msg-received args presented to python users. r))r_from_binding_tuplerrTs r&_from_binding_msg_argsrYsJ 7>>v))!,>G>|,L WlE 22 ?s8c|rtjj|nd} |r |||r|j|y|j dy#|r|j|w|j dwxYw)N)r@)awscrt exceptions from_code set_exception set_result) bound_futurebound_callback error_codees r&_on_message_flushrdsp3=##J/4A*   #   & &q )  # #D )   & &q )  # #D )s A'A=ceZdZdZgdZfdZedddddedede d e d e e d e e d d fdZdZdZdZdZdZdddddde eede eeefdede e ded d f dZddZxZS)raA client connection for the event-stream RPC protocol. Use :meth:`ClientConnection.connect()` to establish a new connection. Note that the network connection stays alive until it is closed, even if no local references to the connection object remain. The user should store a reference to any connections, and call :meth:`close()` when they are done with them to avoid leaking resources. Attributes: host_name (str): Remote host name. port (int): Remote port. shutdown_future (concurrent.futures.Future[None]): Completes when this connection has finished shutting down. Future will contain a result of None, or an exception indicating why shutdown occurred. ) host_nameportshutdown_future_connect_future_handlerct|||_||_t |_|j j t |_|jj ||_yr ) super__init__rfrgr rhset_running_or_notify_cancelrirj)r$rfrghandler __class__s r&rmzClientConnection.__init__s[ " %x 99;%x 99; r)N) bootstrapsocket_optionstls_connection_optionsrorfrgrqrrrsr=concurrent.futures.Futurec|s t}||||}|stj}tj|||||||j S)a,Asynchronously establish a new ClientConnection. Args: handler: Handler for connection events. host_name: Connect to host. port: Connect to port. bootstrap: Client bootstrap to use when initiating socket connection. If None is provided, the default singleton is used. socket_options: Optional socket options. If None is provided, then default options are used. tls_connection_options: Optional TLS connection options. If None is provided, then the connection will be attempted over plain-text. Returns: concurrent.futures.Future: A Future which completes when the connection succeeds or fails. If successful, the Future will contain None. Otherwise it will contain an exception. If the connection is successful, it will be made available via the handler's on_connection_setup callback. Note that this network connection stays alive until it is closed, even if no local references to the connection object remain. The user should store a reference to any connections, and call :meth:`close()` when they are done with them to avoid leaking resources. )rrget_or_create_static_default_awscrt*event_stream_rpc_client_connection_connectri)clsrorfrgrqrrrsr?s r&connectzClientConnection.connects`R*_ND'2 'DDFI ::     "   )))r)c~|r"d}tjj|}n|}d} |jj |||r|j j |y|j jdy#|r|j j |w|j jdwxYw)N)r?r@)r[r\r]rjrBrir^r_)r$rbr?r@s r&_on_connection_setupz%ClientConnection._on_connection_setup;s J%%// ;EJE 6 MM - -5 - Q$$2259$$//5$$2259$$//5s B;B<cr|rtjj|nd} |jj ||r|j j |y|j jdy#|r|j j |w|j jdwxYw)N)rD)r[r\r]rjrFrhr^r_)r$rbrDs r&_on_connection_shutdownz(ClientConnection._on_connection_shutdownLs @r)r=rtc t}|jt||||\}}}}tj|j ||||t t|||S)a Send a continuation message. Use the returned future, or the `on_flush` callback, to be informed when the message is successfully written to the wire, or fails to send. Note that the the first message on a stream-id must be sent with activate(), send_message() is for all messages that follow. Keyword Args: operation: Operation name for this stream. headers: Message headers. payload: Binary message payload. message_type: Message type. flags: Message flags. Values from :class:`MessageFlag` may be XORed together. Not all flags can be used with all message types, consult documentation. on_flush: Callback invoked when the message is successfully written to the wire, or fails to send. The function should take the following arguments and return nothing: * `error` (Optional[Exception]): None if the message was successfully written to the wire, or an Exception if it failed to send. * `**kwargs` (dict): Forward compatibility kwargs. This callback is always invoked on the connection's event-loop thread. Returns: A future which completes with a result of None if the message is successfully written to the wire, or an exception if the message fails to send. )r rnrVrw1event_stream_rpc_client_continuation_send_messagerr rdrs r& send_messagezClientContinuation.send_message(sh`++-0DWgWcej0k-,AA MM     %vx 8  : r)c@tj|jSr )rw.event_stream_rpc_client_continuation_is_closedrrs r& is_closedzClientContinuation.is_closedfsEEdmmTTr)c |jj|jjdy#|jjdwxYwr )rjon_continuation_closedrr_rs r&_on_continuation_closedz*ClientContinuation._on_continuation_closedisB 0 MM 0 0 2    ) )$ /D   ) )$ /s 8Acht||||\}}}}|jj||||yr)rYrjon_continuation_messagers r&_on_continuation_messagez+ClientContinuation._on_continuation_messagepsA0FwPWYegl0m-, --% . r))r*r+r,r-rmr"rrrrPrrrQr rrrrrrrs@r&rrs :)-/3!%EEf% E 5)+, E & EEET)-/3!%<f%<5)+, < & <  <<+F<|U0r)rc NeZdZdZedeededede ddf dZ ed d Z y) raBase class for handling stream continuation events. Inherit from this class and override methods to handle events. All callbacks will be invoked on the same thread (the same thread used by the connection). A common pattern is to store the continuation within its handler. Example:: continuation_handler.continuation = connection.new_stream(continuation_handler) rGrHrIrJr=Nc y)aInvoked when a message is received on this continuation. Args: headers: Message headers. payload: Binary message payload. message_type: Message type. flags: Message flags. Values from :class:`MessageFlag` may be XORed together. Not all flags can be used with all message types, consult documentation. **kwargs: Forward compatibility kwargs. Nr6rLs r&rz1ClientContinuationHandler.on_continuation_messages. r)c y)aYInvoked when the continuation is closed. Once the continuation is closed, no more messages may be sent or received. The continuation is closed when a message is sent or received with the TERMINATE_STREAM flag, or when the connection shuts down. Args: **kwargs: Forward compatibility kwargs. Nr6)r$rAs r&rz0ClientContinuationHandler.on_continuation_closeds r)rN) r*r+r,r-rrrrPrrQrrr6r)r&rrzsd  f%  &       0    r)r)$r-rwabcrrr[rawscrt.exceptionsawscrt.eventstreamr awscrt.iorrr collections.abcr concurrent.futuresr enumr functoolsr typingrrr__all__rrrrVrYrdrrrr6r)r&rs#!%JJ$%,, '''T