Initial commit - copied workspace after database cleanup

scheduler/scheduler.py

@@ -0,0 +1,112 @@
+# scheduler/scheduler.py
+
+import os
+import logging
+from .db_utils import get_active_events
+import paho.mqtt.client as mqtt
+import json
+import datetime
+import time
+
+# Logging-Konfiguration
+ENV = os.getenv("ENV", "development")
+LOG_LEVEL = os.getenv("LOG_LEVEL", "DEBUG" if ENV == "development" else "INFO")
+LOG_PATH = os.path.join(os.path.dirname(__file__), "scheduler.log")
+os.makedirs(os.path.dirname(LOG_PATH), exist_ok=True)
+log_handlers = []
+if ENV == "production":
+    from logging.handlers import RotatingFileHandler
+    log_handlers.append(RotatingFileHandler(
+        LOG_PATH, maxBytes=2*1024*1024, backupCount=5, encoding="utf-8"))
+else:
+    log_handlers.append(logging.FileHandler(LOG_PATH, encoding="utf-8"))
+if os.getenv("DEBUG_MODE", "1" if ENV == "development" else "0") in ("1", "true", "True"):
+    log_handlers.append(logging.StreamHandler())
+logging.basicConfig(
+    level=getattr(logging, LOG_LEVEL.upper(), logging.INFO),
+    format="%(asctime)s [%(levelname)s] %(message)s",
+    handlers=log_handlers
+)
+
+
+def main():
+    # Fix für die veraltete API - explizit callback_api_version setzen
+    client = mqtt.Client(callback_api_version=mqtt.CallbackAPIVersion.VERSION2)
+    client.reconnect_delay_set(min_delay=1, max_delay=30)
+
+    POLL_INTERVAL = 30  # Sekunden, Empfehlung für seltene Änderungen
+    # 0 = aus; z.B. 600 für alle 10 Min
+    REFRESH_SECONDS = int(os.getenv("REFRESH_SECONDS", "0"))
+    last_payloads = {}       # group_id -> payload
+    last_published_at = {}   # group_id -> epoch seconds
+
+    # Beim (Re-)Connect alle bekannten retained Payloads erneut senden
+    def on_connect(client, userdata, flags, reasonCode, properties=None):
+        logging.info(
+            f"MQTT connected (reasonCode={reasonCode}) - republishing {len(last_payloads)} groups")
+        for gid, payload in last_payloads.items():
+            topic = f"infoscreen/events/{gid}"
+            client.publish(topic, payload, retain=True)
+
+    client.on_connect = on_connect
+
+    client.connect("mqtt", 1883)
+    client.loop_start()
+
+    while True:
+        now = datetime.datetime.now(datetime.timezone.utc)
+        # Hole alle aktiven Events (bereits formatierte Dictionaries)
+        events = get_active_events(now, now)
+
+        # Gruppiere Events nach group_id
+        groups = {}
+        for event in events:
+            gid = event.get("group_id")
+            if gid not in groups:
+                groups[gid] = []
+            # Event ist bereits ein Dictionary im gewünschten Format
+            groups[gid].append(event)
+
+        # Sende pro Gruppe die Eventliste als retained Message, nur bei Änderung
+        for gid, event_list in groups.items():
+            # stabile Reihenfolge, um unnötige Publishes zu vermeiden
+            event_list.sort(key=lambda e: (e.get("start"), e.get("id")))
+            payload = json.dumps(
+                event_list, sort_keys=True, separators=(",", ":"))
+            topic = f"infoscreen/events/{gid}"
+
+            should_send = (last_payloads.get(gid) != payload)
+            if not should_send and REFRESH_SECONDS:
+                last_ts = last_published_at.get(gid, 0)
+                if time.time() - last_ts >= REFRESH_SECONDS:
+                    should_send = True
+
+            if should_send:
+                result = client.publish(topic, payload, retain=True)
+                if result.rc != mqtt.MQTT_ERR_SUCCESS:
+                    logging.error(
+                        f"Fehler beim Publish für Gruppe {gid}: {mqtt.error_string(result.rc)}")
+                else:
+                    logging.info(f"Events für Gruppe {gid} gesendet")
+                    last_payloads[gid] = payload
+                    last_published_at[gid] = time.time()
+
+        # Entferne Gruppen, die nicht mehr existieren (leere retained Message senden)
+        for gid in list(last_payloads.keys()):
+            if gid not in groups:
+                topic = f"infoscreen/events/{gid}"
+                result = client.publish(topic, payload="[]", retain=True)
+                if result.rc != mqtt.MQTT_ERR_SUCCESS:
+                    logging.error(
+                        f"Fehler beim Entfernen für Gruppe {gid}: {mqtt.error_string(result.rc)}")
+                else:
+                    logging.info(
+                        f"Events für Gruppe {gid} entfernt (leere retained Message gesendet)")
+                    del last_payloads[gid]
+                    last_published_at.pop(gid, None)
+
+        time.sleep(POLL_INTERVAL)
+
+
+if __name__ == "__main__":
+    main()