#!/usr/bin/env python3 """ Listen for mqtt-events, and trigger for some """ import json from datetime import datetime import common import paho.mqtt.client as mqtt mqtt_server = common.env("el_mqtt_server") mqtt_port = int(common.env("el_mqtt_port")) keepalive = int(common.env("el_mqtt_keepalive")) mqtt_topic = common.env("el_mqtt_topic") mqtt_user = common.env("el_mqtt_user") mqtt_pass = common.env("el_mqtt_pass") tempsensors = [ "Bad Temp", "Barnerom Temp", "Isobod Temp", "Kjøkken Temp Matskap", "Kontor Temp", "Loft Temp", "Soverom Temp", "Stue Temp Display", "Stue Temp Stuebord", "Stue Temp Teleskap", "Toalett motion", "Utebod Temp", "Vaskerom Temp", ] # The callback for when the client receives a CONNACK response from the server. def on_connect(client, userdata, flags, rc): print(f"Connected with result code {rc}") # Subscribing in on_connect() means that if we lose the connection and # reconnect then subscriptions will be renewed. client.subscribe(mqtt_topic) # The callback for when a PUBLISH message is received from the server. def on_message(client, userdata, msg): name = msg.topic.split("/")[1] data = json.loads(msg.payload) if name == "HAN" and "current" in data: sql = """INSERT INTO mqtt_han (name, current, power, voltage, linkquality, time) VALUES(%s,%s,%s,%s,%s,%s)""" values = ( name, data["current"], data["power"], data["voltage"], data["linkquality"], datetime.utcnow(), ) common.dbi(sql, values, verbose=True) if name in tempsensors: if "temperature" in data: common.statein(name, data["temperature"], "temperature", "°C", verbose=True) if "humidity" in data: common.statein(name, data["humidity"], "humidity", "%", verbose=True) # mqtt client = mqtt.Client() client.on_connect = on_connect client.on_message = on_message client.username_pw_set(mqtt_user, password=mqtt_pass) client.connect(mqtt_server, mqtt_port, keepalive) # Blocking call that processes network traffic, dispatches callbacks and # handles reconnecting. # Other loop*() functions are available that give a threaded interface and a # manual interface. client.loop_forever()