Initial dashboard organization setup

backend/app/db.py

@@ -0,0 +1,75 @@
+from contextlib import contextmanager
+import time
+from typing import Iterator
+
+from psycopg.rows import dict_row
+import psycopg
+
+from .config import DATABASE_URL
+
+
+SCHEMA_SQL = """
+CREATE TABLE IF NOT EXISTS members (
+    id SERIAL PRIMARY KEY,
+    name TEXT NOT NULL,
+    company TEXT,
+    rank TEXT,
+    role TEXT,
+    department TEXT,
+    grp TEXT,
+    division TEXT,
+    team TEXT,
+    cell TEXT,
+    work_status TEXT,
+    work_time TEXT,
+    phone TEXT,
+    email TEXT,
+    seat_label TEXT,
+    photo_url TEXT,
+    sort_order INTEGER NOT NULL DEFAULT 0,
+    created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+    updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
+);
+
+CREATE TABLE IF NOT EXISTS seat_positions (
+    member_id INTEGER PRIMARY KEY REFERENCES members(id) ON DELETE CASCADE,
+    x INTEGER NOT NULL DEFAULT 0,
+    y INTEGER NOT NULL DEFAULT 0,
+    floor_label TEXT,
+    updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
+);
+
+CREATE TABLE IF NOT EXISTS snapshots (
+    id SERIAL PRIMARY KEY,
+    snapshot_month TEXT NOT NULL,
+    file_path TEXT NOT NULL,
+    created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
+);
+"""
+
+MIGRATION_SQL = """
+ALTER TABLE members ADD COLUMN IF NOT EXISTS sort_order INTEGER NOT NULL DEFAULT 0;
+"""
+
+
+@contextmanager
+def get_conn() -> Iterator[psycopg.Connection]:
+    with psycopg.connect(DATABASE_URL, row_factory=dict_row) as conn:
+        yield conn
+
+
+def init_db(max_retries: int = 20, retry_delay: float = 2.0) -> None:
+    last_error: Exception | None = None
+    for _ in range(max_retries):
+        try:
+            with get_conn() as conn:
+                with conn.cursor() as cur:
+                    cur.execute(SCHEMA_SQL)
+                    cur.execute(MIGRATION_SQL)
+                conn.commit()
+            return
+        except psycopg.OperationalError as exc:
+            last_error = exc
+            time.sleep(retry_delay)
+    if last_error is not None:
+        raise last_error