initial commit

dashboard/utils/mqtt_client.py

@@ -0,0 +1,121 @@
+# dashboard/utils/mqtt_client.py
+
+import os
+import threading
+import time
+from dotenv import load_dotenv
+import paho.mqtt.client as mqtt
+
+# 1. Laden der Umgebungsvariablen aus .env
+load_dotenv(dotenv_path=os.path.join(os.path.dirname(__file__), "..", ".env"))
+
+# 2. Lese MQTT‐Einstellungen
+MQTT_BROKER_HOST    = os.getenv("MQTT_BROKER_HOST", "localhost")
+MQTT_BROKER_PORT    = int(os.getenv("MQTT_BROKER_PORT", "1883"))
+MQTT_USERNAME       = os.getenv("MQTT_USERNAME", None)
+MQTT_PASSWORD       = os.getenv("MQTT_PASSWORD", None)
+MQTT_KEEPALIVE      = int(os.getenv("MQTT_KEEPALIVE", "60"))
+MQTT_CLIENT_ID      = os.getenv("MQTT_CLIENT_ID", f"dash-{int(time.time())}")
+
+# 3. Erstelle eine globale Client‐Instanz
+client = mqtt.Client(client_id=MQTT_CLIENT_ID)
+
+# Falls Nutzer/Passwort gesetzt sind, authentifizieren
+if MQTT_USERNAME and MQTT_PASSWORD:
+    client.username_pw_set(MQTT_USERNAME, MQTT_PASSWORD)
+
+
+# 4. Callback‐Stubs (kannst du bei Bedarf anpassen)
+def _on_connect(client, userdata, flags, rc):
+    if rc == 0:
+        print(f"[mqtt_client.py] Erfolgreich mit MQTT‐Broker verbunden (Code {rc})")
+    else:
+        print(f"[mqtt_client.py] Verbindungsfehler, rc={rc}")
+
+
+def _on_disconnect(client, userdata, rc):
+    print(f"[mqtt_client.py] Verbindung getrennt (rc={rc}). Versuche, neu zu verbinden …")
+
+
+def _on_message(client, userdata, msg):
+    """
+    Diese Callback‐Funktion wird aufgerufen, sobald eine Nachricht auf einem
+    Topic ankommt, auf das wir subscribed haben. Du kannst hier eine Queue
+    füllen oder direkt eine Datenbank‐Funktion aufrufen.
+    """
+    topic = msg.topic
+    payload = msg.payload.decode("utf-8", errors="ignore")
+    print(f"[mqtt_client.py] Nachricht eingegangen – Topic: {topic}, Payload: {payload}")
+    # Beispiel: Wenn du Live‐Statusdaten in die Datenbank schreibst,
+    # könntest du hier utils/db.execute_non_query(...) aufrufen.
+
+
+# 5. Setze die Callbacks
+client.on_connect    = _on_connect
+client.on_disconnect = _on_disconnect
+client.on_message    = _on_message
+
+
+def start_loop():
+    """
+    Startet die Endlos‐Schleife, in der der Client auf eingehende
+    MQTT‐Nachrichten hört und automatisch reconnectet.
+    Muss idealerweise in einem eigenen Thread laufen, damit Dash‐Callbacks
+    nicht blockieren.
+    """
+    try:
+        client.connect(MQTT_BROKER_HOST, MQTT_BROKER_PORT, keepalive=MQTT_KEEPALIVE)
+        client.loop_start()
+    except Exception as e:
+        print(f"[mqtt_client.py] Konnte keine Verbindung zum MQTT‐Broker herstellen: {e}")
+
+
+def stop_loop():
+    """
+    Stoppt die MQTT‐Loop und trennt die Verbindung.
+    """
+    try:
+        client.loop_stop()
+        client.disconnect()
+    except Exception as e:
+        print(f"[mqtt_client.py] Fehler beim Stoppen der MQTT‐Schleife: {e}")
+
+
+def publish(topic: str, payload: str, qos: int = 0, retain: bool = False) -> bool:
+    """
+    Verschickt eine MQTT‐Nachricht:
+    - topic: z. B. "clients/{client_id}/control"
+    - payload: z. B. '{"command":"restart"}'
+    - qos: 0, 1 oder 2
+    - retain: True/False
+    Rückgabe: True, falls Veröffentlichung bestätigt wurde; sonst False.
+    """
+    try:
+        result = client.publish(topic, payload, qos=qos, retain=retain)
+        status = result.rc  # 0=Erfolg, sonst Fehler
+        if status == mqtt.MQTT_ERR_SUCCESS:
+            return True
+        else:
+            print(f"[mqtt_client.py] Publish-Fehler für Topic {topic}, rc={status}")
+            return False
+    except Exception as e:
+        print(f"[mqtt_client.py] Exception beim Publish: {e}")
+        return False
+
+
+def subscribe(topic: str, qos: int = 0) -> bool:
+    """
+    Abonniert ein MQTT‐Topic, sodass _on_message gerufen wird, sobald Nachrichten
+    ankommen.
+    Rückgabe: True bei Erfolg, ansonsten False.
+    """
+    try:
+        result, mid = client.subscribe(topic, qos=qos)
+        if result == mqtt.MQTT_ERR_SUCCESS:
+            return True
+        else:
+            print(f"[mqtt_client.py] Subscribe‐Fehler für Topic {topic}, rc={result}")
+            return False
+    except Exception as e:
+        print(f"[mqtt_client.py] Exception beim Subscribe: {e}")
+        return False