Add files via upload

app.py

@@ -0,0 +1,131 @@
+from flask import Flask, request, jsonify, send_from_directory
+import sqlite3
+import datetime as dt
+import os
+
+app = Flask(__name__)
+DB_PATH = os.environ.get("DB_PATH", "events.db")
+
+
+def get_db():
+    conn = sqlite3.connect(DB_PATH)
+    conn.row_factory = sqlite3.Row
+    return conn
+
+
+@app.before_first_request
+def init_db():
+    conn = get_db()
+    conn.execute(
+        """
+        CREATE TABLE IF NOT EXISTS events (
+          id INTEGER PRIMARY KEY AUTOINCREMENT,
+          ts TEXT NOT NULL,
+          event TEXT NOT NULL,
+          door_name TEXT,
+          device_name TEXT,
+          actor_id TEXT,
+          actor_name TEXT,
+          auth_type TEXT,
+          result TEXT
+        )
+        """
+    )
+    conn.commit()
+    conn.close()
+
+
+@app.post("/unifi-access-webhook")
+def unifi_access_webhook():
+    payload = request.get_json(force=True, silent=True) or {}
+    event = payload.get("event")
+    data = payload.get("data", {})
+
+    if event != "access.door.unlock":
+        return "", 204
+
+    actor = data.get("actor", {})
+    location = data.get("location", {})
+    device = data.get("device", {})
+    obj = data.get("object", {})
+
+    ts = dt.datetime.utcnow().isoformat(timespec="seconds") + "Z"
+
+    conn = get_db()
+    conn.execute(
+        """
+        INSERT INTO events (ts, event, door_name, device_name,
+                            actor_id, actor_name, auth_type, result)
+        VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+        """,
+        (
+            ts,
+            event,
+            location.get("name"),
+            device.get("name"),
+            actor.get("id"),
+            actor.get("name"),
+            obj.get("authentication_type"),
+            obj.get("result"),
+        ),
+    )
+    conn.commit()
+    conn.close()
+
+    return "", 204
+
+
+@app.get("/api/first-badge-status")
+def first_badge_status():
+    date = request.args.get("date") or dt.date.today().isoformat()
+    cutoff = request.args.get("cutoff", "09:00")  # HH:MM
+    start = f"{date}T00:00:00Z"
+    end = f"{date}T23:59:59Z"
+
+    conn = get_db()
+    rows = conn.execute(
+        """
+        SELECT actor_name, actor_id, MIN(ts) AS first_ts
+        FROM events
+        WHERE event = 'access.door.unlock'
+          AND result = 'Access Granted'
+          AND ts BETWEEN ? AND ?
+          AND actor_id IS NOT NULL
+        GROUP BY actor_id, actor_name
+        ORDER BY first_ts
+        """,
+        (start, end),
+    ).fetchall()
+    conn.close()
+
+    result = []
+    for r in rows:
+        first_ts = r["first_ts"]
+        t = dt.datetime.fromisoformat(first_ts.replace("Z", "+00:00"))
+        badge_time_str = t.strftime("%H:%M")
+        on_time = badge_time_str <= cutoff
+        result.append(
+            {
+                "actor_name": r["actor_name"],
+                "actor_id": r["actor_id"],
+                "first_badge": first_ts,
+                "badge_time": badge_time_str,
+                "on_time": on_time,
+            }
+        )
+
+    return jsonify(result)
+
+
+@app.get("/")
+def index():
+    return send_from_directory("static", "index.html")
+
+
+@app.get("/static/<path:path>")
+def send_static(path):
+    return send_from_directory("static", path)
+
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 8000)))