Initial commit - copied workspace after database cleanup

server/mqtt_helper.py

@@ -0,0 +1,142 @@
+"""
+Einfache MQTT-Hilfsfunktion für Client-Gruppenzuordnungen
+"""
+import os
+import json
+import logging
+import paho.mqtt.client as mqtt
+
+logger = logging.getLogger(__name__)
+
+
+def publish_client_group(client_uuid: str, group_id: int) -> bool:
+    """
+    Publiziert die Gruppenzuordnung eines Clients als retained message
+
+    Args:
+        client_uuid: UUID des Clients
+        group_id: ID der Gruppe
+
+    Returns:
+        bool: True bei Erfolg, False bei Fehler
+    """
+    try:
+        # MQTT-Konfiguration aus .env
+        broker_host = os.getenv("MQTT_BROKER_HOST", "mqtt")
+        broker_port = int(os.getenv("MQTT_BROKER_PORT", 1883))
+        username = os.getenv("MQTT_USER")
+        password = os.getenv("MQTT_PASSWORD")
+
+        # Topic und Payload
+        topic = f"infoscreen/{client_uuid}/group_id"
+        payload = json.dumps({
+            "group_id": group_id,
+            "client_uuid": client_uuid
+        })
+
+        # MQTT-Client erstellen und verbinden
+        client = mqtt.Client()
+        if username and password:
+            client.username_pw_set(username, password)
+
+        client.connect(broker_host, broker_port, 60)
+
+        # Retained message publizieren
+        result = client.publish(topic, payload, qos=1, retain=True)
+        result.wait_for_publish(timeout=5.0)
+
+        client.disconnect()
+
+        logger.info(
+            f"Group assignment published for client {client_uuid}: group_id={group_id}")
+        return True
+
+    except Exception as e:
+        logger.error(
+            f"Error publishing group assignment for client {client_uuid}: {e}")
+        return False
+
+
+def publish_multiple_client_groups(client_group_mappings: dict) -> tuple[int, int]:
+    """
+    Publiziert Gruppenzuordnungen für mehrere Clients in einer Verbindung
+
+    Args:
+        client_group_mappings: Dict mit {client_uuid: group_id}
+
+    Returns:
+        tuple: (success_count, failed_count)
+    """
+    try:
+        broker_host = os.getenv("MQTT_BROKER_HOST", "mqtt")
+        broker_port = int(os.getenv("MQTT_BROKER_PORT", 1883))
+        username = os.getenv("MQTT_USER")
+        password = os.getenv("MQTT_PASSWORD")
+
+        client = mqtt.Client()
+        if username and password:
+            client.username_pw_set(username, password)
+
+        client.connect(broker_host, broker_port, 60)
+
+        success_count = 0
+        failed_count = 0
+
+        for client_uuid, group_id in client_group_mappings.items():
+            try:
+                topic = f"infoscreen/{client_uuid}/group_id"
+                payload = json.dumps({
+                    "group_id": group_id,
+                    "client_uuid": client_uuid
+                })
+
+                result = client.publish(topic, payload, qos=1, retain=True)
+                result.wait_for_publish(timeout=5.0)
+                success_count += 1
+
+            except Exception as e:
+                logger.error(f"Failed to publish for {client_uuid}: {e}")
+                failed_count += 1
+
+        client.disconnect()
+
+        logger.info(
+            f"Bulk publish completed: {success_count} success, {failed_count} failed")
+        return success_count, failed_count
+
+    except Exception as e:
+        logger.error(f"Error in bulk publish: {e}")
+        return 0, len(client_group_mappings)
+
+
+def delete_client_group_message(client_uuid: str) -> bool:
+    """
+    Löscht die retained message für einen Client (bei Client-Löschung)
+    """
+    try:
+        broker_host = os.getenv("MQTT_BROKER_HOST", "mqtt")
+        broker_port = int(os.getenv("MQTT_BROKER_PORT", 1883))
+        username = os.getenv("MQTT_USER")
+        password = os.getenv("MQTT_PASSWORD")
+
+        topic = f"infoscreen/{client_uuid}/group_id"
+
+        client = mqtt.Client()
+        if username and password:
+            client.username_pw_set(username, password)
+
+        client.connect(broker_host, broker_port, 60)
+
+        # Leere retained message löscht die vorherige
+        result = client.publish(topic, "", qos=1, retain=True)
+        result.wait_for_publish(timeout=5.0)
+
+        client.disconnect()
+
+        logger.info(f"Deleted retained group message for client {client_uuid}")
+        return True
+
+    except Exception as e:
+        logger.error(
+            f"Error deleting group message for client {client_uuid}: {e}")
+        return False