106 lines
3.6 KiB
Python
106 lines
3.6 KiB
Python
"""
|
|
检查数据库中的现有数据，确认匹配情况
|
|
"""
|
|
import os
|
|
import json
|
|
import pymysql
|
|
from pathlib import Path
|
|
|
|
# 数据库连接配置
|
|
DB_CONFIG = {
|
|
'host': os.getenv('DB_HOST', '152.136.177.240'),
|
|
'port': int(os.getenv('DB_PORT', 5012)),
|
|
'user': os.getenv('DB_USER', 'finyx'),
|
|
'password': os.getenv('DB_PASSWORD', '6QsGK6MpePZDE57Z'),
|
|
'database': os.getenv('DB_NAME', 'finyx'),
|
|
'charset': 'utf8mb4'
|
|
}
|
|
|
|
TENANT_ID = 615873064429507639
|
|
|
|
def check_existing_data():
|
|
"""检查数据库中的现有数据"""
|
|
print("="*80)
|
|
print("检查数据库中的现有数据")
|
|
print("="*80)
|
|
|
|
try:
|
|
conn = pymysql.connect(**DB_CONFIG)
|
|
cursor = conn.cursor(pymysql.cursors.DictCursor)
|
|
|
|
# 查询所有记录
|
|
sql = """
|
|
SELECT id, name, parent_id, template_code, input_data, file_path, state
|
|
FROM f_polic_file_config
|
|
WHERE tenant_id = %s
|
|
ORDER BY name
|
|
"""
|
|
cursor.execute(sql, (TENANT_ID,))
|
|
configs = cursor.fetchall()
|
|
|
|
print(f"\n共找到 {len(configs)} 条记录\n")
|
|
|
|
# 按 parent_id 分组统计
|
|
with_parent = []
|
|
without_parent = []
|
|
|
|
for config in configs:
|
|
# 尝试从 input_data 中提取 template_code
|
|
template_code = config.get('template_code')
|
|
if not template_code and config.get('input_data'):
|
|
try:
|
|
input_data = json.loads(config['input_data']) if isinstance(config['input_data'], str) else config['input_data']
|
|
if isinstance(input_data, dict):
|
|
template_code = input_data.get('template_code')
|
|
except:
|
|
pass
|
|
|
|
config['extracted_template_code'] = template_code
|
|
|
|
if config.get('parent_id'):
|
|
with_parent.append(config)
|
|
else:
|
|
without_parent.append(config)
|
|
|
|
print(f"有 parent_id 的记录: {len(with_parent)} 条")
|
|
print(f"无 parent_id 的记录: {len(without_parent)} 条\n")
|
|
|
|
# 显示无 parent_id 的记录
|
|
print("="*80)
|
|
print("无 parent_id 的记录列表：")
|
|
print("="*80)
|
|
for i, config in enumerate(without_parent, 1):
|
|
print(f"\n{i}. {config['name']}")
|
|
print(f" ID: {config['id']}")
|
|
print(f" template_code: {config.get('extracted_template_code') or config.get('template_code') or '无'}")
|
|
print(f" file_path: {config.get('file_path', '无')}")
|
|
print(f" state: {config.get('state')}")
|
|
|
|
# 显示有 parent_id 的记录（树状结构）
|
|
print("\n" + "="*80)
|
|
print("有 parent_id 的记录（树状结构）：")
|
|
print("="*80)
|
|
|
|
# 构建ID到名称的映射
|
|
id_to_name = {config['id']: config['name'] for config in configs}
|
|
|
|
for config in with_parent:
|
|
parent_name = id_to_name.get(config['parent_id'], f"ID:{config['parent_id']}")
|
|
print(f"\n{config['name']}")
|
|
print(f" ID: {config['id']}")
|
|
print(f" 父节点: {parent_name} (ID: {config['parent_id']})")
|
|
print(f" template_code: {config.get('extracted_template_code') or config.get('template_code') or '无'}")
|
|
|
|
cursor.close()
|
|
conn.close()
|
|
|
|
except Exception as e:
|
|
print(f"错误: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
check_existing_data()
|
|
|