add initial dev version of ha addon

nspanel-lovelace-ui/rootfs/usr/bin/mqtt-manager/main.py

@@ -0,0 +1,112 @@
+#!/usr/bin/env python
+import logging
+import paho.mqtt.client as mqtt
+import time
+import json
+import subprocess
+import libs.home_assistant
+import libs.panel_cmd
+import yaml
+from uuid import getnode as get_mac
+from panel import LovelaceUIPanel
+import os
+
+settings = {}
+panels = {}
+last_settings_file_mtime = 0
+mqtt_connect_time = 0
+has_sent_reload_command = False
+mqtt_client_name = "NSPanelLovelaceManager_" + str(get_mac())
+client = mqtt.Client(mqtt_client_name)
+logging.basicConfig(level=logging.DEBUG)
+
+
+def on_connect(client, userdata, flags, rc):
+    # global settings
+    logging.info("Connected to MQTT Server")
+    # for panel in settings["nspanels"].values():
+    #    print(panel)
+    #    client.subscribe(panel["panelRecvTopic"])
+    client.subscribe("tele/tasmota_nspdev2/RESULT")
+
+
+def on_message(client, userdata, msg):
+    try:
+        if msg.payload.decode() == "":
+            return
+        parts = msg.topic.split('/')
+        if msg.topic == "tele/tasmota_nspdev2/RESULT":
+            data = json.loads(msg.payload.decode('utf-8'))
+            if "CustomRecv" in data:
+                # print(data["CustomRecv"])
+                if parts[1] in panels:
+                    panels[parts[1]].customrecv_event_callback(
+                        data["CustomRecv"])
+                else:
+                    logging.error(
+                        "Got message for unknown panel: %s - %s", parts[1], data["CustomRecv"])
+        else:
+            logging.debug("Received unhandled message on topic: %s", msg.topic)
+
+    except Exception:  # pylint: disable=broad-exception-caught
+        logging.exception("Something went wrong during processing of message:")
+        try:
+            logging.error(msg.payload.decode('utf-8'))
+        except:  # pylint: disable=bare-except
+            logging.error(
+                "Something went wrong when processing the exception message, couldn't decode payload to utf-8.")
+
+
+def get_config():
+    global settings
+    CONFIG_FILE = os.getenv('CONFIG_FILE')
+    if not CONFIG_FILE:
+        CONFIG_FILE = 'config.yml'
+    with open(CONFIG_FILE, 'r', encoding="utf8") as file:
+        settings = yaml.safe_load(file)
+
+
+def connect_and_loop():
+    global settings, home_assistant
+    client.on_connect = on_connect
+    client.on_message = on_message
+    client.username_pw_set(
+        settings["mqtt_username"], settings["mqtt_password"])
+    # Wait for connection
+    connection_return_code = 0
+    mqtt_server = settings["mqtt_server"]
+    mqtt_port = int(settings["mqtt_port"])
+    logging.info("Connecting to %s:%i as %s",
+                 mqtt_server, mqtt_port, mqtt_client_name)
+    while True:
+        try:
+            client.connect(mqtt_server, mqtt_port, 5)
+            break  # Connection call did not raise exception, connection is sucessfull
+        except:  # pylint: disable=bare-except
+            logging.exception(
+                "Failed to connect to MQTT %s:%i. Will try again in 10 seconds. Code: %s", mqtt_server, mqtt_port, connection_return_code)
+            time.sleep(10.)
+
+    # MQTT Connected, start APIs if configured
+    if settings["home_assistant_address"] != "" and settings["home_assistant_token"] != "":
+        libs.home_assistant.init(settings, client)
+        libs.home_assistant.connect()
+    else:
+        logging.info("Home Assistant values not configured, will not connect.")
+
+    libs.panel_cmd.init(client)
+
+    # Create NsPanel object
+    for name, settings in settings["nspanels"].items():
+        panels[name] = LovelaceUIPanel(client, name, settings)
+        libs.panel_cmd.page_type(
+            settings["panelSendTopic"], "pageStartup")
+
+    # Loop MQTT
+    client.loop_forever()
+
+
+if __name__ == '__main__':
+    get_config()
+    # print(settings)
+    connect_and_loop()