109 lines
3.4 KiB
Python
109 lines
3.4 KiB
Python
"""
|
||
检查数据库表结构，确认template_code字段是否存在
|
||
"""
|
||
import pymysql
|
||
import os
|
||
|
||
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_table_structure():
|
||
"""检查表结构"""
|
||
print("="*60)
|
||
print("检查数据库表结构")
|
||
print("="*60)
|
||
|
||
try:
|
||
conn = pymysql.connect(**DB_CONFIG)
|
||
cursor = conn.cursor()
|
||
|
||
# 检查f_polic_file_config表结构
|
||
print("\n1. 检查 f_polic_file_config 表结构：")
|
||
cursor.execute("DESCRIBE f_polic_file_config")
|
||
columns = cursor.fetchall()
|
||
|
||
has_template_code = False
|
||
has_input_data = False
|
||
|
||
for col in columns:
|
||
col_name = col[0]
|
||
col_type = col[1]
|
||
print(f" - {col_name}: {col_type}")
|
||
|
||
if col_name == 'template_code':
|
||
has_template_code = True
|
||
if col_name == 'input_data':
|
||
has_input_data = True
|
||
|
||
print(f"\n template_code 字段: {'✓ 存在' if has_template_code else '✗ 不存在'}")
|
||
print(f" input_data 字段: {'✓ 存在' if has_input_data else '✗ 不存在'}")
|
||
|
||
if not has_template_code:
|
||
print("\n ⚠ 警告：template_code字段不存在，可能需要添加该字段")
|
||
print(" 请联系数据库管理员添加该字段")
|
||
|
||
# 检查是否有数据
|
||
print("\n2. 检查文件配置数据：")
|
||
cursor.execute("""
|
||
SELECT COUNT(*)
|
||
FROM f_polic_file_config
|
||
WHERE tenant_id = %s
|
||
""", (TENANT_ID,))
|
||
count = cursor.fetchone()[0]
|
||
print(f" 找到 {count} 个文件配置记录")
|
||
|
||
if count > 0:
|
||
# 检查template_code字段是否有值
|
||
if has_template_code:
|
||
cursor.execute("""
|
||
SELECT COUNT(*)
|
||
FROM f_polic_file_config
|
||
WHERE tenant_id = %s AND template_code IS NOT NULL AND template_code != ''
|
||
""", (TENANT_ID,))
|
||
code_count = cursor.fetchone()[0]
|
||
print(f" 其中 {code_count} 个记录有template_code值")
|
||
|
||
if code_count < count:
|
||
print(f" ⚠ 建议运行: python update_template_code_field.py")
|
||
|
||
# 检查字段数据
|
||
print("\n3. 检查字段数据：")
|
||
cursor.execute("""
|
||
SELECT COUNT(*)
|
||
FROM f_polic_field
|
||
WHERE tenant_id = %s
|
||
""", (TENANT_ID,))
|
||
field_count = cursor.fetchone()[0]
|
||
print(f" 找到 {field_count} 个字段记录")
|
||
|
||
if field_count == 0:
|
||
print(" ⚠ 建议运行: python init_all_fields_from_excel.py")
|
||
|
||
cursor.close()
|
||
conn.close()
|
||
|
||
print("\n" + "="*60)
|
||
print("检查完成")
|
||
print("="*60)
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"\n✗ 检查失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
if __name__ == '__main__':
|
||
check_table_structure()
|
||
|
||
|