status update consumption done
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
2020-07-20 17:09:05 +02:00
parent 689d530996
commit 009e3e4933
7 changed files with 164 additions and 2 deletions

33
src/mqtt_methods.py Normal file
View File

@@ -0,0 +1,33 @@
#!/usr/bin/env python3
import logging
from mqtt_flask_instance import mqtt
from db import db
from schemas import DeviceSchema
"""
MQTT Stuff
"""
__author__ = "@tormakris"
__copyright__ = "Copyright 2020, Birbnetes Team"
__module_name__ = "mqtt_methods"
__version__text__ = "1"
deviceschema = DeviceSchema(many=False)
@mqtt.on_message()
def handle_status_message(client, userdata, message):
data = dict(
topic=message.topic,
payload=message.payload.decode()
)
try:
status_message = deviceschema.load(data['payload'], session=db.session, transient=True).data
logging.info(f"Recieved status message from {data['payload']['id']}, persisting to db.")
db.session.merge(status_message)
except Exception as e:
db.session.rollback()
logging.exception(e)
else:
db.session.commit()