Scheduler: Refactor database utilities and

scheduler logic
This commit is contained in:
2025-07-15 05:05:09 +00:00
parent 2fa84c1e2b
commit 661d25d70c
3 changed files with 54 additions and 7 deletions

25
scheduler/db_utils.py Normal file
View File

@@ -0,0 +1,25 @@
# scheduler/db_utils.py
from models import Event
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from datetime import datetime
import os
import sys
sys.path.append('/workspace/server')
# DB-URL aus Umgebungsvariable oder Fallback
DB_CONN = os.environ.get("DB_CONN", "mysql+pymysql://user:password@db/dbname")
engine = create_engine(DB_CONN)
Session = sessionmaker(bind=engine)
def get_active_events(start: datetime, end: datetime, group_id: int = None):
session = Session()
query = session.query(Event).filter(Event.is_active == True)
if start and end:
query = query.filter(Event.start < end, Event.end > start)
if group_id:
query = query.filter(Event.group_id == group_id)
events = query.all()
session.close()
return events

View File

@@ -1,19 +1,41 @@
# scheduler/scheduler.py
import time
from db_utils import get_active_events
import paho.mqtt.client as mqtt
import json
import datetime
import time
def main():
# Fix für die veraltete API - explizit callback_api_version setzen
client = mqtt.Client(callback_api_version=mqtt.CallbackAPIVersion.VERSION2)
# Im Docker-Netzwerk: Hostname des MQTT-Brokers ist "mqtt", nicht "localhost"
client.connect("mqtt", 1883)
POLL_INTERVAL = 10 # Sekunden
last_sent_event_ids = set()
while True:
# Hier später: Events aus DB lesen und MQTT-Nachricht senden
print("Scheduler läuft...")
time.sleep(10)
now = datetime.datetime.now()
# Hole alle Events, die jetzt aktiv sind (start < now < end)
events = get_active_events(now, now)
current_event_ids = set(event.id for event in events)
# Sende nur neue Events (die noch nicht gesendet wurden)
new_event_ids = current_event_ids - last_sent_event_ids
for event in events:
if event.id in new_event_ids:
# Beispiel: Sende Event-Daten als JSON auf Topic "infoscreen/events"
payload = json.dumps({
"id": event.id,
"title": getattr(event, "title", ""),
"start": str(getattr(event, "start", "")),
"end": str(getattr(event, "end", "")),
"group_id": getattr(event, "group_id", None),
})
client.publish("infoscreen/events", payload)
print(f"Event {event.id} gesendet: {payload}")
last_sent_event_ids = current_event_ids
time.sleep(POLL_INTERVAL)
if __name__ == "__main__":