258 lines
7.4 KiB
Python
258 lines
7.4 KiB
Python
import json
|
|
|
|
import aiosqlite
|
|
from config import DB_PATH
|
|
|
|
CREATE_GUIDES = """
|
|
CREATE TABLE IF NOT EXISTS guides (
|
|
id TEXT PRIMARY KEY,
|
|
topic TEXT NOT NULL,
|
|
format TEXT NOT NULL,
|
|
instructions TEXT NOT NULL DEFAULT '',
|
|
status TEXT NOT NULL DEFAULT 'queued',
|
|
progress TEXT,
|
|
step INTEGER,
|
|
error_msg TEXT,
|
|
created_at TEXT NOT NULL,
|
|
updated_at TEXT NOT NULL
|
|
)
|
|
"""
|
|
|
|
CREATE_PROGRESS = """
|
|
CREATE TABLE IF NOT EXISTS guide_progress (
|
|
guide_id TEXT NOT NULL,
|
|
chapter TEXT NOT NULL,
|
|
created_at TEXT NOT NULL,
|
|
PRIMARY KEY (guide_id, chapter)
|
|
)
|
|
"""
|
|
|
|
CREATE_TOPICS = """
|
|
CREATE TABLE IF NOT EXISTS topics (
|
|
name TEXT PRIMARY KEY,
|
|
created_at TEXT NOT NULL
|
|
)
|
|
"""
|
|
|
|
CREATE_ELEMENTS = """
|
|
CREATE TABLE IF NOT EXISTS elements (
|
|
id TEXT PRIMARY KEY,
|
|
topic TEXT NOT NULL,
|
|
title TEXT NOT NULL,
|
|
description TEXT NOT NULL DEFAULT '',
|
|
examples TEXT NOT NULL DEFAULT '[]',
|
|
hints TEXT NOT NULL DEFAULT '[]',
|
|
created_at TEXT NOT NULL,
|
|
updated_at TEXT NOT NULL
|
|
)
|
|
"""
|
|
|
|
_db: aiosqlite.Connection | None = None
|
|
|
|
|
|
async def get_db() -> aiosqlite.Connection:
|
|
global _db
|
|
if _db is None:
|
|
_db = await aiosqlite.connect(DB_PATH)
|
|
_db.row_factory = None
|
|
return _db
|
|
|
|
|
|
async def init_db():
|
|
db = await get_db()
|
|
# WAL übersteht Crashes deutlich besser; busy_timeout fängt kurze Locks ab.
|
|
await db.execute("PRAGMA journal_mode=WAL")
|
|
await db.execute("PRAGMA busy_timeout=5000")
|
|
await db.execute(CREATE_GUIDES)
|
|
await db.execute(CREATE_PROGRESS)
|
|
await db.execute(CREATE_TOPICS)
|
|
await db.execute(CREATE_ELEMENTS)
|
|
try: # Migration für Bestands-DBs ohne step-Spalte
|
|
await db.execute("ALTER TABLE guides ADD COLUMN step INTEGER")
|
|
except aiosqlite.OperationalError:
|
|
pass
|
|
await db.execute(
|
|
"UPDATE guides SET status = 'error', progress = NULL, error_msg = 'Server-Neustart' "
|
|
"WHERE status IN ('queued', 'generating')"
|
|
)
|
|
await db.commit()
|
|
|
|
|
|
async def close_db():
|
|
global _db
|
|
if _db is not None:
|
|
await _db.close()
|
|
_db = None
|
|
|
|
|
|
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:
|
|
db = await get_db()
|
|
await db.execute(
|
|
"""INSERT INTO guides (id, topic, format, instructions, status, progress, created_at, updated_at)
|
|
VALUES (:id, :topic, :format, :instructions, :status, :progress, :created_at, :updated_at)""",
|
|
guide,
|
|
)
|
|
await db.commit()
|
|
return guide
|
|
|
|
|
|
async def get_guide(guide_id: str) -> dict | None:
|
|
db = await get_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]:
|
|
db = await get_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
|
|
db = await get_db()
|
|
await db.execute(f"UPDATE guides SET {sets} WHERE id = :id", fields)
|
|
await db.commit()
|
|
|
|
|
|
async def delete_guide(guide_id: str) -> bool:
|
|
db = await get_db()
|
|
cursor = await db.execute("DELETE FROM guides WHERE id = ?", (guide_id,))
|
|
await db.commit()
|
|
return cursor.rowcount > 0
|
|
|
|
|
|
# --- Themen ---
|
|
|
|
async def create_topic(name: str) -> None:
|
|
from datetime import datetime, timezone
|
|
db = await get_db()
|
|
await db.execute(
|
|
"INSERT OR IGNORE INTO topics (name, created_at) VALUES (?, ?)",
|
|
(name, datetime.now(timezone.utc).isoformat()),
|
|
)
|
|
await db.commit()
|
|
|
|
|
|
async def list_topics() -> list[str]:
|
|
db = await get_db()
|
|
cursor = await db.execute("SELECT name FROM topics ORDER BY created_at DESC")
|
|
rows = await cursor.fetchall()
|
|
return [row[0] for row in rows]
|
|
|
|
|
|
async def delete_topic(name: str) -> None:
|
|
db = await get_db()
|
|
await db.execute("DELETE FROM topics WHERE name = ?", (name,))
|
|
await db.commit()
|
|
|
|
|
|
# --- Elemente ---
|
|
|
|
def _element_row(row, cursor) -> dict:
|
|
el = _row_to_dict(row, cursor)
|
|
el["examples"] = json.loads(el["examples"] or "[]")
|
|
el["hints"] = json.loads(el["hints"] or "[]")
|
|
return el
|
|
|
|
|
|
async def create_element(element: dict) -> dict:
|
|
db = await get_db()
|
|
await db.execute(
|
|
"""INSERT INTO elements (id, topic, title, description, examples, hints, created_at, updated_at)
|
|
VALUES (:id, :topic, :title, :description, :examples, :hints, :created_at, :updated_at)""",
|
|
{**element, "examples": json.dumps(element["examples"], ensure_ascii=False),
|
|
"hints": json.dumps(element["hints"], ensure_ascii=False)},
|
|
)
|
|
await db.commit()
|
|
return element
|
|
|
|
|
|
async def list_elements(topic: str) -> list[dict]:
|
|
db = await get_db()
|
|
cursor = await db.execute(
|
|
"SELECT * FROM elements WHERE topic = ? ORDER BY updated_at DESC", (topic,)
|
|
)
|
|
rows = await cursor.fetchall()
|
|
return [_element_row(row, cursor) for row in rows]
|
|
|
|
|
|
async def get_element(element_id: str) -> dict | None:
|
|
db = await get_db()
|
|
cursor = await db.execute("SELECT * FROM elements WHERE id = ?", (element_id,))
|
|
row = await cursor.fetchone()
|
|
if row is None:
|
|
return None
|
|
return _element_row(row, cursor)
|
|
|
|
|
|
async def update_element(element_id: str, **fields) -> None:
|
|
for key in ("examples", "hints"):
|
|
if key in fields:
|
|
fields[key] = json.dumps(fields[key], ensure_ascii=False)
|
|
sets = ", ".join(f"{k} = :{k}" for k in fields)
|
|
fields["id"] = element_id
|
|
db = await get_db()
|
|
await db.execute(f"UPDATE elements SET {sets} WHERE id = :id", fields)
|
|
await db.commit()
|
|
|
|
|
|
async def delete_element(element_id: str) -> bool:
|
|
db = await get_db()
|
|
cursor = await db.execute("DELETE FROM elements WHERE id = ?", (element_id,))
|
|
await db.commit()
|
|
return cursor.rowcount > 0
|
|
|
|
|
|
# --- Kapitel-Fortschritt ---
|
|
|
|
async def list_progress_all() -> dict[str, set[str]]:
|
|
"""Kompletter Kapitel-Fortschritt in einem Query: guide_id → Kapitel-Titel."""
|
|
db = await get_db()
|
|
cursor = await db.execute("SELECT guide_id, chapter FROM guide_progress")
|
|
rows = await cursor.fetchall()
|
|
out: dict[str, set[str]] = {}
|
|
for guide_id, chapter in rows:
|
|
out.setdefault(guide_id, set()).add(chapter)
|
|
return out
|
|
|
|
|
|
async def list_progress(guide_id: str) -> list[str]:
|
|
db = await get_db()
|
|
cursor = await db.execute(
|
|
"SELECT chapter FROM guide_progress WHERE guide_id = ?", (guide_id,)
|
|
)
|
|
rows = await cursor.fetchall()
|
|
return [row[0] for row in rows]
|
|
|
|
|
|
async def set_progress(guide_id: str, chapter: str, done: bool) -> None:
|
|
from datetime import datetime, timezone
|
|
db = await get_db()
|
|
if done:
|
|
await db.execute(
|
|
"INSERT OR IGNORE INTO guide_progress (guide_id, chapter, created_at) VALUES (?, ?, ?)",
|
|
(guide_id, chapter, datetime.now(timezone.utc).isoformat()),
|
|
)
|
|
else:
|
|
await db.execute(
|
|
"DELETE FROM guide_progress WHERE guide_id = ? AND chapter = ?", (guide_id, chapter)
|
|
)
|
|
await db.commit()
|
|
|
|
|
|
async def delete_progress(guide_id: str) -> None:
|
|
db = await get_db()
|
|
await db.execute("DELETE FROM guide_progress WHERE guide_id = ?", (guide_id,))
|
|
await db.commit()
|