feat(db): add per-channel ZC-freq columns to events (+ migration)

2026-07-29 17:13:59 +00:00
parent 12401b917c
commit 248276d141
2 changed files with 61 additions and 0 deletions
@@ -0,0 +1,45 @@
+from __future__ import annotations
+import os, sys, sqlite3
+from pathlib import Path
+import pytest
+
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+from sfm.database import SeismoDb
+
+ZC_COLS = ["tran_zc_freq", "vert_zc_freq", "long_zc_freq", "mic_zc_freq",
+           "tran_zc_above_range", "vert_zc_above_range",
+           "long_zc_above_range", "mic_zc_above_range"]
+
+
+def test_fresh_db_has_zc_freq_columns(tmp_path: Path):
+    db = SeismoDb(tmp_path / "seismo_relay.db")
+    with sqlite3.connect(str(tmp_path / "seismo_relay.db")) as conn:
+        cols = {r[1] for r in conn.execute("PRAGMA table_info(events)").fetchall()}
+    for c in ZC_COLS:
+        assert c in cols, f"missing column {c}"
+
+
+def test_migration_adds_zc_freq_columns_to_legacy_db(tmp_path: Path):
+    db_path = tmp_path / "old.db"
+    # A "modern-minus-zc" events table (has blastware_* but not the zc columns).
+    with sqlite3.connect(str(db_path)) as conn:
+        conn.executescript("""
+            CREATE TABLE events (
+                id TEXT PRIMARY KEY, serial TEXT NOT NULL, waveform_key TEXT NOT NULL,
+                session_id TEXT, timestamp TEXT,
+                tran_ppv REAL, vert_ppv REAL, long_ppv REAL, peak_vector_sum REAL, mic_ppv REAL,
+                project TEXT, client TEXT, operator TEXT, sensor_location TEXT,
+                sample_rate INTEGER, record_type TEXT,
+                false_trigger INTEGER NOT NULL DEFAULT 0,
+                created_at TEXT NOT NULL DEFAULT (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
+                UNIQUE(serial, timestamp)
+            );
+            INSERT INTO events (id, serial, waveform_key, timestamp)
+              VALUES ('legacy-id', 'BE11529', '01110000', '2026-04-01T12:00:00');
+        """)
+    db = SeismoDb(db_path)              # triggers _migrate
+    rows = db.query_events(serial="BE11529")
+    assert len(rows) == 1
+    for c in ZC_COLS:
+        assert c in rows[0]            # present
+        assert rows[0][c] is None      # NULL for the legacy row