From 7d9be91aabd67f205c9c80dc76fa419ecfc32135 Mon Sep 17 00:00:00 2001 From: pyr0ball Date: Fri, 20 Mar 2026 02:36:23 -0700 Subject: [PATCH] fix: fix indentation and add try/finally in digest startup --- dev-api.py | 22 ++++++++++++---------- scripts/db.py | 8 ++++---- 2 files changed, 16 insertions(+), 14 deletions(-) diff --git a/dev-api.py b/dev-api.py index 0ca974e..3ec042d 100644 --- a/dev-api.py +++ b/dev-api.py @@ -57,16 +57,18 @@ def _strip_html(text: str | None) -> str | None: 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() + try: + 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() + finally: + db.close() def _row_to_job(row) -> dict: diff --git a/scripts/db.py b/scripts/db.py index 197bb8d..4afbd77 100644 --- a/scripts/db.py +++ b/scripts/db.py @@ -139,10 +139,10 @@ 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) + id INTEGER PRIMARY KEY, + job_contact_id INTEGER NOT NULL REFERENCES job_contacts(id), + created_at TEXT DEFAULT (datetime('now')), + UNIQUE(job_contact_id) ) """