进行路径计算

2026-06-29 11:37:04 +08:00
parent 1f01ceb062
commit e9169dfd13
6 changed files with 745 additions and 0 deletions
@@ -0,0 +1,120 @@
+"""
+灾害影响范围数据仓库
+负责 xian_hazard_zones 表的数据读取和写入
+"""
+from typing import List, Dict, Any, Optional
+
+from app.utils.db_helper import db_helper
+
+
+class HazardRepository:
+    """灾害影响范围数据访问层"""
+
+    # ==================== 隐患点查询 ====================
+
+    @staticmethod
+    def get_all_hidden_danger_spots() -> List[Dict[str, Any]]:
+        """
+        获取所有隐患点（含静态因子）
+
+        Returns:
+            隐患点列表，每个元素含: source_id, disaster_name, disaster_type,
+            scale_grade, county, lon, lat, static_factors
+        """
+        sql = """
+            SELECT
+                h.id AS source_id,
+                h.disaster_name,
+                h.disaster_type,
+                h.scale_grade,
+                h.county,
+                rf.lon,
+                rf.lat,
+                rf.static_factors
+            FROM xian_hidden_danger_spots h
+            JOIN xian_risk_factors rf
+                ON rf.source_id = h.id AND rf.source_type = 1
+            WHERE h.is_delete = 0
+              AND rf.is_delete = 0
+              AND h.disaster_type IN ('滑坡', '泥石流', '山洪', '内涝', '崩塌')
+              AND rf.lon IS NOT NULL
+              AND rf.lat IS NOT NULL
+            ORDER BY h.id
+        """
+        rows = db_helper.execute_query(sql)
+        return [{
+            'source_id': r['source_id'],
+            'disaster_name': r['disaster_name'],
+            'disaster_type': r['disaster_type'],
+            'scale_grade': r['scale_grade'],
+            'county': r['county'],
+            'lon': float(r['lon']),
+            'lat': float(r['lat']),
+            'static_factors': r.get('static_factors') or {}
+        } for r in rows]
+
+    # ==================== 影响范围写入 ====================
+
+    @staticmethod
+    def delete_existing_zones() -> int:
+        """逻辑删除已有影响范围记录"""
+        sql = "UPDATE xian_hazard_zones SET is_delete = 1 WHERE is_delete = 0"
+        return db_helper.execute_update(sql)
+
+    @staticmethod
+    def insert_hazard_zone(source_id: int, disaster_type: str,
+                           zone_wkt: str, path_wkt: Optional[str]) -> int:
+        """
+        插入单条影响范围记录
+
+        Args:
+            source_id: 隐患点ID
+            disaster_type: 灾害类型
+            zone_wkt: 影响范围 POLYGON WKT
+            path_wkt: 影响路径 LINESTRING WKT (可为None)
+
+        Returns:
+            新插入记录的ID
+        """
+        if path_wkt:
+            sql = """
+                INSERT INTO xian_hazard_zones (source_id, disaster_type, zone_geom, path_geom)
+                VALUES (%s, %s, ST_GeomFromText(%s, 4326), ST_GeomFromText(%s, 4326))
+                RETURNING id
+            """
+            row = db_helper.execute_query_one(sql, (source_id, disaster_type, zone_wkt, path_wkt))
+        else:
+            sql = """
+                INSERT INTO xian_hazard_zones (source_id, disaster_type, zone_geom)
+                VALUES (%s, %s, ST_GeomFromText(%s, 4326))
+                RETURNING id
+            """
+            row = db_helper.execute_query_one(sql, (source_id, disaster_type, zone_wkt))
+        return row['id'] if row else 0
+
+    @staticmethod
+    def batch_insert_zones(records: List[Dict[str, Any]]) -> int:
+        """
+        批量插入影响范围记录
+
+        Args:
+            records: [{source_id, disaster_type, zone_wkt, path_wkt}, ...]
+
+        Returns:
+            成功插入的记录数
+        """
+        count = 0
+        for rec in records:
+            hid = HazardRepository.insert_hazard_zone(
+                rec['source_id'],
+                rec['disaster_type'],
+                rec['zone_wkt'],
+                rec.get('path_wkt')
+            )
+            if hid > 0:
+                count += 1
+        return count
+
+
+# 全局实例
+hazard_repository = HazardRepository()