university-final-iot-backend/app/mod_devices/__init__.py

35 lines
901 B
Python

from flask_mqtt import Mqtt
mqtt = Mqtt()
def setup_mqtt(app):
mqtt.init_app(app)
mqtt.client.on_message = handle_mqtt_message
mqtt.client.on_subscribe = handle_subscribe
print('MQTT client initialized')
def tear_down_mqtt():
mqtt.unsubscribe_all()
if hasattr(mqtt, 'client') and mqtt.client is not None:
mqtt.client.disconnect()
print('MQTT client destroyed')
@mqtt.on_connect()
def handle_connect(client, userdata, flags, rc):
print('MQTT client connected')
mqtt.subscribe('topic/state')
@mqtt.on_disconnect()
def handle_disconnect():
print('MQTT client disconnected')
def handle_subscribe(client, userdata, mid, granted_qos):
print('MQTT client subscribed')
def handle_mqtt_message(client, userdata, message):
data = dict(
topic=message.topic,
payload=message.payload.decode()
)
print(message.payload.decode())