49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
import logging
|
|
from mqtt_flask_instance import mqtt
|
|
import config
|
|
from db import db
|
|
from schemas import DeviceSchema, SensorSchema
|
|
|
|
"""
|
|
MQTT Stuff
|
|
"""
|
|
|
|
__author__ = "@tormakris"
|
|
__copyright__ = "Copyright 2020, Birbnetes Team"
|
|
__module_name__ = "mqtt_methods"
|
|
__version__text__ = "1"
|
|
|
|
deviceschema = DeviceSchema(many=False)
|
|
sensorschema = SensorSchema(many=False)
|
|
|
|
LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
@mqtt.on_message()
|
|
def handle_status_message(client, userdata, message):
|
|
data = dict(
|
|
topic=message.topic,
|
|
payload=message.payload.decode()
|
|
)
|
|
LOGGER.info(data)
|
|
try:
|
|
ids = data['topic'].replace(f"{config.MQTT_STATUS_TOPIC}/", "").split("/")
|
|
if len(ids) == 1:
|
|
data['payload']['deviceID'] = ids[0]
|
|
status_message = deviceschema.load(data['payload'], session=db.session, transient=True).data
|
|
LOGGER.info(f"Recieved status message from {data['payload']['deviceID']}, persisting to db.")
|
|
db.session.merge(status_message)
|
|
else:
|
|
if len(ids) == 2:
|
|
data['payload']['deviceID'] = ids[0]
|
|
data['payload']['sensorID'] = ids[1]
|
|
status_message = sensorschema.load(data['payload'], session=db.session, transient=True).data
|
|
LOGGER.info(f"Recieved status message from sensor {data['payload']['sensorID']}, persisting to db.")
|
|
db.session.merge(status_message)
|
|
except Exception as e:
|
|
db.session.rollback()
|
|
LOGGER.exception(e)
|
|
else:
|
|
db.session.commit()
|