wzy-warehouse
121 lines
3.8 KiB
Python
121 lines
3.8 KiB
Python
"""
|
|
灾害影响范围数据仓库
|
|
负责 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()
|