ai-business-write/check_template_all_relations.py

99 lines
3.8 KiB
Python

"""
检查模板的所有关联关系(包括未启用的)
"""
import pymysql
import os
from dotenv import load_dotenv
load_dotenv()
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
TEMPLATE_ID = 1765432134276990 # 1.请示报告卡(初核谈话)
def check_all_relations():
"""检查模板的所有关联关系"""
conn = pymysql.connect(**DB_CONFIG)
cursor = conn.cursor(pymysql.cursors.DictCursor)
try:
print(f"检查模板 ID: {TEMPLATE_ID}")
print("=" * 80)
# 查询模板信息
cursor.execute("""
SELECT id, name, state
FROM f_polic_file_config
WHERE tenant_id = %s AND id = %s
""", (TENANT_ID, TEMPLATE_ID))
template = cursor.fetchone()
if template:
print(f"模板名称: {template['name']}")
print(f"模板状态: {template['state']}")
else:
print("模板不存在")
return
# 查询所有关联关系(包括 state=0 的)
cursor.execute("""
SELECT
fff.file_id,
fff.filed_id,
fff.state as relation_state,
f.name as field_name,
f.field_type,
f.state as field_state
FROM f_polic_file_field fff
INNER JOIN f_polic_field f ON fff.filed_id = f.id AND fff.tenant_id = f.tenant_id
WHERE fff.tenant_id = %s AND fff.file_id = %s
ORDER BY f.field_type, f.name
""", (TENANT_ID, TEMPLATE_ID))
all_relations = cursor.fetchall()
print(f"\n所有关联关系数: {len(all_relations)}")
# 按状态分组
enabled_relations = [r for r in all_relations if r['relation_state'] == 1 or (isinstance(r['relation_state'], bytes) and r['relation_state'] == b'\x01')]
disabled_relations = [r for r in all_relations if r not in enabled_relations]
print(f"启用的关联关系: {len(enabled_relations)}")
print(f"未启用的关联关系: {len(disabled_relations)}")
# 按字段类型分组
input_fields = [r for r in enabled_relations if r['field_type'] == 1]
output_fields = [r for r in enabled_relations if r['field_type'] == 2]
print(f"\n启用的输入字段关联: {len(input_fields)}")
for r in input_fields:
state_str = str(r['relation_state']) if not isinstance(r['relation_state'], bytes) else 'bytes'
print(f" - {r['field_name']} (ID: {r['filed_id']}, relation_state: {state_str}, field_state: {r['field_state']})")
print(f"\n启用的输出字段关联: {len(output_fields)}")
for r in output_fields[:10]:
state_str = str(r['relation_state']) if not isinstance(r['relation_state'], bytes) else 'bytes'
print(f" - {r['field_name']} (ID: {r['filed_id']}, relation_state: {state_str}, field_state: {r['field_state']})")
if len(output_fields) > 10:
print(f" ... 还有 {len(output_fields) - 10} 个输出字段")
# 检查未启用的关联关系
if disabled_relations:
print(f"\n未启用的关联关系: {len(disabled_relations)}")
disabled_input = [r for r in disabled_relations if r['field_type'] == 1]
disabled_output = [r for r in disabled_relations if r['field_type'] == 2]
print(f" 输入字段: {len(disabled_input)}, 输出字段: {len(disabled_output)}")
finally:
cursor.close()
conn.close()
if __name__ == '__main__':
check_all_relations()