QGIS完成初步重构

2026-06-20 15:50:24 +08:00
parent d20b5744bb
commit 18d8bcb1a3
45 changed files with 1688 additions and 454 deletions
@@ -0,0 +1,71 @@
+"""检查暴雨避难场所分布图的所有动态图层是否有数据"""
+import zipfile, re, os, psycopg2
+
+TEMPLATE_PATH = r"F:\project\xian\xian_algorithm_new\app\data\template\rainfall\暴雨避难场所分布图.qgz"
+
+z = zipfile.ZipFile(TEMPLATE_PATH)
+qgs_name = [n for n in z.namelist() if n.endswith('.qgs')][0]
+content = z.read(qgs_name).decode('utf-8')
+
+maplayer_re = re.compile(r'<maplayer[^>]*>(.*?)</maplayer>', re.DOTALL)
+
+# 收集所有 PostgreSQL 动态图层
+layers = []
+for m in maplayer_re.finditer(content):
+    block = m.group(1)
+    name_m = re.search(r'<layername>([^<]+)</layername>', block)
+    provider_m = re.search(r'<provider[^>]*>(\w+)</provider>', block)
+    ds_m = re.search(r'<datasource>(.*?)</datasource>', block, re.DOTALL)
+    
+    provider = provider_m.group(1) if provider_m else '?'
+    if provider != 'postgres':
+        continue
+    
+    name = name_m.group(1) if name_m else '?'
+    ds = ds_m.group(1).strip() if ds_m else ''
+    table_m = re.search(r'table="(\w+)"\."(\w+)"', ds)
+    table_key = f"{table_m.group(1)}.{table_m.group(2)}" if table_m else '?'
+
+    layers.append((name, table_key))
+
+# 已知 GPKG 静态层
+static_tables = {
+    'base.rivers', 'base.river', 'base.sx', 'base.sx_capital',
+    'base.sx_street', 'base.sx_xa_county', 'base.sx_xa_county_boundary',
+    'base.sx_zb_county_boundary', 'base.sx_zb_city', 'base.sx_zb_county',
+    'base.active_fault', 'base.traffic_expressway', 'base.traffic_provincial',
+    'base.traffic_railway', 'base.traffic_township', 'base.traffic_trunk_line',
+}
+
+c = psycopg2.connect(host='47.92.216.173', port=7654, user='postgres', password='zhangsan', database='xian_new')
+c.autocommit = True
+cur = c.cursor()
+
+print(f"模板: 暴雨避难场所分布图")
+print(f"{'图层名':20s} {'原表':40s} {'qgis有数据':10s} {'行数':>6s}")
+print("-" * 85)
+
+for name, table_key in layers:
+    if table_key in static_tables:
+        continue
+    
+    schema, table = table_key.split('.', 1)
+    mapped_table = 'hazard_hydrops' if table == 'hazard_waterlogging' else table
+    
+    try:
+        cur.execute(f'SELECT count(*) FROM qgis.{mapped_table}')
+        count = cur.fetchone()[0]
+        exists = "YES" if count > 0 else "EMPTY"
+    except:
+        try:
+            cur.execute(f'SELECT count(*) FROM {schema}.{table}')
+            count = cur.fetchone()[0]
+            exists = f"在{schema}"
+        except:
+            exists = "NO"
+            count = 0
+    
+    marker = " <-- 无数据!" if count == 0 else ""
+    print(f"{name:20s} {table_key:40s} {exists:10s} {count:>6,d}{marker}")
+
+c.close()