add printout

printout.py

@@ -0,0 +1,109 @@
+import logging
+from time import sleep
+
+import paho.mqtt.client as mqtt
+from rich.align import Align
+from rich.layout import Layout
+from rich.live import Live
+from rich.panel import Panel
+from rich.table import Table
+
+MQTT_HOST = "nas.home.kunbox.net"
+MQTT_SUB = "sdm630/"
+TITLE = "SDM630"
+TABLE_LAYOUT = {
+    "row1": {
+        "voltage": {
+            "name": "Voltage",
+            "suffix": "V",
+        },
+        "current": {
+            "name": "Current",
+            "suffix": "A",
+        },
+        "current-demand": {
+            "name": "Current Demand",
+            "suffix": "A",
+        },
+    },
+    "row2": {
+        "apparent-power": {
+            "name": "Apparent Power",
+            "suffix": "W",
+        },
+        "reactive-power": {
+            "name": "Reactive Power",
+            "suffix": "VAr",
+        },
+        "power-factor": {
+            "name": "Power Factor",
+            "suffix": "",
+        },
+    },
+}
+
+# logging.basicConfig(level=logging.DEBUG)
+LOG = logging.getLogger(__name__)
+
+mqtt_data = {}
+
+
+def on_connect(client, userdata, flags, rc):
+    LOG.info(f"Connected to mqtt server")
+    mqtt.subscribe(f"{MQTT_SUB}#")
+
+
+def on_disconnect(client, userdata, rc):
+    LOG.info(f"Disconnected from mqtt server")
+
+
+def on_message(client, userdata, msg):
+    value = msg.payload.decode()
+    try:
+        value = round(float(value), 2)
+    except Exception:
+        pass
+    mqtt_data[msg.topic[len(MQTT_SUB) :]] = value
+
+
+def generate_layout():
+    layout = Layout()
+
+    layout.split_column(
+        Layout(Panel(Align.center(TITLE)), name="header"),
+        Layout(name="main"),
+    )
+    layout["header"].size = 3
+
+    layout["main"].split_column(*[Layout(name=k) for k in TABLE_LAYOUT.keys()])
+    for k, i in TABLE_LAYOUT.items():
+        layout["main"][k].split_row(*[Layout(name=v) for v in i.keys()])
+
+    for k, i in TABLE_LAYOUT.items():
+        for k2, v in i.items():
+            table = Table(
+                title=v["name"],
+                show_header=False,
+                box=None,
+            )
+            for topic, value in sorted(mqtt_data.items()):
+                if topic.startswith(f"{k2}/"):
+                    table.add_row(topic[len(f"{k2}/") :], f"{value} {v['suffix']}")
+            layout["main"][k][k2].update(Panel(Align.center(table)))
+    return layout
+
+
+try:
+    mqtt = mqtt.Client()
+    mqtt.on_connect = on_connect
+    mqtt.on_disconnect = on_disconnect
+    mqtt.on_message = on_message
+    mqtt.connect(MQTT_HOST, 1883, 10)
+    mqtt.loop_start()
+
+    with Live(generate_layout(), screen=True) as live:
+        while True:
+            live.update(generate_layout())
+            sleep(0.5)
+except Exception:
+    LOG.exception("oops")