feat(workbench): add runtime GIS and sample integrations

2026-09-22 10:44:31 +08:00
parent a3b4e75317
commit 7ab9b36a5e
62 changed files with 6685 additions and 51 deletions
@@ -0,0 +1,153 @@
+"""Idempotently seed synthetic peripheral demo resources for one explicit project."""
+from __future__ import annotations
+
+import argparse
+import json
+import sys
+from pathlib import Path
+from uuid import NAMESPACE_URL, UUID, uuid5
+
+from psycopg.types.json import Jsonb
+
+ROOT = Path(__file__).resolve().parents[1]
+sys.path.insert(0, str(ROOT))
+
+from app.services.content_digest import digest
+from samples.peripheral.auth import Identity
+from samples.peripheral.config import SampleSettings
+from samples.peripheral.demo_data import build_demo_content
+from samples.peripheral.schemas import CreateTestDataRecord, CreateWorkOrder
+from samples.peripheral.store import PostgresStore
+
+
+SEED_NAMESPACE = uuid5(NAMESPACE_URL, "tjwater/peripheral-sample/seed/v1")
+ACTOR_ID = uuid5(SEED_NAMESPACE, "actor")
+
+
+def request_id(kind: str) -> UUID:
+    return uuid5(SEED_NAMESPACE, kind)
+
+
+WORK_ORDERS = (
+    {
+        "key": "work-order/demo-communication",
+        "title": "合成通信核查",
+        "summary": "检查 DEMO-P001 的预置通信中断情节，仅用于外围样板联调。",
+        "status": "open",
+        "priority": "high",
+        "assignee": "外围样板班组",
+        "scheduled_at": "2026-09-17T03:00:00Z",
+    },
+    {
+        "key": "work-order/demo-pressure",
+        "title": "合成压力偏差复核",
+        "summary": "复核 DEMO-P001、DEMO-P002 相邻测试点的固定合成读数，不代表现场异常。",
+        "status": "in_progress",
+        "priority": "urgent",
+        "assignee": "演示核查班组",
+        "scheduled_at": "2026-09-17T04:00:00Z",
+    },
+    {
+        "key": "work-order/flow-zero",
+        "title": "合成零流量样本确认",
+        "summary": "确认 DEMO-F001 中零值与缺测值可被界面明确区分。",
+        "status": "completed",
+        "priority": "normal",
+        "assignee": "外围样板班组",
+        "scheduled_at": "2026-09-17T06:00:00Z",
+    },
+)
+
+
+TEST_DATA = (
+    ("DEMO-P001", "合成压力 1 · 00:00", "2026-09-17T00:00:00Z", 24.37, "m", "valid"),
+    ("DEMO-P001", "合成压力 1 · 00:15", "2026-09-17T00:15:00Z", 24.18, "m", "valid"),
+    ("DEMO-P001", "合成压力 1 · 00:30 缺测", "2026-09-17T00:30:00Z", None, "m", "missing"),
+    ("DEMO-P001", "合成压力 1 · 00:45", "2026-09-17T00:45:00Z", 24.42, "m", "valid"),
+    ("DEMO-P002", "合成压力 2 · 00:00", "2026-09-17T00:00:00Z", 31.26, "m", "valid"),
+    ("DEMO-P002", "合成压力 2 · 00:15", "2026-09-17T00:15:00Z", 31.31, "m", "valid"),
+    ("DEMO-P002", "合成压力 2 · 00:30", "2026-09-17T00:30:00Z", 31.18, "m", "valid"),
+    ("DEMO-P002", "合成压力 2 · 00:45", "2026-09-17T00:45:00Z", 31.36, "m", "valid"),
+    ("DEMO-F001", "合成流量 00:00", "2026-09-17T00:00:00Z", 186.43, "m3/h", "valid"),
+    ("DEMO-F001", "合成流量 00:15", "2026-09-17T00:15:00Z", 185.90, "m3/h", "valid"),
+    ("DEMO-F001", "合成零流量 00:30", "2026-09-17T00:30:00Z", 0.0, "m3/h", "valid"),
+    ("DEMO-F001", "合成流量 00:45", "2026-09-17T00:45:00Z", 187.12, "m3/h", "valid"),
+)
+
+
+def ensure_demo_dataset(store: PostgresStore, project_id: str) -> str:
+    content = build_demo_content()
+    content_digest = digest(content)
+    with store.connection() as connection:
+        row = connection.execute(
+            """SELECT digest FROM peripheral_sample.demo_datasets
+               WHERE project_id=%s AND dataset_id='water-demo-v1'""",
+            (project_id,),
+        ).fetchone()
+        if row is None:
+            connection.execute(
+                """INSERT INTO peripheral_sample.demo_datasets
+                   (project_id,dataset_id,digest,document)
+                   VALUES (%s,'water-demo-v1',%s,%s)""",
+                (project_id, content_digest, Jsonb(content)),
+            )
+        elif row["digest"] != content_digest:
+            connection.execute(
+                """UPDATE peripheral_sample.demo_datasets
+                   SET digest=%s, document=%s
+                   WHERE project_id=%s AND dataset_id='water-demo-v1'""",
+                (content_digest, Jsonb(content), project_id),
+            )
+    return content_digest
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--project-id", required=True, type=UUID)
+    args = parser.parse_args()
+    project_id = str(args.project_id)
+    store = PostgresStore(SampleSettings(db_name="tjwater_v2", enable_faults=False))
+    who = Identity(project_id, str(ACTOR_ID), True)
+
+    dataset_digest = ensure_demo_dataset(store, project_id)
+    order_ids = []
+    for item in WORK_ORDERS:
+        command = CreateWorkOrder(
+            request_id=request_id(item["key"]),
+            title=item["title"],
+            summary=item["summary"],
+            status=item["status"],
+            priority=item["priority"],
+            assignee=item["assignee"],
+            scheduled_at=item["scheduled_at"],
+        )
+        order_ids.append(store.create_work_order(who, command)["work_order_id"])
+
+    record_ids = []
+    for point_id, label, observed_at, value, unit, quality in TEST_DATA:
+        command = CreateTestDataRecord(
+            request_id=request_id(f"test-data/{point_id}/{observed_at}"),
+            label=label,
+            point_id=point_id,
+            observed_at=observed_at,
+            value=value,
+            unit=unit,
+            quality=quality,
+            notes="外围样板固定合成数据，不是现场监测记录。",
+        )
+        record_ids.append(store.create_test_data(who, command)["record_id"])
+
+    result = {
+        "project_id": project_id,
+        "actor_id": str(ACTOR_ID),
+        "origin": "synthetic",
+        "demo_dataset_digest": dataset_digest,
+        "work_orders": {"count": len(order_ids), "ids": order_ids},
+        "test_data_records": {"count": len(record_ids), "ids": record_ids},
+    }
+    print(json.dumps(result, ensure_ascii=False, indent=2, sort_keys=True))
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())