update

backend/database.py

@@ -0,0 +1,70 @@
+import aiosqlite
+from config import DB_PATH
+
+CREATE_TABLE = """
+CREATE TABLE IF NOT EXISTS guides (
+    id TEXT PRIMARY KEY,
+    topic TEXT NOT NULL,
+    format TEXT NOT NULL,
+    status TEXT NOT NULL DEFAULT 'queued',
+    progress TEXT,
+    error_msg TEXT,
+    html_path TEXT,
+    pdf_path TEXT,
+    created_at TEXT NOT NULL,
+    updated_at TEXT NOT NULL
+)
+"""
+
+
+async def init_db():
+    async with aiosqlite.connect(DB_PATH) as db:
+        await db.execute(CREATE_TABLE)
+        await db.commit()
+
+
+def _row_to_dict(row, cursor):
+    columns = [d[0] for d in cursor.description]
+    return dict(zip(columns, row))
+
+
+async def create_guide(guide: dict) -> dict:
+    async with aiosqlite.connect(DB_PATH) as db:
+        await db.execute(
+            """INSERT INTO guides (id, topic, format, status, progress, html_path, pdf_path, created_at, updated_at)
+               VALUES (:id, :topic, :format, :status, :progress, :html_path, :pdf_path, :created_at, :updated_at)""",
+            guide,
+        )
+        await db.commit()
+    return guide
+
+
+async def get_guide(guide_id: str) -> dict | None:
+    async with aiosqlite.connect(DB_PATH) as db:
+        cursor = await db.execute("SELECT * FROM guides WHERE id = ?", (guide_id,))
+        row = await cursor.fetchone()
+        if row is None:
+            return None
+        return _row_to_dict(row, cursor)
+
+
+async def list_guides() -> list[dict]:
+    async with aiosqlite.connect(DB_PATH) as db:
+        cursor = await db.execute("SELECT * FROM guides ORDER BY created_at DESC")
+        rows = await cursor.fetchall()
+        return [_row_to_dict(row, cursor) for row in rows]
+
+
+async def update_guide(guide_id: str, **fields) -> None:
+    sets = ", ".join(f"{k} = :{k}" for k in fields)
+    fields["id"] = guide_id
+    async with aiosqlite.connect(DB_PATH) as db:
+        await db.execute(f"UPDATE guides SET {sets} WHERE id = :id", fields)
+        await db.commit()
+
+
+async def delete_guide(guide_id: str) -> bool:
+    async with aiosqlite.connect(DB_PATH) as db:
+        cursor = await db.execute("DELETE FROM guides WHERE id = ?", (guide_id,))
+        await db.commit()
+        return cursor.rowcount > 0