first commit

app/db.py

@@ -0,0 +1,83 @@
+from __future__ import annotations
+
+import json
+import sqlite3
+from datetime import datetime
+from pathlib import Path
+
+from .config import DB_PATH
+
+
+def connect() -> sqlite3.Connection:
+    DB_PATH.parent.mkdir(parents=True, exist_ok=True)
+    conn = sqlite3.connect(DB_PATH)
+    conn.row_factory = sqlite3.Row
+    return conn
+
+
+def init_db() -> None:
+    conn = connect()
+    conn.execute(
+        """
+        CREATE TABLE IF NOT EXISTS analyses (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            title TEXT,
+            filename TEXT,
+            created_at TEXT,
+            step INTEGER DEFAULT 1,
+            parsed_path TEXT,
+            settings_json TEXT,
+            result_path TEXT
+        )
+        """
+    )
+    conn.commit()
+    conn.close()
+
+
+def create_analysis(title: str, filename: str, parsed_path: str) -> int:
+    conn = connect()
+    cur = conn.execute(
+        """
+        INSERT INTO analyses (title, filename, created_at, step, parsed_path)
+        VALUES (?, ?, ?, 1, ?)
+        """,
+        (title, filename, datetime.now().strftime("%Y-%m-%d %H:%M:%S"), parsed_path),
+    )
+    conn.commit()
+    aid = cur.lastrowid
+    conn.close()
+    return int(aid)
+
+
+def list_analyses() -> list[dict]:
+    conn = connect()
+    rows = conn.execute("SELECT * FROM analyses ORDER BY id DESC").fetchall()
+    conn.close()
+    return [dict(r) for r in rows]
+
+
+def get_analysis(aid: int) -> dict | None:
+    conn = connect()
+    row = conn.execute("SELECT * FROM analyses WHERE id=?", (aid,)).fetchone()
+    conn.close()
+    return dict(row) if row else None
+
+
+def update_analysis(aid: int, **fields) -> None:
+    if not fields:
+        return
+    keys = ", ".join(f"{k}=?" for k in fields)
+    conn = connect()
+    conn.execute(f"UPDATE analyses SET {keys} WHERE id=?", (*fields.values(), aid))
+    conn.commit()
+    conn.close()
+
+
+def load_json(path: str):
+    return json.loads(Path(path).read_text(encoding="utf-8"))
+
+
+def save_json(path: str, obj) -> None:
+    Path(path).parent.mkdir(parents=True, exist_ok=True)
+    Path(path).write_text(json.dumps(obj, ensure_ascii=False), encoding="utf-8")