87 lines
3.4 KiB
Python
87 lines
3.4 KiB
Python
# 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.connect("mqtt", 1883)
|
|
client.loop_start()
|
|
|
|
POLL_INTERVAL = 30 # Sekunden, Empfehlung für seltene Änderungen
|
|
last_payloads = {} # group_id -> payload
|
|
while True:
|
|
now = datetime.datetime.now(datetime.timezone.utc)
|
|
# Hole alle aktiven Events (Vergleich mit UTC)
|
|
events = get_active_events(now, now)
|
|
# Gruppiere Events nach group_id
|
|
groups = {}
|
|
for event in events:
|
|
gid = getattr(event, "group_id", None)
|
|
if gid not in groups:
|
|
groups[gid] = []
|
|
groups[gid].append({
|
|
"id": event.id,
|
|
"title": getattr(event, "title", ""),
|
|
"start": str(getattr(event, "start", "")),
|
|
"end": str(getattr(event, "end", "")),
|
|
"group_id": gid,
|
|
})
|
|
# Sende pro Gruppe die Eventliste als retained Message, nur bei Änderung
|
|
for gid, event_list in groups.items():
|
|
topic = f"infoscreen/events/{gid}"
|
|
payload = json.dumps(event_list)
|
|
if last_payloads.get(gid) != payload:
|
|
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: {payload}")
|
|
last_payloads[gid] = payload
|
|
# Entferne Gruppen, die nicht mehr existieren (optional: retained Message löschen)
|
|
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]
|
|
time.sleep(POLL_INTERVAL)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|