mirror of
https://github.com/smittix/intercept.git
synced 2026-06-18 18:39:47 -07:00
James Smith
74d5663f73
- busy_timeout so concurrent app+agent writers wait instead of raising - seed from _connect() so update-before-first-read can't drop the seed - regression tests: seed ordering, concurrent writer, default DB path Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
116 lines
3.7 KiB
Python
116 lines
3.7 KiB
Python
"""Unified TLE store.
|
|
|
|
Single source of truth for TLE data, shared by satellite tracking,
|
|
weather-satellite prediction, SSTV doppler, and the remote agent.
|
|
Backed by SQLite; seeded once from the static data/satellites.py.
|
|
|
|
Replaces three previous stores: routes/satellite._tle_cache (which
|
|
persisted by rewriting data/satellites.py at runtime),
|
|
utils/weather_sat_predict._tle_cache, and the agent's own download.
|
|
"""
|
|
|
|
import sqlite3
|
|
import threading
|
|
from pathlib import Path
|
|
|
|
from utils.logging import get_logger
|
|
|
|
logger = get_logger("intercept.tle_store")
|
|
|
|
_DB_PATH = Path(__file__).resolve().parent.parent / "instance" / "tle.db"
|
|
_lock = threading.Lock()
|
|
_cache: dict[str, tuple[str, str, str]] | None = None
|
|
|
|
|
|
def _connect() -> sqlite3.Connection:
|
|
_DB_PATH.parent.mkdir(parents=True, exist_ok=True)
|
|
conn = sqlite3.connect(str(_DB_PATH), check_same_thread=False, timeout=5)
|
|
conn.execute("PRAGMA journal_mode = WAL")
|
|
conn.execute("PRAGMA busy_timeout = 5000")
|
|
conn.execute(
|
|
"""
|
|
CREATE TABLE IF NOT EXISTS tle_entries (
|
|
key TEXT PRIMARY KEY,
|
|
name TEXT NOT NULL,
|
|
line1 TEXT NOT NULL,
|
|
line2 TEXT NOT NULL,
|
|
updated_at TEXT NOT NULL DEFAULT (datetime('now'))
|
|
)
|
|
"""
|
|
)
|
|
_seed_if_empty(conn)
|
|
return conn
|
|
|
|
|
|
def _seed_if_empty(conn: sqlite3.Connection) -> None:
|
|
count = conn.execute("SELECT COUNT(*) FROM tle_entries").fetchone()[0]
|
|
if count:
|
|
return
|
|
try:
|
|
from data.satellites import TLE_SATELLITES
|
|
except ImportError:
|
|
logger.warning("data/satellites.py unavailable; TLE store starts empty")
|
|
return
|
|
conn.executemany(
|
|
"INSERT OR REPLACE INTO tle_entries (key, name, line1, line2) VALUES (?, ?, ?, ?)",
|
|
[(key, name, l1, l2) for key, (name, l1, l2) in TLE_SATELLITES.items()],
|
|
)
|
|
conn.commit()
|
|
logger.info(f"Seeded TLE store with {len(TLE_SATELLITES)} entries")
|
|
|
|
|
|
def _load() -> dict[str, tuple[str, str, str]]:
|
|
global _cache
|
|
if _cache is None:
|
|
with _lock:
|
|
if _cache is None:
|
|
conn = _connect()
|
|
try:
|
|
rows = conn.execute("SELECT key, name, line1, line2 FROM tle_entries").fetchall()
|
|
# Single-statement assignment of the fully built dict —
|
|
# the DCL pattern is only safe because readers never see
|
|
# a partially populated cache
|
|
_cache = {key: (name, l1, l2) for key, name, l1, l2 in rows}
|
|
finally:
|
|
conn.close()
|
|
return _cache
|
|
|
|
|
|
def all_tles() -> dict[str, tuple[str, str, str]]:
|
|
"""Return all TLEs as {key: (name, line1, line2)}."""
|
|
return dict(_load())
|
|
|
|
|
|
def get_tle(key: str) -> tuple[str, str, str] | None:
|
|
"""Return (name, line1, line2) for a satellite key, or None."""
|
|
return _load().get(key)
|
|
|
|
|
|
def update(entries: dict[str, tuple[str, str, str]]) -> None:
|
|
"""Insert or replace TLE entries and refresh the cache."""
|
|
if not entries:
|
|
return
|
|
with _lock:
|
|
conn = _connect()
|
|
try:
|
|
conn.executemany(
|
|
"INSERT OR REPLACE INTO tle_entries (key, name, line1, line2, updated_at)"
|
|
" VALUES (?, ?, ?, ?, datetime('now'))",
|
|
[(key, name, l1, l2) for key, (name, l1, l2) in entries.items()],
|
|
)
|
|
conn.commit()
|
|
finally:
|
|
conn.close()
|
|
global _cache
|
|
_cache = None # rebuilt on next read
|
|
|
|
|
|
def _reset_for_tests() -> None:
|
|
"""Clear the in-memory cache so the next read hits the database.
|
|
|
|
Seeding only runs when the table is empty, so a reset never overwrites
|
|
entries written by a test.
|
|
"""
|
|
global _cache
|
|
_cache = None
|