新增爆管侦测功能及相关API接口

2026-03-11 10:31:24 +08:00
parent 90216a762a
commit bef1c74782
7 changed files with 743 additions and 1 deletions
@@ -401,6 +401,85 @@ def query_burst_location_scheme_detail(name: str, scheme_name: str) -> dict:
    }


+def query_burst_detection_schemes(
+    name: str,
+    network: str,
+    scheme_type: str = "burst_detection",
+    query_date: date | None = None,
+) -> list[dict]:
+    conn_string = get_pgconn_string(db_name=name)
+    with psycopg.connect(conn_string) as conn:
+        with conn.cursor() as cur:
+            if query_date is None:
+                cur.execute(
+                    """
+                    SELECT scheme_id, scheme_name, scheme_type, username, create_time, scheme_start_time, scheme_detail
+                    FROM public.scheme_list
+                    WHERE scheme_type = %s
+                    ORDER BY create_time DESC
+                    """,
+                    (scheme_type,),
+                )
+            else:
+                cur.execute(
+                    """
+                    SELECT scheme_id, scheme_name, scheme_type, username, create_time, scheme_start_time, scheme_detail
+                    FROM public.scheme_list
+                    WHERE scheme_type = %s AND DATE(create_time) = %s
+                    ORDER BY create_time DESC
+                    """,
+                    (scheme_type, query_date),
+                )
+            rows = cur.fetchall()
+    result = []
+    for row in rows:
+        detail = row[6] if isinstance(row[6], dict) else {}
+        if network and detail.get("network") not in (None, network):
+            continue
+        result.append(
+            {
+                "scheme_id": row[0],
+                "scheme_name": row[1],
+                "scheme_type": row[2],
+                "username": row[3],
+                "create_time": row[4],
+                "scheme_start_time": row[5],
+                "scheme_detail": detail,
+            }
+        )
+    return result
+
+
+def query_burst_detection_scheme_detail(name: str, scheme_name: str) -> dict:
+    conn_string = get_pgconn_string(db_name=name)
+    with psycopg.connect(conn_string) as conn:
+        with conn.cursor() as cur:
+            cur.execute(
+                """
+                SELECT scheme_id, scheme_name, scheme_type, username, create_time, scheme_start_time, scheme_detail
+                FROM public.scheme_list
+                WHERE scheme_name = %s
+                LIMIT 1
+                """,
+                (scheme_name,),
+            )
+            base_row = cur.fetchone()
+            if base_row is None:
+                return {}
+    detail = base_row[6] if isinstance(base_row[6], dict) else {}
+    return {
+        "scheme_id": base_row[0],
+        "scheme_name": base_row[1],
+        "scheme_type": base_row[2],
+        "username": base_row[3],
+        "create_time": base_row[4],
+        "scheme_start_time": base_row[5],
+        "scheme_detail": detail,
+        "network": detail.get("network"),
+        "result_payload": detail.get("result_payload", {}),
+    }
+
+
 # 2025/03/23
 def upload_shp_to_pg(name: str, table_name: str, role: str, shp_file_path: str):
    """