feat: implement real-time observation patch normalization and live temperature threshold visualization logic

2026-05-27 10:17:33 +08:00
parent e6a673e27d
commit 820dabfbf3
23 changed files with 709 additions and 29 deletions
@@ -0,0 +1,98 @@
+import threading
+
+import src.data_collection.settlement_sources as settlement_sources
+import src.database.db_manager as db_manager
+from src.data_collection.settlement_sources import SettlementSourceMixin
+
+
+class _FakeResponse:
+    content = b"{}"
+
+    def raise_for_status(self):
+        return None
+
+    def json(self):
+        return {
+            "records": {
+                "Station": [
+                    {
+                        "StationId": "466920",
+                        "StationName": "中央氣象署臺北站",
+                        "ObsTime": {"DateTime": "2026-05-27T10:00:00+08:00"},
+                        "WeatherElement": {
+                            "AirTemperature": "34.2",
+                            "RelativeHumidity": "60",
+                            "WindSpeed": "2.0",
+                            "WindDirection": "90",
+                            "DailyExtreme": {
+                                "DailyHigh": {
+                                    "TemperatureInfo": {
+                                        "AirTemperature": "34.2",
+                                        "Occurred_at": {
+                                            "DateTime": "2026-05-27T10:00:00+08:00"
+                                        },
+                                    }
+                                },
+                                "DailyLow": {
+                                    "TemperatureInfo": {"AirTemperature": "27.1"}
+                                },
+                            },
+                        },
+                    }
+                ]
+            }
+        }
+
+
+class _FakeOfficialIntradayRepo:
+    def __init__(self):
+        self.rows = [{"time": "09:50", "temp": 34.0}]
+        self.upserts = []
+
+    def upsert_point(self, **kwargs):
+        self.upserts.append(kwargs)
+        self.rows.append(
+            {
+                "time": kwargs["observation_time"],
+                "temp": kwargs["value"],
+            }
+        )
+
+    def load_points(self, **kwargs):
+        return list(self.rows)
+
+
+class _FakeDBManager:
+    def append_airport_obs(self, **kwargs):
+        return None
+
+
+class _FakeCollector(SettlementSourceMixin):
+    cwa_open_data_auth = "token"
+    timeout = 1
+    settlement_cache_ttl_sec = 0
+
+    def __init__(self):
+        self._settlement_cache = {}
+        self._settlement_cache_lock = threading.Lock()
+
+    def _http_get(self, *args, **kwargs):
+        return _FakeResponse()
+
+
+def test_cwa_settlement_returns_recorded_intraday_history(monkeypatch):
+    repo = _FakeOfficialIntradayRepo()
+    monkeypatch.setattr(settlement_sources, "_official_intraday_repo", repo)
+    monkeypatch.setattr(db_manager, "DBManager", lambda: _FakeDBManager())
+
+    collector = _FakeCollector()
+    payload = collector.fetch_cwa_taipei_settlement_current()
+
+    assert payload is not None
+    assert "cwa:466920" in collector._settlement_cache
+    assert repo.upserts[0]["source_code"] == "cwa"
+    assert repo.upserts[0]["station_code"] == "466920"
+    assert payload["today_obs"] == [
+        {"time": "09:50", "temp": 34.0},
+        {"time": "10:00", "temp": 34.2},
+    ]