"""检查暴雨避难场所分布图的所有动态图层是否有数据"""
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()