remove state capabilities
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
2021-06-12 19:55:27 +02:00
parent 6deb6321ed
commit 028313380f
10 changed files with 13 additions and 241 deletions

View File

@@ -1,8 +1,6 @@
#!/usr/bin/env python3
from flask import current_app as app
import config
from db import db
from schemas import DeviceSchema, SensorSchema
"""
MQTT Stuff
@@ -13,9 +11,6 @@ __copyright__ = "Copyright 2020, Birbnetes Team"
__module_name__ = "mqtt_methods"
__version__text__ = "1"
deviceschema = DeviceSchema(many=False)
sensorschema = SensorSchema(many=False)
def handle_status_message(client, userdata, message):
data = dict(
@@ -27,18 +22,11 @@ def handle_status_message(client, userdata, message):
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
app.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
app.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()
app.logger.exception(e)
else:
db.session.commit()