U Zj-@sdZddlmZddlmZddZddZd d Zd d d dddd d d ejddd f ddZ d ddd dddd d d ejddd fddZ d S)a This module provides some helper functions to allow straightforward subscribing to topics and retrieving messages. The two functions are simple(), which returns one or messages matching a set of topics, and callback() which allows you to pass a callback for processing of messages. )mqtt)clientcCs^|dkrtt|t|dtrF|dD]}|||dq.n||d|ddS)Internal callbacktopicsqosN)rZ MQTTExceptionpahoZconnack_string isinstancelist subscribe)ruserdataflagsZ reason_code propertiesZtopicrV/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/paho/mqtt/subscribe.py _on_connects  rcCs|d||d|dS)rcallbackr Nrrr messagerrr_on_message_callback&srcCs|ddkrdS|jr"|ds"dS|dd|d<|ddkr^|ddkr^||d<|dS|d||ddkr|dS)r msg_countrNretainedrmessages)ZretainZ disconnectappendrrrr_on_message_simple+s  rrN localhosti[<ZtcpTcCs |dks|dkrtd||||d}tjtjj||| | | d}|t|_t|_ |dk rh|j f|| r| d}|r| d}| ||nt d |dk r|jf|| dk rt| tr| d d }|jf| |r||n || |||||dS) aT Subscribe to a list of topics and process them in a callback function. This function creates an MQTT client, connects to a broker and subscribes to a list of topics. Incoming messages are processed by the user provided callback. This is a blocking function and will never return. :param callback: function with the same signature as `on_message` for processing the messages received. :param topics: either a string containing a single topic to subscribe to, or a list of topics to subscribe to. :param int qos: the qos to use when subscribing. This is applied to all topics. :param userdata: passed to the callback :param str hostname: the address of the broker to connect to. Defaults to localhost. :param int port: the port to connect to the broker on. Defaults to 1883. :param str client_id: the MQTT client id to use. If "" or None, the Paho library will generate a client id automatically. :param int keepalive: the keepalive timeout value for the client. Defaults to 60 seconds. :param will: a dict containing will parameters for the client: will = {'topic': "", 'payload':", 'qos':, 'retain':}. Topic is required, all other parameters are optional and will default to None, 0 and False respectively. Defaults to None, which indicates no will should be used. :param auth: a dict containing authentication parameters for the client: auth = {'username':"", 'password':""} Username is required, password is optional and will default to None if not provided. Defaults to None, which indicates no authentication is to be used. :param tls: a dict containing TLS configuration parameters for the client: dict = {'ca_certs':"", 'certfile':"", 'keyfile':"", 'tls_version':"", 'ciphers':", 'insecure':""} ca_certs is required, all other parameters are optional and will default to None if not provided, which results in the client using the default behaviour - see the paho.mqtt.client documentation. Alternatively, tls input can be an SSLContext object, which will be processed using the tls_set_context method. Defaults to None, which indicates that TLS should not be used. :param str transport: set to "tcp" to use the default setting of transport which is raw TCP. Set to "websockets" to use WebSockets as the transport. :param clean_session: a boolean that determines the client type. If True, the broker will remove all information about this client when it disconnects. If False, the client is a persistent client and subscription information and queued messages will be retained when the client disconnects. Defaults to True. :param proxy_args: a dictionary that will be given to the client. rrzqos must be in the range 0-2)rrrr ) client_idr protocol transport clean_sessionNusernamepasswordz;The 'username' key was not found, this is required for authinsecureF) ValueErrorr ZClientZCallbackAPIVersionZVERSION2Z enable_loggerrZ on_messagerZ on_connectZ proxy_setgetZusername_pw_setKeyErrorZwill_setr dictpopZtls_setZtls_insecure_setZtls_set_contextconnectZ loop_forever)rrrr hostnameportr keepalivewillauthtlsr r!r" proxy_argsZcallback_userdatarr#r$r%rrrrAsJD         rcCsh|dkrtd|dkrd}ng}| tjkr0d} |||d}tt||||||||| | | | | ||dS)aSubscribe to a list of topics and return msg_count messages. This function creates an MQTT client, connects to a broker and subscribes to a list of topics. Once "msg_count" messages have been received, it disconnects cleanly from the broker and returns the messages. :param topics: either a string containing a single topic to subscribe to, or a list of topics to subscribe to. :param int qos: the qos to use when subscribing. This is applied to all topics. :param int msg_count: the number of messages to retrieve from the broker. if msg_count == 1 then a single MQTTMessage will be returned. if msg_count > 1 then a list of MQTTMessages will be returned. :param bool retained: If set to True, retained messages will be processed the same as non-retained messages. If set to False, retained messages will be ignored. This means that with retained=False and msg_count=1, the function will return the first message received that does not have the retained flag set. :param str hostname: the address of the broker to connect to. Defaults to localhost. :param int port: the port to connect to the broker on. Defaults to 1883. :param str client_id: the MQTT client id to use. If "" or None, the Paho library will generate a client id automatically. :param int keepalive: the keepalive timeout value for the client. Defaults to 60 seconds. :param will: a dict containing will parameters for the client: will = {'topic': "", 'payload':", 'qos':, 'retain':}. Topic is required, all other parameters are optional and will default to None, 0 and False respectively. Defaults to None, which indicates no will should be used. :param auth: a dict containing authentication parameters for the client: auth = {'username':"", 'password':""} Username is required, password is optional and will default to None if not provided. Defaults to None, which indicates no authentication is to be used. :param tls: a dict containing TLS configuration parameters for the client: dict = {'ca_certs':"", 'certfile':"", 'keyfile':"", 'tls_version':"", 'ciphers':", 'insecure':""} ca_certs is required, all other parameters are optional and will default to None if not provided, which results in the client using the default behaviour - see the paho.mqtt.client documentation. Alternatively, tls input can be an SSLContext object, which will be processed using the tls_set_context method. Defaults to None, which indicates that TLS should not be used. :param protocol: the MQTT protocol version to use. Defaults to MQTTv311. :param transport: set to "tcp" to use the default setting of transport which is raw TCP. Set to "websockets" to use WebSockets as the transport. :param clean_session: a boolean that determines the client type. If True, the broker will remove all information about this client when it disconnects. If False, the client is a persistent client and subscription information and queued messages will be retained when the client disconnects. Defaults to True. If protocol is MQTTv50, clean_session is ignored. :param proxy_args: a dictionary that will be given to the client. rzmsg_count must be > 0N)rrrr)r&r ZMQTTv5rr)rrrrr,r-rr.r/r0r1r r!r"r2rr rrrsimples(K  r3) __doc__rrrr rrrZMQTTv311rr3rrrrs:    y