[verified] Add in-app automatic data scheduler (runs on its own server, no Hermes)

- scheduler.py: daemon thread inside Flask refreshes all real Thai collectors on interval (default 60min, REFRESH_INTERVAL_SECONDS) via shared daily cache + writes timestamped marker
- create_app starts scheduler (skipped in TESTING); shared daily_cache now an extension
- GET /api/v1/data/last-refresh: automation status + last refresh (every N hours)
- Live verified: refresh_all pulls 5/5 real sources (tourism/auto/NPL/energy/macro)
- 2 tests; full suite 197 OK
This commit is contained in:
Kunthawat Greethong
2026-08-25 21:00:13 +07:00
parent 4b6ce2c09e
commit abc06af3a1
3 changed files with 215 additions and 0 deletions

134
backend/app/scheduler.py Normal file
View File

@@ -0,0 +1,134 @@
"""App-internal background scheduler — refreshes real data on its own schedule.
This app runs on its own server (Docker/EasyPanel), independent of Hermes. So
the periodic data-fetching/analysis must live INSIDE the app. This module runs a
daemon thread inside the Flask process that:
- on an interval (default every 60 min), triggers each collector through the
shared daily cache (so fresh real data lands as the "current" state),
- records each successful refresh as a timestamped snapshot under the data dir
(persistent history the dashboard/sources table can report),
- logs progress; swallows transient failures (a source being down once must
not crash the app or the loop).
Thread-safety: the loop only calls collectors + writes cache/snapshots; the HTTP
layer reads the daily cache. No shared mutable in-process state beyond that.
"""
from __future__ import annotations
import logging
import threading
import time
from pathlib import Path
from typing import Callable, List, Optional
log = logging.getLogger("set50.scheduler")
# collectors returning a .to_dict()/dict, keyed by cache key
# (imported lazily to avoid import cycles at module load)
_REFRESH_JOBS: List[dict] = [
{"key": "bot_tourism", "label": "ท่องเที่ยว (BOT)", "module": "bot_tourism", "fn": "BotTourismSource().fetch"},
{"key": "auto_credit/tourism", "label": "ยอดขายรถ (TradingEconomics)", "module": "auto_credit", "fn": "fetch_auto_credit"},
{"key": "auto_npl", "label": "NPL รถยนต์ (BOT)", "module": "auto_npl", "fn": "fetch_auto_npl"},
{"key": "energy_thai", "label": "โรงกลั่น TOP", "module": "energy_thai", "fn": "fetch_energy_thai"},
{"key": "macro_thai", "label": "ภาพรวมประเทศไทย (BOT)", "module": "macro_thai", "fn": "fetch_macro_thai"},
]
class AppDataScheduler:
"""Runs periodic refresh of the real data collectors inside the app."""
def __init__(self, cache, data_root: Path, interval_seconds: int = 3600):
self.cache = cache
self.data_root = data_root
self.interval = max(30, int(interval_seconds)) # never faster than 30s
self._stop = threading.Event()
self._thread: Optional[threading.Thread] = None
self._snap_dir = data_root / "scheduler"
self._snap_dir.mkdir(parents=True, exist_ok=True)
# -- public lifecycle ---------------------------------------------------
def start(self) -> None:
if self._thread and self._thread.is_alive():
return
self._stop.clear()
self._thread = threading.Thread(target=self._loop, name="set50-refresh", daemon=True)
self._thread.start()
log.info("set50 data scheduler started (interval=%ss)", self.interval)
def stop(self) -> None:
self._stop.set()
# -- internals ----------------------------------------------------------
def _loop(self) -> None:
# run an immediate refresh shortly after boot, then on interval
while not self._stop.wait(self.interval):
try:
self.refresh_all()
except Exception: # noqa: BLE001 — loop must survive individual failures
log.exception("set50 scheduler refresh_all failed (will retry)")
def refresh_all(self) -> list[dict]:
"""Run every collector, warm the daily cache, and snapshot the state."""
results: list[dict] = []
for job in _REFRESH_JOBS:
results.append(self._run_job(job))
self._write_marker(results)
return results
def _run_job(self, job: dict) -> dict:
key = job["key"]
label = job["label"]
try:
value = self.cache.fetch_or_stale(key, self._make_fetcher(job))
return {"key": key, "label": label, "ok": True, "at": self._now()}
except Exception as exc: # noqa: BLE001
log.warning("set50 refresh failed for %s: %s", key, exc)
return {"key": key, "label": label, "ok": False, "error": str(exc), "at": self._now()}
def _make_fetcher(self, job: dict) -> Callable[[], dict]:
module_name = job["module"]
fn = job["fn"]
def fetcher() -> dict:
mod = __import__(f"app.{module_name}", fromlist=["*"])
obj = mod
# Support "ClassName().method" (instantiate then call) and plain
# "func_name" and static object attribute chains.
parts = fn.split(".")
for i, part in enumerate(parts):
is_instantiate = part.endswith("()")
attr = part[:-2] if is_instantiate else part
obj = getattr(obj, attr)
if is_instantiate:
obj = obj() # construct instance
if callable(obj):
value = obj()
else:
value = obj
if hasattr(value, "to_dict"):
value = value.to_dict()
return value or {}
return fetcher
def _write_marker(self, results: list[dict]) -> None:
import json
import datetime as _dt
ok = [r for r in results if r["ok"]]
marker = {
"at": self._now(),
"ok": len(ok),
"total": len(results),
"sources": results,
}
path = self._snap_dir / "last_refresh.json"
try:
path.write_text(json.dumps(marker, ensure_ascii=False, indent=2), encoding="utf-8")
except OSError:
log.exception("could not write scheduler marker")
@staticmethod
def _now() -> str:
import datetime as _dt
return _dt.datetime.now(_dt.timezone.utc).isoformat(timespec="seconds")