5iU.dZddlmZddlmZddlmZddlmZm Z m Z ddl m Z m Z mZmZddlmZddlZGd d eZeGd d ZeGd dZeegdfZ eegdfZ eGddZeGddZeGddZeGddZeGddZGdde ZGdde Zy)z MQTT Request Response module )Sequence)IntEnum) dataclass)CallableUnionOptional)NativeResourcemqtt5mqtt exceptions)FutureNc eZdZdZdZ dZ dZy)SubscriptionStatusEventTypezO The type of change to the state of a streaming operation subscription rN)__name__ __module__ __qualname____doc__SUBSCRIPTION_ESTABLISHEDSUBSCRIPTION_LOSTSUBSCRIPTION_HALTED~/home/marpiech/ifpan-abm-pgxpred/analysis/marpiech-gwas-test/venv/lib/python3.12/site-packages/awscrt/mqtt_request_response.pyrrs. !rrc.eZdZUdZdZeed<dZded<y)SubscriptionStatusEventaR An event that describes a change in subscription status for a streaming operation. Args: type (SubscriptionStatusEventType): The type of status change represented by the event error (Optional[Exception]): Describes an underlying reason for the event. Only set for SubscriptionLost and SubscriptionHalted. NtypezOptional[Exception]error)rrrrrr__annotations__rrrrrr's)-D %,#'E 'rrc*eZdZUdZeed<dZded<y)IncomingPublishEventz An event that describes an incoming message on a streaming operation. Args: topic (str): MQTT Topic that the response was received on. payload (Optional[bytes]): The payload of the incoming message. topicNOptional[bytes]payloadrrrrstrr r%rrrr"r"4 J!%G %rr"c>eZdZUdZeed<dZded<dZded<dZy) StreamingOperationOptionsa Configuration options for an MQTT-based streaming operation. Args: subscription_topic_filter (str): Topic filter that the streaming operation should listen on subscription_status_listener (SubscriptionStatusListener): function object to invoke when the operation's subscription status changes incoming_publish_listener (IncomingPublishListener): function object to invoke when a publish packet arrives that matches the subscription topic filter subscription_topic_filterNz$Optional[SubscriptionStatusListener]subscription_status_listenerz!Optional[IncomingPublishListener]incoming_publish_listenerct|jtsJt|js|jJt|j s|j JyyzE Stringently type-checks an instance's field values. N) isinstancer+r'callabler,r-selfs rvalidatez"StreamingOperationOptions.validateZs^$88#>>>99:d>_>_>ggg6674;Y;Y;aaa;a7r) rrrrr'r r,r-r4rrrr*r*Ls- #"KO "HOEIBIbrr*c*eZdZUdZeed<dZded<y)Responsez Encapsulates a response to an AWS IoT Core MQTT-based service request Args: topic (str): MQTT Topic that the response was received on. payload (Optional[bytes]): The payload of the response. r#Nr$r%r&rrrr6r6cr(rr6c0eZdZUdZeed<dZded<dZy) ResponsePatha A response path is a pair of values - MQTT topic and a JSON path - that describe how a response to an MQTT-based request may arrive. For a given request type, there may be multiple response paths and each one is associated with a separate JSON schema for the response body. Args: topic (str): MQTT topic that a response may arrive on. correlation_token_json_path (Optional[str]): JSON path for finding correlation tokens within payloads that arrive on this path's topic. r#N Optional[str]correlation_token_json_pathct|jtsJt|jts|jJyyr/)r0r#r'r:r2s rr4zResponsePath.validate~s?$**c***$::C@DDdDdDlllDl@r)rrrrr'r r:r4rrrr8r8ps J377mrr8cNeZdZUdZded<ded<eed<eed<dZd ed <d Zy) RequestOptionsa Configuration options for an MQTT-based request-response operation. Args: subscription_topic_filters (Sequence[str]): Set of topic filters that should be subscribed to in order to cover all possible response paths. Sometimes using wildcards can cut down on the subscriptions needed; other times that isn't valid. response_paths (Sequence[ResponsePath]): Set of all possible response paths associated with this request type. publish_topic (str): Topic to publish the request to once response subscriptions have been established. payload (bytes): Payload to publish to 'publishTopic' in order to initiate the request correlation_token (Optional[str]): Correlation token embedded in the request that must be found in a response message. This can be null to support certain services which don't use correlation tokens. In that case, the client only allows one token-less request at a time. z Sequence[str]subscription_topic_filterszSequence[ResponsePath]response_paths publish_topicr%Nr9correlation_tokenct|jtsJ|jD]}t|trJt|jtsJ|jD]}|j t|j tsJt|jtsJt|jts|jJyyr/) r0r>rr'r?r4r@r%bytesrA)r3 topic_filter response_paths rr4zRequestOptions.validates$998DDD ;; 1LlC0 00 1$--x888!00 %M  " " $ %$,,c222$,,...$00#6$:P:P:XXX:X6r) rrrrr r'rCrAr4rrrr=r=s1 !0/,, N)--Yrr=c:eZdZUdZeed<eed<dZded<dZy) ClientOptionsa  MQTT-based request-response client configuration options Args: max_request_response_subscriptions (int): Maximum number of subscriptions that the client will concurrently use for request-response operations max_streaming_subscriptions (int): Maximum number of subscriptions that the client will concurrently use for streaming operations operation_timeout_in_seconds (Optional[int]): Duration, in seconds, that a request-response operation will wait for completion before giving up "max_request_response_subscriptionsmax_streaming_subscriptions<z Optional[int]operation_timeout_in_secondsct|jtsJt|jtsJt|jtsJyr/)r0rHintrIrKr2s rr4zClientOptions.validatesF$AA3GGG$::C@@@$;;SAAArN)rrrrrMr rKr4rrrrGrGs&),+!$$46 /6BrrGcveZdZdZdeej ejfde ffd Z de fdZ de fdZxZS)ClientaL MQTT-based request-response client tuned for AWS MQTT services. Supports streaming operations (listen to a stream of modeled events from an MQTT topic) and request-response operations (performs the subscribes, publish, and incoming publish correlation and error checking needed to perform simple request-response operations over MQTT). Args: protocol_client (Union[mqtt5.Client, mqtt.Connection]): MQTT client to use as transport client_options (ClientOptions): The ClientOptions dataclass to used to configure the new request response Client. protocol_clientclient_optionsctt|tjst|tjsJt|t sJ|j t|!t|tjrtj|||_ ytj|||_ yN) r0r rOr ConnectionrGr4super__init___awscrt'mqtt_request_response_client_new_from_5_binding)mqtt_request_response_client_new_from_311)r3rPrQ __class__s rrVzClient.__init__s/5<<8JX\XgXg.on_request_completes=Q$$Z%9%9*%EF#%A!!(+r) r4r rW)mqtt_request_response_client_make_requestrYr>r?r@r%rA)r3r\rerds @r make_requestzClient.make_requestsi  , 99$--:A:\:\:A:P:P:A:O:O:A//:A:S:S:M  O rcjfd}fd}tj|jj||}t |S)a Creates a new streaming operation Args: options (StreamingOperationOptions): Configuration options for the streaming operation Returns: StreamingOperation: a new streaming operation. Opening the operation triggers the client to maintain an MQTT subscription for relevant events. Matching publishes and subscription status changes are communicated by invoking configuration-controlled callbacks. cj.on_subscription_status_event sF33?/ ;?","6"6z"BEK44U; @rcZjt||}j|yyrS)r-r")r#r%rkr\s ron_incoming_publish_eventz7Client.create_stream..on_incoming_publish_events.00<,UG<11%8=r)r4rW*mqtt_request_response_client_create_streamrYr+StreamingOperation)r3r\rlrnstream_bindings ` r create_streamzClient.create_streamsP  < 9 !KK MM7<<>Z\uw".11r)rrrrrr rOr rTrGrVr=rgr*rr __classcell__r[s@rrOrOsK  oellDOO.K(L o!. oN@2%>2rrOc(eZdZdZfdZdZxZS)rpzT An operation that represents a stream of events broadcast to an MQTT topic c0t|||_yrS)rUrVrY)r3bindingr[s rrVzStreamingOperation.__init__!s  rcBtj|jy)z Triggers the streaming operation to maintain an MQTT subscription for relevant events. Until a stream is opened, no events can be received. N)rWmqtt_streaming_operation_openrYr2s ropenzStreamingOperation.open&s --dmmrs@%!,,::%',  ( (  (  & &  &&'>&?&EF#$8#94#?@  bb b,  & &  & mm m* YY YD BB B,Y2^Y2x==r