def on_broker_connect(client: mqtt_client.Client, userdata: dict, flags, rc):
print('Connected with MQTT broker with result code ' + str(rc))
if userdata[USER_DATA_FIRST_CONNECT]:
if int(rc) != 0:
raise Exception('Connect to MQTT broker failed.')
client.user_data_set({USER_DATA_FIRST_CONNECT: False})
class MqttHandler:
def __init__(self, client_id='DEFAULT_CLIENT_ID', topic='DEFAULT_TOPIC', broker_host='localhost',
broker_port=MQTT_BROKER_PORT):
self.subscribed = False
self.client_id = client_id
self.client = Client(client_id=self.client_id, protocol=MQTT_PROTOCOL_VERSION)
self.client.on_message = self.on_message_callback
self.client.on_publish = self.on_publish_callback
self.client.on_connect = self.connect_callback
self.client.on_disconnect = self.disconnect_callback
self.topic = topic
self.broker_host = broker_host
self.broker_port = broker_port
self.message_received = 0
userdata = {
USER_DATA_MESSAGE_RECEIVED: 0,
}
self.client.user_data_set(userdata)
def connect(self):
self.client.connect(host=self.broker_host, port=self.broker_port)
def connect_async(self):
self.client.connect_async(host=self.broker_host, port=self.broker_port)
def connect_callback(self, client, userdata, flags, rc):
print('connect_callback: result code[' + str(rc) + ']')
(result, _) = client.subscribe(topic=self.topic)
self.subscribed = result
def disconnect(self):
self.client.disconnect()
def disconnect_callback(self, client, userdata, rc):
print('disconnect_callback')
def is_valid(self, my_json: json):
if app.config.get('DEBUG', False):
print("json_validation")
# try:
# if my_json['id'] is None or my_json['byte_stream'] is None:
# return False
# except KeyError:
# return False
return True
def on_message_callback(self, client, userdata, message):
from core.socketio_runner import emit_command
userdata[USER_DATA_MESSAGE_RECEIVED] += 1
topic = message.topic
payload = json.loads(message.payload)
if app.config.get('DEBUG', False):
print('on_message_callback: topic[' + topic + ']')
if self.is_valid(payload):
emit_command(topic, payload)
else:
raise Exception('Message payload not valid')
@staticmethod
def publish_single_message(topic, payload=None, qos=0, retain=False, hostname="localhost",
port=MQTT_BROKER_PORT, client_id="", keepalive=60, will=None, auth=None, tls=None):
if app.config.get('DEBUG', False):
print("publish_single_message")
single(topic=topic, payload=payload, qos=qos, retain=retain, hostname=hostname, port=port, client_id=client_id,
keepalive=keepalive, will=will, auth=auth, tls=tls)
def on_publish_callback(self, client, userdata, mid):
print('on_publish_callback')
def loop_for_ever(self):
self.client.loop_forever()
def loop_start(self):
self.client.loop_start()
def loop_stop(self, force=False):
self.client.loop_stop(force=force)
def on_broker_connect(client: mqtt_client.Client, userdata: dict, flags, rc):
print('Connected with MQTT broker with result code ' + str(rc))
if userdata[USER_DATA_FIRST_CONNECT]:
if int(rc) != 0:
raise Exception('Failed to connect to MQTT broker.')
client.user_data_set({USER_DATA_FIRST_CONNECT: False})
