hermes
075b72ec6c
- FastAPI + APScheduler + SQLite, single container - Daily generation at 06:00 Europe/Copenhagen + cold-start generation - Retry with backoff (3 attempts/15 min), stale fallback with banner - OpenAI-compatible endpoint via env (OPENAI_BASE_URL/MODEL/API_KEY) - docker-compose with named volume for joke persistence
87 lines
2.1 KiB
Python
87 lines
2.1 KiB
Python
import sqlite3
|
|
from contextlib import contextmanager
|
|
from datetime import date
|
|
|
|
from .config import settings
|
|
|
|
SCHEMA = """
|
|
CREATE TABLE IF NOT EXISTS jokes (
|
|
day TEXT PRIMARY KEY,
|
|
article_title TEXT NOT NULL,
|
|
article_url TEXT NOT NULL,
|
|
jokes TEXT NOT NULL,
|
|
generated_at TEXT NOT NULL
|
|
);
|
|
"""
|
|
|
|
|
|
def init_db() -> None:
|
|
with _conn() as conn:
|
|
conn.executescript(SCHEMA)
|
|
|
|
|
|
@contextmanager
|
|
def _conn():
|
|
conn = sqlite3.connect(settings.db_path)
|
|
try:
|
|
yield conn
|
|
conn.commit()
|
|
finally:
|
|
conn.close()
|
|
|
|
|
|
def save_batch(day: date, article_title: str, article_url: str, jokes: list[str]) -> None:
|
|
import json
|
|
from datetime import datetime, timezone
|
|
|
|
with _conn() as conn:
|
|
conn.execute(
|
|
"INSERT OR REPLACE INTO jokes (day, article_title, article_url, jokes, generated_at) "
|
|
"VALUES (?, ?, ?, ?, ?)",
|
|
(
|
|
day.isoformat(),
|
|
article_title,
|
|
article_url,
|
|
json.dumps(jokes, ensure_ascii=False),
|
|
datetime.now(timezone.utc).isoformat(timespec="seconds"),
|
|
),
|
|
)
|
|
|
|
|
|
def get_day(day: date) -> dict | None:
|
|
import json
|
|
|
|
with _conn() as conn:
|
|
row = conn.execute(
|
|
"SELECT day, article_title, article_url, jokes, generated_at FROM jokes WHERE day = ?",
|
|
(day.isoformat(),),
|
|
).fetchone()
|
|
if row is None:
|
|
return None
|
|
return {
|
|
"day": row[0],
|
|
"article_title": row[1],
|
|
"article_url": row[2],
|
|
"jokes": json.loads(row[3]),
|
|
"generated_at": row[4],
|
|
}
|
|
|
|
|
|
def get_latest() -> dict | None:
|
|
import json
|
|
|
|
with _conn() as conn:
|
|
row = conn.execute(
|
|
"SELECT day, article_title, article_url, jokes, generated_at "
|
|
"FROM jokes ORDER BY day DESC LIMIT 1"
|
|
).fetchone()
|
|
if row is None:
|
|
return None
|
|
return {
|
|
"day": row[0],
|
|
"article_title": row[1],
|
|
"article_url": row[2],
|
|
"jokes": json.loads(row[3]),
|
|
"generated_at": row[4],
|
|
}
|