feat: add digest_queue table to schema and dev-api startup

This commit is contained in:
pyr0ball 2026-03-20 02:34:41 -07:00
parent c6f810fb30
commit 6a1ee3ed28
2 changed files with 26 additions and 0 deletions

View file

@ -53,6 +53,22 @@ def _strip_html(text: str | None) -> str | None:
return cleaned.strip() or None return cleaned.strip() or None
@app.on_event("startup")
def _startup():
"""Ensure digest_queue table exists (dev-api may run against an existing DB)."""
db = _get_db()
db.execute("""
CREATE TABLE IF NOT EXISTS digest_queue (
id INTEGER PRIMARY KEY,
job_contact_id INTEGER NOT NULL REFERENCES job_contacts(id),
created_at TEXT DEFAULT (datetime('now')),
UNIQUE(job_contact_id)
)
""")
db.commit()
db.close()
def _row_to_job(row) -> dict: def _row_to_job(row) -> dict:
d = dict(row) d = dict(row)
d["is_remote"] = bool(d.get("is_remote", 0)) d["is_remote"] = bool(d.get("is_remote", 0))

View file

@ -137,6 +137,15 @@ CREATE TABLE IF NOT EXISTS survey_responses (
); );
""" """
CREATE_DIGEST_QUEUE = """
CREATE TABLE IF NOT EXISTS digest_queue (
id INTEGER PRIMARY KEY,
job_contact_id INTEGER NOT NULL REFERENCES job_contacts(id),
created_at TEXT DEFAULT (datetime('now')),
UNIQUE(job_contact_id)
)
"""
_MIGRATIONS = [ _MIGRATIONS = [
("cover_letter", "TEXT"), ("cover_letter", "TEXT"),
("applied_at", "TEXT"), ("applied_at", "TEXT"),
@ -193,6 +202,7 @@ def init_db(db_path: Path = DEFAULT_DB) -> None:
conn.execute(CREATE_COMPANY_RESEARCH) conn.execute(CREATE_COMPANY_RESEARCH)
conn.execute(CREATE_BACKGROUND_TASKS) conn.execute(CREATE_BACKGROUND_TASKS)
conn.execute(CREATE_SURVEY_RESPONSES) conn.execute(CREATE_SURVEY_RESPONSES)
conn.execute(CREATE_DIGEST_QUEUE)
conn.commit() conn.commit()
conn.close() conn.close()
_migrate_db(db_path) _migrate_db(db_path)