"""
修复剩余的层级结构问题
"""
import pymysql

DB_CONFIG = {
    'host': '152.136.177.240',
    'port': 5012,
    'user': 'finyx',
    'password': '6QsGK6MpePZDE57Z',
    'database': 'finyx',
    'charset': 'utf8mb4'
}

TENANT_ID = 615873064429507639
UPDATED_BY = 655162080928945152

conn = pymysql.connect(**DB_CONFIG)
cursor = conn.cursor()

try:
    # 1. 修复"2保密承诺书"的parent_id（应该在"走读式谈话流程"下）
    # "走读式谈话流程"的ID是 1765273962716807
    cursor.execute("""
        UPDATE f_polic_file_config
        SET parent_id = %s, updated_time = NOW(), updated_by = %s
        WHERE tenant_id = %s AND id = %s
    """, (1765273962716807, UPDATED_BY, TENANT_ID, 1765425919729046))
    print(f"[UPDATE] 更新'2保密承诺书'的parent_id: {cursor.rowcount} 条")
    
    # 2. 检查"8.XXX初核情况报告"的位置（应该在"3.初核结论"下，而不是"走读式谈话流程"下）
    # "3.初核结论"的ID是 1765431559135346
    # 先查找"8.XXX初核情况报告"的ID
    cursor.execute("""
        SELECT id, name, parent_id
        FROM f_polic_file_config
        WHERE tenant_id = %s AND name LIKE %s
    """, (TENANT_ID, '%XXX初核情况报告%'))
    result = cursor.fetchone()
    if result:
        file_id, file_name, current_parent = result
        if current_parent != 1765431559135346:
            cursor.execute("""
                UPDATE f_polic_file_config
                SET parent_id = %s, updated_time = NOW(), updated_by = %s
                WHERE tenant_id = %s AND id = %s
            """, (1765431559135346, UPDATED_BY, TENANT_ID, file_id))
            print(f"[UPDATE] 更新'{file_name}'的parent_id: {cursor.rowcount} 条")
    
    conn.commit()
    print("\n[OK] 修复完成")
    
except Exception as e:
    conn.rollback()
    print(f"[ERROR] 修复失败: {e}")
    import traceback
    traceback.print_exc()
finally:
    cursor.close()
    conn.close()