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,42 @@
+"""
+在所有 qgis 动态表中添加 tid 自增主键列
+QGIS 的 datasource key='tid' 需要此列
+"""
+import psycopg2
+
+TABLES = [
+    'hazard_hydrops',
+    'lifeline_outfall', 
+    'lifeline_pipe',
+    'risk_census_population',
+    'sx_xa_towns',
+]
+
+c = psycopg2.connect(
+    host='47.92.216.173', port=7654,
+    user='postgres', password='zhangsan',
+    database='xian_new'
+)
+c.autocommit = True
+cur = c.cursor()
+
+for t in TABLES:
+    try:
+        # 检查 tid 列是否存在
+        cur.execute(f"""
+            SELECT EXISTS (
+                SELECT FROM information_schema.columns 
+                WHERE table_schema='qgis' AND table_name='{t}' AND column_name='tid'
+            )
+        """)
+        exists = cur.fetchone()[0]
+        
+        if exists:
+            print(f"qgis.{t}: tid 列已存在, 跳过")
+        else:
+            cur.execute(f'ALTER TABLE qgis."{t}" ADD COLUMN tid SERIAL')
+            print(f"qgis.{t}: tid 列添加成功")
+    except Exception as e:
+        print(f"qgis.{t}: 失败 - {e}")
+
+c.close()