#!/usr/bin/env python3 """Busylight WLED Controller - Web Application.""" import hashlib import json import logging import os import secrets import shutil import sys from contextlib import asynccontextmanager from datetime import datetime from urllib.parse import quote from fastapi import FastAPI, Request, Form, UploadFile, File, BackgroundTasks from fastapi.responses import HTMLResponse, RedirectResponse, FileResponse, JSONResponse from fastapi.staticfiles import StaticFiles from fastapi.templating import Jinja2Templates from apscheduler.schedulers.background import BackgroundScheduler import database import scheduler from graph_client import GraphAPI logging.basicConfig( level=logging.INFO, format="%(asctime)s %(levelname)-8s %(name)s: %(message)s", datefmt="%Y-%m-%d %H:%M:%S", ) logger = logging.getLogger(__name__) bg_scheduler = BackgroundScheduler() @asynccontextmanager async def lifespan(application: FastAPI): database.init_db() # Init Graph client try: scheduler.graph = GraphAPI() except Exception as e: logger.error("Graph-Client konnte nicht initialisiert werden: %s", e) # Verzeichnisse erstellen os.makedirs(BACKUP_DIR, exist_ok=True) os.makedirs(WLED_BACKUP_DIR, exist_ok=True) os.makedirs(FIRMWARE_DIR, exist_ok=True) os.makedirs(CUSTOM_DIR, exist_ok=True) # Start background polling interval = int(database.get_setting("poll_interval", "15")) bg_scheduler.add_job(scheduler.poll_status, "interval", seconds=interval, id="poll") bg_scheduler.add_job(scheduler.check_devices, "interval", seconds=240, id="device_check") bg_scheduler.add_job(auto_backup_db, "cron", hour=2, minute=0, id="auto_backup") bg_scheduler.add_job(auto_backup_wled, "cron", day=1, hour=2, minute=5, id="auto_backup_wled") # Initial device check (nicht blockierend) bg_scheduler.add_job(scheduler.check_devices, "date", id="device_check_init") bg_scheduler.start() logger.info("Polling gestartet (Intervall: %ds)", interval) yield bg_scheduler.shutdown() BACKUP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "backups") WLED_BACKUP_DIR = os.path.join(BACKUP_DIR, "wled") FIRMWARE_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "firmware") # Kundenspezifische Dateien (Logo). Liegt bewusst ausserhalb der Versionierung, # damit ein git pull das Kundenlogo nicht ueberschreibt. CUSTOM_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "static", "custom") LOGO_EXTENSIONS = (".png", ".jpg", ".jpeg", ".svg", ".webp", ".gif") LOGO_MAX_BYTES = 2 * 1024 * 1024 def auto_backup_db(): """Create daily DB backup. Keeps last 14 backups.""" ts = datetime.now().strftime("%Y%m%d_%H%M%S") dst = os.path.join(BACKUP_DIR, f"busylight_{ts}.db") shutil.copy2(database.DB_PATH, dst) # Keep only last 14 backups = sorted( [f for f in os.listdir(BACKUP_DIR) if f.startswith("busylight_") and f.endswith(".db")], reverse=True, ) for old in backups[14:]: os.remove(os.path.join(BACKUP_DIR, old)) logger.info("Auto-Backup erstellt: %s", dst) def auto_backup_wled(): """Backup all WLED configs. Keeps last 14 per device.""" from wled_client import get_wled_config conn = database.get_db() devices = conn.execute("SELECT id, hostname FROM wled_devices").fetchall() conn.close() ts = datetime.now().strftime("%Y%m%d_%H%M%S") count = 0 for dev in devices: config = get_wled_config(dev["hostname"]) if config: safe_name = dev["hostname"].replace(".", "_") path = os.path.join(WLED_BACKUP_DIR, f"{safe_name}_{ts}.json") with open(path, "w") as f: json.dump(config, f, indent=2) count += 1 # Keep only last 14 per device prefix = safe_name + "_" device_backups = sorted( [f for f in os.listdir(WLED_BACKUP_DIR) if f.startswith(prefix)], reverse=True, ) for old in device_backups[14:]: os.remove(os.path.join(WLED_BACKUP_DIR, old)) logger.info("WLED Auto-Backup: %d Geraete gesichert", count) app = FastAPI(title="Busylight Manager", lifespan=lifespan) app.mount("/static", StaticFiles(directory="static"), name="static") templates = Jinja2Templates(directory="templates") SESSION_TOKEN = secrets.token_hex(32) # Regenerated on restart DEFAULT_LOGO = "/static/logo.png" def _logo_url(): """URL des anzuzeigenden Logos - kundenspezifisch, sonst das mitgelieferte. Als Jinja-Global registriert, damit auch Seiten ohne _ctx() (Login, Zentrale) dasselbe Logo zeigen. """ name = database.get_setting("logo_file", "") if name: # Kein Pfad, nur Dateiname - der Wert kommt aus der eigenen DB, aber # ein Traversal soll auch dann nicht moeglich sein. name = os.path.basename(name) if os.path.exists(os.path.join(CUSTOM_DIR, name)): return f"/static/custom/{name}" return DEFAULT_LOGO templates.env.globals["logo_url"] = _logo_url def _hash_password(pw): return hashlib.sha256(pw.encode()).hexdigest() def _is_logged_in(request): return request.cookies.get("session") == SESSION_TOKEN def _require_login(request): """Redirect to login if not authenticated. Returns RedirectResponse or None.""" if not _is_logged_in(request): # Check if a password is set admin_pw = database.get_setting("admin_password", "") if admin_pw: return RedirectResponse(f"/login?next={request.url.path}", status_code=303) return None def _ctx(request, page, **kwargs): """Build template context with common variables.""" authenticated = scheduler.graph is not None and scheduler.graph.is_authenticated() logged_in = _is_logged_in(request) return request, f"{page}.html", {"page": page, "authenticated": authenticated, "logged_in": logged_in, **kwargs} # ── Login ────────────────────────────────────────────────────────────────── @app.get("/login", response_class=HTMLResponse) async def login_page(request: Request): next_url = request.query_params.get("next", "/") return templates.TemplateResponse(request, "login.html", { "page": "login", "next": next_url, "error": request.query_params.get("error"), }) @app.post("/login") async def login_submit( request: Request, password: str = Form(...), next: str = Form("/"), ): admin_pw = database.get_setting("admin_password", "") if admin_pw and _hash_password(password) == admin_pw: response = RedirectResponse(next, status_code=303) response.set_cookie("session", SESSION_TOKEN, httponly=True, max_age=86400) return response return RedirectResponse("/login?error=1&next=" + next, status_code=303) @app.get("/logout") async def logout(): response = RedirectResponse("/", status_code=303) response.delete_cookie("session") return response # ── Dashboard ────────────────────────────────────────────────────────────── @app.get("/", response_class=HTMLResponse) async def dashboard(request: Request): conn = database.get_db() devices = conn.execute(""" SELECT d.id, d.hostname, d.name, d.segment_count FROM wled_devices d ORDER BY d.name, d.hostname """).fetchall() device_data = [] for dev in devices: segments = [] for seg_idx in range(dev["segment_count"]): row = conn.execute(""" SELECT u.display_name, u.azure_id, s.last_attendance_status, s.last_teams_status, COALESCE(s.phone_active, 0) as phone_active, cr_att.r as ar, cr_att.g as ag, cr_att.b as ab, cr_teams.r as tr, cr_teams.g as tg, cr_teams.b as tb FROM assignments a JOIN users u ON u.azure_id = a.user_azure_id LEFT JOIN state s ON s.user_azure_id = u.azure_id LEFT JOIN color_rules cr_att ON cr_att.status = s.last_attendance_status LEFT JOIN color_rules cr_teams ON cr_teams.status = s.last_teams_status WHERE a.wled_device_id = ? AND a.segment = ? """, (dev["id"], seg_idx)).fetchone() if row: phone = row["phone_active"] if phone: pc = conn.execute("SELECT r,g,b FROM color_rules WHERE status='phone_active'").fetchone() r, g, b = (pc["r"], pc["g"], pc["b"]) if pc else (255, 0, 0) elif row["last_attendance_status"] == "anwesend" and row["tr"] is not None: r, g, b = row["tr"], row["tg"], row["tb"] elif row["ar"] is not None: r, g, b = row["ar"], row["ag"], row["ab"] else: r, g, b = 128, 128, 128 segments.append({ "index": seg_idx, "user": row["display_name"], "attendance": row["last_attendance_status"] or "-", "teams": row["last_teams_status"] or "-", "phone": phone, "r": r, "g": g, "b": b, }) else: segments.append({ "index": seg_idx, "user": None, "attendance": "-", "teams": "-", "r": 128, "g": 128, "b": 128, }) device_data.append({ "id": dev["id"], "hostname": dev["hostname"], "name": dev["name"] or dev["hostname"], "segments": segments, }) conn.close() return templates.TemplateResponse(*_ctx(request, "dashboard", devices=device_data)) # ── Users ────────────────────────────────────────────────────────────────── @app.get("/users", response_class=HTMLResponse) async def users_page(request: Request): redir = _require_login(request) if redir: return redir conn = database.get_db() users = conn.execute(""" SELECT u.*, a.id as assignment_id FROM users u LEFT JOIN assignments a ON a.user_azure_id = u.azure_id ORDER BY u.active DESC, u.display_name """).fetchall() conn.close() return templates.TemplateResponse(*_ctx(request, "users", users=users)) @app.post("/api/users/sync") async def sync_users(): if scheduler.graph is None or not scheduler.graph.is_authenticated(): return RedirectResponse("/users?error=not_authenticated", status_code=303) azure_users = scheduler.graph.get_users() conn = database.get_db() count = 0 for au in azure_users: existing = conn.execute( "SELECT azure_id FROM users WHERE azure_id = ?", (au["id"],) ).fetchone() if not existing: conn.execute( "INSERT INTO users (azure_id, display_name, email) VALUES (?, ?, ?)", (au["id"], au.get("displayName", ""), au.get("mail", "")), ) count += 1 else: conn.execute( "UPDATE users SET display_name=?, email=? WHERE azure_id=?", (au.get("displayName", ""), au.get("mail", ""), au["id"]), ) conn.commit() conn.close() return RedirectResponse(f"/users?synced={count}", status_code=303) @app.post("/api/users/{azure_id}/update") async def update_user( azure_id: str, personal_nr: str = Form(""), phone_extension: str = Form(""), active: str = Form("0"), ): conn = database.get_db() conn.execute( "UPDATE users SET personal_nr=?, phone_extension=?, active=? WHERE azure_id=?", (personal_nr, phone_extension, 1 if active == "1" else 0, azure_id), ) conn.commit() conn.close() return RedirectResponse("/users", status_code=303) @app.post("/api/users/{azure_id}/toggle") async def toggle_user(azure_id: str): conn = database.get_db() conn.execute( "UPDATE users SET active = CASE WHEN active=1 THEN 0 ELSE 1 END WHERE azure_id=?", (azure_id,), ) conn.commit() conn.close() return RedirectResponse("/users", status_code=303) # ── Devices ──────────────────────────────────────────────────────────────── @app.get("/devices", response_class=HTMLResponse) async def devices_page(request: Request): redir = _require_login(request) if redir: return redir conn = database.get_db() devices_raw = conn.execute( "SELECT * FROM wled_devices ORDER BY name, hostname" ).fetchall() # Letzte Aktivitaet pro Device aus activity_log devices = [] for dev in devices_raw: last_act = conn.execute(""" SELECT MAX(l.timestamp) as la FROM activity_log l WHERE l.wled_url LIKE '%' || ? || '%' """, (dev["hostname"],)).fetchone() d = dict(dev) d["last_activity"] = last_act["la"] if last_act and last_act["la"] else None # Segment assignments for this device d["segments"] = conn.execute(""" SELECT a.segment, u.display_name FROM assignments a JOIN users u ON u.azure_id = a.user_azure_id WHERE a.wled_device_id = ? ORDER BY a.segment """, (dev["id"],)).fetchall() devices.append(d) last_check_row = conn.execute( "SELECT MAX(last_check) as lc FROM wled_devices WHERE last_check IS NOT NULL" ).fetchone() conn.close() last_check = last_check_row["lc"] if last_check_row and last_check_row["lc"] else None # Firmware files firmwares = [] if os.path.exists(FIRMWARE_DIR): for f in sorted(os.listdir(FIRMWARE_DIR), reverse=True): if f.endswith(".bin"): size = os.path.getsize(os.path.join(FIRMWARE_DIR, f)) / 1024 firmwares.append({"name": f, "size": f"{size:.0f} KB"}) return templates.TemplateResponse(*_ctx(request, "devices", devices=devices, last_check=last_check, firmwares=firmwares)) @app.post("/api/devices") async def add_device( hostname: str = Form(...), name: str = Form(""), segment_count: int = Form(2), ): conn = database.get_db() conn.execute( "INSERT OR IGNORE INTO wled_devices (hostname, name, segment_count) VALUES (?, ?, ?)", (hostname, name or hostname, segment_count), ) conn.commit() conn.close() return RedirectResponse("/devices", status_code=303) @app.post("/api/devices/{device_id}/update") async def update_device( device_id: int, hostname: str = Form(...), name: str = Form(""), segment_count: int = Form(2), ): conn = database.get_db() conn.execute( "UPDATE wled_devices SET hostname=?, name=?, segment_count=? WHERE id=?", (hostname, name, segment_count, device_id), ) conn.commit() conn.close() return RedirectResponse("/devices", status_code=303) @app.post("/api/devices/{device_id}/delete") async def delete_device(device_id: int): conn = database.get_db() conn.execute("DELETE FROM wled_devices WHERE id=?", (device_id,)) conn.commit() conn.close() return RedirectResponse("/devices", status_code=303) @app.post("/api/devices/check-all") async def check_all_devices(): scheduler.check_devices() return RedirectResponse("/devices", status_code=303) @app.post("/api/devices/{device_id}/check") async def check_single_device(device_id: int): from wled_client import get_wled_info conn = database.get_db() dev = conn.execute("SELECT hostname FROM wled_devices WHERE id=?", (device_id,)).fetchone() if dev: from datetime import datetime info = get_wled_info(dev["hostname"]) if info.get("online"): conn.execute(""" UPDATE wled_devices SET online=1, leds=?, version=?, mac=?, last_check=? WHERE id=? """, (info.get("leds", 0), info.get("version", ""), info.get("mac", ""), datetime.now(), device_id)) else: conn.execute(""" UPDATE wled_devices SET online=0, last_check=? WHERE id=? """, (datetime.now(), device_id)) conn.commit() conn.close() return RedirectResponse("/devices", status_code=303) # ── Assignments ──────────────────────────────────────────────────────────── @app.get("/assignments", response_class=HTMLResponse) async def assignments_page(request: Request): redir = _require_login(request) if redir: return redir conn = database.get_db() assignments = conn.execute(""" SELECT a.id, a.segment, u.display_name, u.azure_id, d.hostname, d.name as device_name, d.id as device_id FROM assignments a JOIN users u ON u.azure_id = a.user_azure_id JOIN wled_devices d ON d.id = a.wled_device_id ORDER BY d.name, a.segment """).fetchall() users = conn.execute( "SELECT azure_id, display_name FROM users WHERE active=1 ORDER BY display_name" ).fetchall() devices = conn.execute( "SELECT id, hostname, name, segment_count FROM wled_devices ORDER BY name" ).fetchall() conn.close() return templates.TemplateResponse(*_ctx(request, "assignments", assignments=assignments, users=users, devices=devices)) @app.post("/api/assignments") async def create_assignment( user_azure_id: str = Form(...), wled_device_id: int = Form(...), segment: int = Form(0), ): conn = database.get_db() # Remove existing assignment for this user conn.execute("DELETE FROM assignments WHERE user_azure_id=?", (user_azure_id,)) conn.execute( "INSERT OR REPLACE INTO assignments (user_azure_id, wled_device_id, segment) VALUES (?, ?, ?)", (user_azure_id, wled_device_id, segment), ) # Eine ausdrueckliche Admin-Zuordnung setzt den fuer die Zentrale gemerkten # Heimatplatz ausser Kraft, sonst wuerde die Rueckkehr auf einen veralteten # Platz zeigen. conn.execute("DELETE FROM home_assignments WHERE user_azure_id=?", (user_azure_id,)) conn.commit() conn.close() return RedirectResponse("/assignments", status_code=303) @app.post("/api/assignments/{assignment_id}/delete") async def delete_assignment(assignment_id: int): conn = database.get_db() # Der fuer die Zentrale gemerkte Heimatplatz bleibt hier absichtlich stehen: # wer von der Zentrale entfernt wird, soll noch zurueckkehren koennen. conn.execute("DELETE FROM assignments WHERE id=?", (assignment_id,)) conn.commit() conn.close() return RedirectResponse("/assignments", status_code=303) # ── Zentrale (Self-Service per Durchwahl) ────────────────────── def _selfservice_target(conn): """Konfiguriertes Zentral-Geraet + Segment, oder None wenn nicht eingerichtet.""" raw_id = database.get_setting("selfservice_device_id", "") if not raw_id: return None try: device_id = int(raw_id) except (TypeError, ValueError): return None dev = conn.execute( "SELECT id, hostname, name FROM wled_devices WHERE id = ?", (device_id,) ).fetchone() if not dev: return None try: segment = int(database.get_setting("selfservice_segment", "0")) except (TypeError, ValueError): segment = 0 return { "id": dev["id"], "hostname": dev["hostname"], "name": dev["name"] or dev["hostname"], "segment": segment, } def _user_by_extension(extension, conn): """Aktiven Benutzer anhand der Telefon-Durchwahl finden.""" ext = (extension or "").strip() if not ext: return None return conn.execute(""" SELECT azure_id, display_name, phone_extension FROM users WHERE active = 1 AND TRIM(phone_extension) = ? """, (ext,)).fetchone() def _zentrale_owner_row(conn, target): """Heimateintrag, der auf das Zentral-Segment zeigt - der Stamm-Besetzer.""" if not target: return None return conn.execute(""" SELECT h.user_azure_id, u.display_name FROM home_assignments h JOIN users u ON u.azure_id = h.user_azure_id WHERE h.wled_device_id = ? AND h.segment = ? AND u.active = 1 AND NOT EXISTS (SELECT 1 FROM assignments a WHERE a.user_azure_id = h.user_azure_id) ORDER BY h.created_at, h.user_azure_id LIMIT 1 """, (target["id"], target["segment"])).fetchone() def _restore_zentrale_owner(conn, target): """Stamm-Besetzer zurueckholen, sobald das Zentral-Segment frei ist. Als Stamm-Besetzer gilt, fuer wen das Zentral-Segment als Heimatplatz gemerkt ist. Sein Eintrag bleibt dabei absichtlich stehen: die Zentrale ist sein normaler Platz, nicht eine Vertretung, aus der er zurueckkehrt. """ if not target: return None busy = conn.execute( "SELECT 1 FROM assignments WHERE wled_device_id = ? AND segment = ?", (target["id"], target["segment"]), ).fetchone() if busy: return None owner = _zentrale_owner_row(conn, target) if not owner: return None conn.execute(""" INSERT INTO assignments (user_azure_id, wled_device_id, segment) VALUES (?, ?, ?) """, (owner["user_azure_id"], target["id"], target["segment"])) return owner def _zentrale_state(conn, target, user): """Alles, was die Self-Service-Seite ueber die aktuelle Lage wissen muss. Wird bei jedem GET frisch berechnet. Die Aktionen leiten deshalb nach dem Absenden auf /zentrale/{durchwahl} um: die Seite zeigt dann den echten Zustand und nicht das Ergebnis eines abgeschickten Formulars. """ st = { "occupant": None, "is_here": False, "home": None, "home_blocker": None, "is_owner": False, "owner_waiting": None, "occupant_is_owner": False, "occupant_has_home": False, } if not target: return st st["occupant"] = conn.execute(""" SELECT u.azure_id, u.display_name FROM assignments a JOIN users u ON u.azure_id = a.user_azure_id WHERE a.wled_device_id = ? AND a.segment = ? """, (target["id"], target["segment"])).fetchone() owner = _zentrale_owner_row(conn, target) if owner and (not st["occupant"] or owner["user_azure_id"] != st["occupant"]["azure_id"]): st["owner_waiting"] = owner["display_name"] # Sitzt der aktuelle Besetzer auf seinem Stammplatz? Dann kommt er nach einer # Uebernahme von allein zurueck - das gehoert auf den Knopf geschrieben. if st["occupant"]: occ_home = conn.execute(""" SELECT wled_device_id, segment FROM home_assignments WHERE user_azure_id = ? """, (st["occupant"]["azure_id"],)).fetchone() st["occupant_has_home"] = occ_home is not None st["occupant_is_owner"] = bool(occ_home) and \ (occ_home["wled_device_id"], occ_home["segment"]) == (target["id"], target["segment"]) if not user: return st st["is_here"] = bool(st["occupant"] and st["occupant"]["azure_id"] == user["azure_id"]) st["home"] = conn.execute(""" SELECT h.wled_device_id, h.segment, d.name, d.hostname FROM home_assignments h JOIN wled_devices d ON d.id = h.wled_device_id WHERE h.user_azure_id = ? """, (user["azure_id"],)).fetchone() if st["home"]: st["is_owner"] = (st["home"]["wled_device_id"], st["home"]["segment"]) \ == (target["id"], target["segment"]) if not st["is_owner"]: blocker = conn.execute(""" SELECT u.display_name FROM assignments a JOIN users u ON u.azure_id = a.user_azure_id WHERE a.wled_device_id = ? AND a.segment = ? AND a.user_azure_id != ? """, (st["home"]["wled_device_id"], st["home"]["segment"], user["azure_id"])).fetchone() st["home_blocker"] = blocker["display_name"] if blocker else None return st def _zentrale_page(request, conn, target, extension="", user=None, done=None): """Self-Service-Seite rendern.""" st = _zentrale_state(conn, target, user) return templates.TemplateResponse(request, "zentrale.html", { "page": "zentrale", "target": target, "extension": extension, "user": user, "done": done, **st, }) def _zentrale_redirect(extension, done): """Nach jeder Aktion zurueck auf /zentrale (POST/Redirect/GET). Es gibt absichtlich nur diese eine Adresse: wer die Zentrale freigeben will, soll sich keine zweite URL gemerkt haben muessen. """ ext = quote((extension or "").strip(), safe="") if not ext: return RedirectResponse(f"/zentrale?done={done}", status_code=303) return RedirectResponse(f"/zentrale?durchwahl={ext}&done={done}", status_code=303) @app.get("/zentrale", response_class=HTMLResponse) async def zentrale_page(request: Request): """Self-Service-Seite fuer den Empfangsbereich. Absichtlich ohne Admin-Login. Die eine Adresse fuer alles: Durchwahl eintragen, und die Seite zeigt die Knoepfe, die fuer diese Person gerade gelten - Uebernehmen oder Freigeben. """ typed = (request.query_params.get("durchwahl") or "").strip() done = request.query_params.get("done") conn = database.get_db() try: target = _selfservice_target(conn) user = _user_by_extension(typed, conn) if typed else None if typed and not user and not done: done = "err_unbekannt" return _zentrale_page(request, conn, target, extension=typed, user=user, done=done) finally: conn.close() @app.get("/zentrale/{extension}") async def zentrale_page_for_extension(extension: str): """Kurzform fuer QR-Codes. Leitet auf die eine Adresse um.""" return RedirectResponse(f"/zentrale?durchwahl={quote(extension, safe='')}", status_code=303) @app.post("/api/zentrale/uebernehmen") async def zentrale_takeover(extension: str = Form(...)): """Mitarbeiter setzt sich selbst auf das Busylight der Zentrale.""" conn = database.get_db() try: target = _selfservice_target(conn) if not target: return _zentrale_redirect(extension, "err_nicht_eingerichtet") user = _user_by_extension(extension, conn) if not user: return _zentrale_redirect(extension, "err_unbekannt") already = conn.execute(""" SELECT 1 FROM assignments WHERE user_azure_id = ? AND wled_device_id = ? AND segment = ? """, (user["azure_id"], target["id"], target["segment"])).fetchone() if already: scheduler.push_user_light(user["azure_id"], user["display_name"], target["hostname"], target["segment"], conn) return _zentrale_redirect(extension, "bereits") # Bisheriger Platz des Benutzers wird als Heimat-Tuerschild gemerkt own = conn.execute(""" SELECT a.wled_device_id, a.segment, d.hostname, d.name FROM assignments a JOIN wled_devices d ON d.id = a.wled_device_id WHERE a.user_azure_id = ? """, (user["azure_id"],)).fetchone() predecessor = conn.execute(""" SELECT u.azure_id, u.display_name FROM assignments a JOIN users u ON u.azure_id = a.user_azure_id WHERE a.wled_device_id = ? AND a.segment = ? """, (target["id"], target["segment"])).fetchone() if own: conn.execute(""" INSERT OR REPLACE INTO home_assignments (user_azure_id, wled_device_id, segment) VALUES (?, ?, ?) """, (user["azure_id"], own["wled_device_id"], own["segment"])) # Zentral-Segment freimachen, eigene Zuordnung loesen, neu eintragen. # Beide DELETEs sind noetig: UNIQUE(wled_device_id, segment) und UNIQUE(user_azure_id). conn.execute("DELETE FROM assignments WHERE wled_device_id = ? AND segment = ?", (target["id"], target["segment"])) conn.execute("DELETE FROM assignments WHERE user_azure_id = ?", (user["azure_id"],)) conn.execute(""" INSERT INTO assignments (user_azure_id, wled_device_id, segment) VALUES (?, ?, ?) """, (user["azure_id"], target["id"], target["segment"])) # Vorgaenger automatisch auf sein Tuerschild zurueckschicken. Ohne das muesste # er erst selbst "zurueck" klicken, bevor die Uebernahme fuer ihn stimmt - und # bis dahin haette er gar kein Licht. pred_back = None if predecessor and predecessor["azure_id"] != user["azure_id"]: pred_home = conn.execute(""" SELECT h.wled_device_id, h.segment, d.hostname, d.name FROM home_assignments h JOIN wled_devices d ON d.id = h.wled_device_id WHERE h.user_azure_id = ? """, (predecessor["azure_id"],)).fetchone() is_owner = pred_home and (pred_home["wled_device_id"], pred_home["segment"]) \ == (target["id"], target["segment"]) if is_owner: # Sein Stammplatz IST die Zentrale, die jetzt der Nachfolger hat. # Der Heimateintrag bleibt stehen, damit er zurueckkommt, sobald # die Zentrale wieder frei wird. pass elif pred_home: taken = conn.execute(""" SELECT 1 FROM assignments WHERE wled_device_id = ? AND segment = ? """, (pred_home["wled_device_id"], pred_home["segment"])).fetchone() if not taken: conn.execute(""" INSERT INTO assignments (user_azure_id, wled_device_id, segment) VALUES (?, ?, ?) """, (predecessor["azure_id"], pred_home["wled_device_id"], pred_home["segment"])) conn.execute("DELETE FROM home_assignments WHERE user_azure_id = ?", (predecessor["azure_id"],)) pred_back = pred_home else: # Kein anderer Platz gemerkt: die Zentrale ist sein Stammplatz. # Merken, damit er automatisch zurueckkommt, wenn sie frei wird. claimed = conn.execute(""" SELECT 1 FROM home_assignments WHERE wled_device_id = ? AND segment = ? AND user_azure_id != ? """, (target["id"], target["segment"], predecessor["azure_id"])).fetchone() if not claimed: conn.execute(""" INSERT OR REPLACE INTO home_assignments (user_azure_id, wled_device_id, segment) VALUES (?, ?, ?) """, (predecessor["azure_id"], target["id"], target["segment"])) conn.commit() # Licht sofort setzen - das Polling wuerde bei unveraendertem Status nichts senden scheduler.push_user_light(user["azure_id"], user["display_name"], target["hostname"], target["segment"], conn) # Das eigene Segment nur ausschalten, wenn nicht gerade der Vorgaenger # dorthin zurueckgekehrt ist - sonst wuerde sein Licht sofort wieder ausgehen. if own and not (pred_back and pred_back["wled_device_id"] == own["wled_device_id"] and pred_back["segment"] == own["segment"]): scheduler.clear_segment(own["hostname"], own["segment"], conn) if pred_back: scheduler.push_user_light(predecessor["azure_id"], predecessor["display_name"], pred_back["hostname"], pred_back["segment"], conn) logger.info("Zentrale uebernommen von %s (Durchwahl %s), Vorgaenger: %s", user["display_name"], extension, predecessor["display_name"] if predecessor else "-") return _zentrale_redirect(extension, "uebernommen") finally: conn.close() @app.post("/api/zentrale/verlassen") async def zentrale_release(extension: str = Form(...)): """Mitarbeiter verlaesst die Zentrale und geht auf sein Heimat-Tuerschild zurueck.""" conn = database.get_db() try: target = _selfservice_target(conn) user = _user_by_extension(extension, conn) if not user: return _zentrale_redirect(extension, "err_unbekannt") was_here = bool(target) and conn.execute(""" SELECT 1 FROM assignments WHERE user_azure_id = ? AND wled_device_id = ? AND segment = ? """, (user["azure_id"], target["id"], target["segment"])).fetchone() is not None home = conn.execute(""" SELECT h.wled_device_id, h.segment, d.hostname, d.name FROM home_assignments h JOIN wled_devices d ON d.id = h.wled_device_id WHERE h.user_azure_id = ? """, (user["azure_id"],)).fetchone() if not home: if not was_here: return _zentrale_redirect(extension, "err_nichts_zu_tun") # Nur von der Zentrale abmelden, es gibt keinen Platz zum Zurueckgehen conn.execute("DELETE FROM assignments WHERE user_azure_id = ?", (user["azure_id"],)) returning = _restore_zentrale_owner(conn, target) conn.commit() if returning: scheduler.push_user_light(returning["user_azure_id"], returning["display_name"], target["hostname"], target["segment"], conn) else: scheduler.clear_segment(target["hostname"], target["segment"], conn) logger.info("Zentrale verlassen von %s (kein Heimat-Tuerschild), zurueck: %s", user["display_name"], returning["display_name"] if returning else "-") return _zentrale_redirect(extension, "abgemeldet") # Der Stamm-Besetzer hat keinen anderen Platz - sein Heimatplatz IST die Zentrale if target and (home["wled_device_id"], home["segment"]) == (target["id"], target["segment"]): return _zentrale_redirect(extension, "stammplatz_hier" if was_here else "err_stammplatz_fremd") # Ist der Heimatplatz inzwischen von jemand anderem belegt? blocker = conn.execute(""" SELECT 1 FROM assignments WHERE wled_device_id = ? AND segment = ? AND user_azure_id != ? """, (home["wled_device_id"], home["segment"], user["azure_id"])).fetchone() if blocker: return _zentrale_redirect(extension, "err_heimat_belegt") conn.execute("DELETE FROM assignments WHERE user_azure_id = ?", (user["azure_id"],)) conn.execute(""" INSERT INTO assignments (user_azure_id, wled_device_id, segment) VALUES (?, ?, ?) """, (user["azure_id"], home["wled_device_id"], home["segment"])) conn.execute("DELETE FROM home_assignments WHERE user_azure_id = ?", (user["azure_id"],)) # Die Zentrale ist jetzt frei - ihr Stamm-Besetzer kommt zurueck. returning = _restore_zentrale_owner(conn, target) if was_here else None conn.commit() scheduler.push_user_light(user["azure_id"], user["display_name"], home["hostname"], home["segment"], conn) if was_here: if returning: scheduler.push_user_light(returning["user_azure_id"], returning["display_name"], target["hostname"], target["segment"], conn) else: scheduler.clear_segment(target["hostname"], target["segment"], conn) logger.info("Zentrale verlassen von %s, zurueck auf %s Segment %s; Zentrale nun: %s", user["display_name"], home["hostname"], home["segment"], returning["display_name"] if returning else "frei") return _zentrale_redirect(extension, "zurueck") finally: conn.close() # ── Color Rules ──────────────────────────────────────────────────────────── @app.get("/colors", response_class=HTMLResponse) async def colors_page(request: Request): redir = _require_login(request) if redir: return redir conn = database.get_db() rules = conn.execute("SELECT * FROM color_rules ORDER BY status").fetchall() conn.close() return templates.TemplateResponse(*_ctx(request, "colors", rules=rules)) @app.post("/api/colors") async def update_colors(request: Request): form = await request.form() conn = database.get_db() statuses = form.getlist("status") for status in statuses: color_hex = form.get(f"color_{status}", "#000000") # Convert hex to RGB color_hex = color_hex.lstrip("#") r = int(color_hex[0:2], 16) g = int(color_hex[2:4], 16) b = int(color_hex[4:6], 16) effect = form.get(f"effect_{status}", "") conn.execute( "UPDATE color_rules SET r=?, g=?, b=?, effect=? WHERE status=?", (r, g, b, effect, status), ) conn.commit() conn.close() return RedirectResponse("/colors?saved=1", status_code=303) @app.post("/api/colors/add") async def add_color( status: str = Form(...), color: str = Form("#00ff00"), effect: str = Form(""), ): color_hex = color.lstrip("#") r = int(color_hex[0:2], 16) g = int(color_hex[2:4], 16) b = int(color_hex[4:6], 16) conn = database.get_db() conn.execute( "INSERT OR REPLACE INTO color_rules (status, r, g, b, effect) VALUES (?, ?, ?, ?, ?)", (status, r, g, b, effect), ) conn.commit() conn.close() return RedirectResponse("/colors", status_code=303) # ── Settings ─────────────────────────────────────────────────────────────── @app.get("/settings", response_class=HTMLResponse) async def settings_page(request: Request): redir = _require_login(request) if redir: return redir settings = database.get_all_settings() conn = database.get_db() devices = conn.execute( "SELECT id, hostname, name, segment_count FROM wled_devices ORDER BY name" ).fetchall() conn.close() return templates.TemplateResponse(*_ctx(request, "settings", settings=settings, devices=devices)) @app.post("/api/settings") async def update_settings(request: Request): form = await request.form() keys = [ "azure_client_id", "azure_client_secret", "azure_tenant_id", "azure_auth_tenant", "azure_scopes", "poll_interval", "attendance_csv_path", "attendance_status_anwesend", "attendance_status_abwesend", "attendance_status_homeoffice", "selfservice_device_id", "selfservice_segment", ] for key in keys: value = form.get(key, "") database.set_setting(key, value) # Auth Tenant immer gleich Tenant ID setzen tenant_id = form.get("azure_tenant_id", "") if tenant_id: database.set_setting("azure_auth_tenant", tenant_id) # Admin password new_pw = form.get("admin_password", "") if new_pw: database.set_setting("admin_password", _hash_password(new_pw)) logger.info("Admin-Passwort geaendert") # Update scheduler interval try: interval = int(form.get("poll_interval", "15")) bg_scheduler.reschedule_job("poll", trigger="interval", seconds=interval) logger.info("Polling-Intervall geaendert auf %ds", interval) except Exception as e: logger.error("Fehler beim Aendern des Intervalls: %s", e) return RedirectResponse("/settings?saved=1", status_code=303) @app.post("/api/settings/logo") async def upload_logo(request: Request, logo: UploadFile = File(...)): """Kundenspezifisches Logo hochladen.""" redir = _require_login(request) if redir: return redir ext = os.path.splitext(logo.filename or "")[1].lower() if ext not in LOGO_EXTENSIONS: return RedirectResponse("/settings?logo_error=format", status_code=303) content = await logo.read() if not content: return RedirectResponse("/settings?logo_error=empty", status_code=303) if len(content) > LOGO_MAX_BYTES: return RedirectResponse("/settings?logo_error=size", status_code=303) os.makedirs(CUSTOM_DIR, exist_ok=True) # Nur ein Logo behalten, sonst bleiben alte Endungen liegen for old in os.listdir(CUSTOM_DIR): if old.startswith("logo."): try: os.remove(os.path.join(CUSTOM_DIR, old)) except OSError: pass filename = f"logo{ext}" with open(os.path.join(CUSTOM_DIR, filename), "wb") as f: f.write(content) database.set_setting("logo_file", filename) logger.info("Kundenlogo hochgeladen: %s (%d KB)", filename, len(content) // 1024) return RedirectResponse("/settings?logo_saved=1", status_code=303) @app.post("/api/settings/logo/reset") async def reset_logo(request: Request): """Zurueck auf das mitgelieferte Logo.""" redir = _require_login(request) if redir: return redir if os.path.isdir(CUSTOM_DIR): for old in os.listdir(CUSTOM_DIR): if old.startswith("logo."): try: os.remove(os.path.join(CUSTOM_DIR, old)) except OSError: pass database.set_setting("logo_file", "") logger.info("Kundenlogo entfernt, Standardlogo aktiv") return RedirectResponse("/settings?logo_reset=1", status_code=303) # ── Log ──────────────────────────────────────────────────────────────────── @app.get("/log", response_class=HTMLResponse) async def log_page(request: Request): redir = _require_login(request) if redir: return redir conn = database.get_db() logs = conn.execute( "SELECT * FROM activity_log ORDER BY id DESC LIMIT 200" ).fetchall() conn.close() return templates.TemplateResponse(*_ctx(request, "log", logs=logs)) # ── API Status (for dashboard auto-refresh) ─────────────────────────────── @app.get("/api/status") async def api_status(): conn = database.get_db() states = conn.execute(""" SELECT u.display_name, u.azure_id, s.last_attendance_status, s.last_teams_status, s.updated_at, d.hostname, a.segment FROM state s JOIN users u ON u.azure_id = s.user_azure_id LEFT JOIN assignments a ON a.user_azure_id = u.azure_id LEFT JOIN wled_devices d ON d.id = a.wled_device_id ORDER BY u.display_name """).fetchall() conn.close() return [dict(s) for s in states] @app.post("/api/poll") async def trigger_poll(): """Trigger a manual poll.""" scheduler.poll_status() return RedirectResponse("/", status_code=303) # ── Backup ───────────────────────────────────────────────────────────────── @app.get("/backup", response_class=HTMLResponse) async def backup_page(request: Request): redir = _require_login(request) if redir: return redir # Portal backups db_backups = [] if os.path.exists(BACKUP_DIR): for f in sorted(os.listdir(BACKUP_DIR), reverse=True): if f.startswith("busylight_") and f.endswith(".db"): path = os.path.join(BACKUP_DIR, f) size = os.path.getsize(path) / 1024 # Parse date from filename: busylight_YYYYMMDD_HHMMSS.db try: ts = f.replace("busylight_", "").replace(".db", "") dt = datetime.strptime(ts, "%Y%m%d_%H%M%S") date_display = dt.strftime("%d.%m.%Y %H:%M:%S") except ValueError: date_display = f db_backups.append({"name": f, "date": date_display, "size": f"{size:.0f} KB"}) # WLED config backups grouped by device wled_by_device = {} if os.path.exists(WLED_BACKUP_DIR): for f in sorted(os.listdir(WLED_BACKUP_DIR), reverse=True): if f.endswith(".json"): path = os.path.join(WLED_BACKUP_DIR, f) size = os.path.getsize(path) / 1024 # Extract device name and timestamp from filename # Format: hostname_parts_YYYYMMDD_HHMMSS.json parts = f.rsplit("_", 2) if len(parts) >= 3: device_key = parts[0] date_str = parts[1] time_str = parts[2].replace(".json", "") try: dt = datetime.strptime(date_str + time_str, "%Y%m%d%H%M%S") date_display = dt.strftime("%d.%m.%Y %H:%M:%S") except ValueError: date_display = f"{date_str} {time_str}" else: device_key = f.replace(".json", "") date_display = "-" hostname = device_key.replace("_", ".") if hostname not in wled_by_device: wled_by_device[hostname] = [] wled_by_device[hostname].append({ "name": f, "size": f"{size:.1f} KB", "date": date_display, }) conn = database.get_db() devices = conn.execute("SELECT id, hostname, name FROM wled_devices ORDER BY name, hostname").fetchall() conn.close() return templates.TemplateResponse(*_ctx(request, "backup", db_backups=db_backups, wled_by_device=wled_by_device, devices=devices)) @app.get("/api/backup/db/download") async def download_db(): """Download current database.""" ts = datetime.now().strftime("%Y%m%d_%H%M%S") return FileResponse( database.DB_PATH, media_type="application/octet-stream", filename=f"busylight_{ts}.db", ) @app.post("/api/backup/db/create") async def create_db_backup(): """Create a manual DB backup.""" auto_backup_db() return RedirectResponse("/backup?db_created=1", status_code=303) @app.get("/api/backup/db/{filename}") async def download_db_backup(filename: str): """Download a specific DB backup.""" path = os.path.join(BACKUP_DIR, filename) if not os.path.exists(path) or ".." in filename: return RedirectResponse("/backup", status_code=303) return FileResponse(path, media_type="application/octet-stream", filename=filename) @app.post("/api/backup/db/{filename}/delete") async def delete_db_backup(filename: str): path = os.path.join(BACKUP_DIR, filename) if os.path.exists(path) and ".." not in filename: os.remove(path) return RedirectResponse("/backup", status_code=303) @app.post("/api/backup/wled/all") async def backup_all_wled(): """Backup config of all WLED devices.""" from wled_client import get_wled_config conn = database.get_db() devices = conn.execute("SELECT id, hostname, name FROM wled_devices").fetchall() conn.close() ts = datetime.now().strftime("%Y%m%d_%H%M%S") count = 0 for dev in devices: config = get_wled_config(dev["hostname"]) if config: safe_name = dev["hostname"].replace(".", "_") path = os.path.join(WLED_BACKUP_DIR, f"{safe_name}_{ts}.json") with open(path, "w") as f: json.dump(config, f, indent=2) count += 1 return RedirectResponse(f"/backup?wled_backed_up={count}", status_code=303) @app.post("/api/backup/wled/{device_id}") async def backup_single_wled(device_id: int): """Backup config of a single WLED device.""" from wled_client import get_wled_config conn = database.get_db() dev = conn.execute("SELECT hostname FROM wled_devices WHERE id=?", (device_id,)).fetchone() conn.close() if not dev: return RedirectResponse("/backup", status_code=303) config = get_wled_config(dev["hostname"]) if config: ts = datetime.now().strftime("%Y%m%d_%H%M%S") safe_name = dev["hostname"].replace(".", "_") path = os.path.join(WLED_BACKUP_DIR, f"{safe_name}_{ts}.json") with open(path, "w") as f: json.dump(config, f, indent=2) return RedirectResponse("/backup?wled_backed_up=1", status_code=303) @app.get("/api/backup/wled/download/{filename}") async def download_wled_backup(filename: str): path = os.path.join(WLED_BACKUP_DIR, filename) if not os.path.exists(path) or ".." in filename: return RedirectResponse("/backup", status_code=303) return FileResponse(path, media_type="application/json", filename=filename) @app.post("/api/backup/wled/delete/{filename}") async def delete_wled_backup(filename: str): path = os.path.join(WLED_BACKUP_DIR, filename) if os.path.exists(path) and ".." not in filename: os.remove(path) return RedirectResponse("/backup", status_code=303) @app.post("/api/backup/wled/restore/{filename}") async def restore_wled_backup(filename: str, device_id: int = Form(...)): """Restore a WLED config backup to a device.""" from wled_client import restore_wled_config path = os.path.join(WLED_BACKUP_DIR, filename) if not os.path.exists(path) or ".." in filename: return RedirectResponse("/backup", status_code=303) conn = database.get_db() dev = conn.execute("SELECT hostname FROM wled_devices WHERE id=?", (device_id,)).fetchone() conn.close() if not dev: return RedirectResponse("/backup", status_code=303) with open(path) as f: config = json.load(f) success, msg = restore_wled_config(dev["hostname"], config) return RedirectResponse(f"/backup?restored={'1' if success else '0'}", status_code=303) # ── Firmware Update ───────────────────────────────────────────────────────── @app.post("/api/firmware/upload") async def upload_firmware(firmware: UploadFile = File(...)): """Upload a WLED firmware .bin file.""" if not firmware.filename.endswith(".bin"): return RedirectResponse("/devices?fw_error=format", status_code=303) path = os.path.join(FIRMWARE_DIR, firmware.filename) with open(path, "wb") as f: content = await firmware.read() f.write(content) logger.info("Firmware hochgeladen: %s (%d KB)", firmware.filename, len(content) // 1024) return RedirectResponse("/devices?fw_uploaded=1", status_code=303) @app.post("/api/firmware/{device_id}/update") async def ota_update_device(device_id: int, firmware_file: str = Form(...)): """OTA update a single WLED device. Backs up config first.""" from wled_client import get_wled_config, ota_update_wled conn = database.get_db() dev = conn.execute("SELECT hostname, name FROM wled_devices WHERE id=?", (device_id,)).fetchone() conn.close() if not dev: return RedirectResponse("/devices?fw_error=notfound", status_code=303) fw_path = os.path.join(FIRMWARE_DIR, firmware_file) if not os.path.exists(fw_path) or ".." in firmware_file: return RedirectResponse("/devices?fw_error=nofile", status_code=303) # Auto config backup before update config = get_wled_config(dev["hostname"]) if config: ts = datetime.now().strftime("%Y%m%d_%H%M%S") safe_name = dev["hostname"].replace(".", "_") backup_path = os.path.join(WLED_BACKUP_DIR, f"{safe_name}_{ts}_pre_update.json") with open(backup_path, "w") as f: json.dump(config, f, indent=2) logger.info("Pre-Update Backup: %s", backup_path) # Push firmware success, msg = ota_update_wled(dev["hostname"], fw_path) logger.info("OTA Update %s: %s - %s", dev["hostname"], "OK" if success else "FEHLER", msg) if success: return RedirectResponse(f"/devices?fw_updated={dev['hostname']}", status_code=303) else: return RedirectResponse(f"/devices?fw_error=failed", status_code=303) @app.post("/api/firmware/{filename}/delete") async def delete_firmware(filename: str): path = os.path.join(FIRMWARE_DIR, filename) if os.path.exists(path) and ".." not in filename: os.remove(path) return RedirectResponse("/devices", status_code=303) # ── Azure Auth (Web-basiert) ────────────────────────────────────────────── # Global state for device code flow _auth_state = {"status": "idle", "user_code": None, "verification_uri": None, "error": None} def _run_device_code_auth(): """Background task: run device code flow and save auth record.""" from azure.identity import DeviceCodeCredential, TokenCachePersistenceOptions from database import get_setting, set_setting client_id = get_setting("azure_client_id") tenant_id = get_setting("azure_auth_tenant", "common") scopes = get_setting("azure_scopes", "User.Read").split() if not client_id: _auth_state["status"] = "error" _auth_state["error"] = "Anwendungs-ID (Client) nicht konfiguriert" return def prompt_callback(verification_uri, user_code, expires_on): _auth_state["user_code"] = user_code _auth_state["verification_uri"] = verification_uri _auth_state["status"] = "waiting" try: cred = DeviceCodeCredential( client_id=client_id, tenant_id=tenant_id, cache_persistence_options=TokenCachePersistenceOptions( name="busylight", allow_unencrypted_storage=True ), prompt_callback=prompt_callback, ) record = cred.authenticate(tenant_id=tenant_id, scopes=scopes) set_setting("azure_auth_record", record.serialize()) _auth_state["status"] = "success" # Reinit graph client try: scheduler.graph = GraphAPI() logger.info("Graph-Client nach Web-Auth neu initialisiert") except Exception as e: logger.error("Graph-Client Reinit fehlgeschlagen: %s", e) except Exception as e: _auth_state["status"] = "error" _auth_state["error"] = str(e) logger.error("Web-Auth fehlgeschlagen: %s", e) @app.post("/api/azure/auth") async def start_azure_auth(): """Start device code flow in background.""" import threading _auth_state["status"] = "starting" _auth_state["user_code"] = None _auth_state["verification_uri"] = None _auth_state["error"] = None # Clear old auth record database.set_setting("azure_auth_record", "") thread = threading.Thread(target=_run_device_code_auth, daemon=True) thread.start() return JSONResponse({"status": "starting"}) @app.get("/api/azure/auth/status") async def azure_auth_status(): """Poll auth state from frontend.""" return JSONResponse(_auth_state) # ── CLI Auth Command ─────────────────────────────────────────────────────── def cli_auth(): """Run Azure device code authentication from terminal.""" database.init_db() # Clear old auth record to force fresh device code flow database.set_setting("azure_auth_record", "") print("Starte Azure Device Code Authentifizierung...") print("Bitte den angezeigten Code im Browser eingeben.\n") graph = GraphAPI() graph.authenticate() print("\nAuthentifizierung erfolgreich!") if __name__ == "__main__": import uvicorn import threading if len(sys.argv) > 1 and sys.argv[1] == "--auth": cli_auth() else: # Start webhook server on port 8081 in background thread from webhook import webhook_app def run_webhook(): uvicorn.run(webhook_app, host="0.0.0.0", port=8081, log_level="info") webhook_thread = threading.Thread(target=run_webhook, daemon=True) webhook_thread.start() logger.info("Webhook-Server gestartet auf Port 8081") # Main web UI on port 8080 uvicorn.run(app, host="0.0.0.0", port=8080)