207 lines
7.2 KiB
Python
207 lines
7.2 KiB
Python
"""
|
||
验证文档生成接口可以正确生成文档
|
||
测试模板和字段关联是否正确
|
||
"""
|
||
import sys
|
||
import os
|
||
import json
|
||
import pymysql
|
||
sys.path.insert(0, os.path.dirname(__file__))
|
||
|
||
from services.document_service import DocumentService
|
||
|
||
# 数据库连接配置
|
||
DB_CONFIG = {
|
||
'host': '152.136.177.240',
|
||
'port': 5012,
|
||
'user': 'finyx',
|
||
'password': '6QsGK6MpePZDE57Z',
|
||
'database': 'finyx',
|
||
'charset': 'utf8mb4'
|
||
}
|
||
|
||
TENANT_ID = 615873064429507639
|
||
|
||
|
||
def get_template_by_name(conn, template_name: str):
|
||
"""根据模板名称获取模板信息"""
|
||
cursor = conn.cursor(pymysql.cursors.DictCursor)
|
||
|
||
try:
|
||
sql = """
|
||
SELECT id, name, file_path, state
|
||
FROM f_polic_file_config
|
||
WHERE tenant_id = %s AND name = %s AND state = 1
|
||
"""
|
||
cursor.execute(sql, (TENANT_ID, template_name))
|
||
return cursor.fetchone()
|
||
finally:
|
||
cursor.close()
|
||
|
||
|
||
def get_template_fields(conn, file_id: int):
|
||
"""获取模板关联的字段"""
|
||
cursor = conn.cursor(pymysql.cursors.DictCursor)
|
||
|
||
try:
|
||
sql = """
|
||
SELECT f.id, f.name, f.filed_code, f.field_type
|
||
FROM f_polic_field f
|
||
INNER JOIN f_polic_file_field fff ON f.id = fff.filed_id
|
||
WHERE fff.file_id = %s AND fff.tenant_id = %s AND fff.state = 1
|
||
ORDER BY f.field_type, f.filed_code
|
||
"""
|
||
cursor.execute(sql, (file_id, TENANT_ID))
|
||
return cursor.fetchall()
|
||
finally:
|
||
cursor.close()
|
||
|
||
|
||
def test_document_generation(template_name: str, test_data: list):
|
||
"""测试文档生成"""
|
||
print("=" * 80)
|
||
print(f"测试文档生成: {template_name}")
|
||
print("=" * 80)
|
||
|
||
# 连接数据库
|
||
conn = pymysql.connect(**DB_CONFIG)
|
||
|
||
try:
|
||
# 获取模板信息
|
||
template = get_template_by_name(conn, template_name)
|
||
if not template:
|
||
print(f"[ERROR] 未找到模板: {template_name}")
|
||
return False
|
||
|
||
print(f"\n模板信息:")
|
||
print(f" ID: {template['id']}")
|
||
print(f" 名称: {template['name']}")
|
||
print(f" 文件路径: {template['file_path']}")
|
||
print(f" 状态: {template['state']}")
|
||
|
||
# 获取模板关联的字段
|
||
fields = get_template_fields(conn, template['id'])
|
||
print(f"\n关联的字段数量: {len(fields)}")
|
||
if fields:
|
||
print(" 字段列表:")
|
||
for field in fields[:10]: # 只显示前10个
|
||
field_type = "输出字段" if field['field_type'] == 2 else "输入字段"
|
||
print(f" - {field['name']} ({field['filed_code']}) [{field_type}]")
|
||
if len(fields) > 10:
|
||
print(f" ... 还有 {len(fields) - 10} 个字段")
|
||
|
||
# 准备测试数据
|
||
print(f"\n测试数据字段数量: {len(test_data)}")
|
||
|
||
# 创建文档服务
|
||
doc_service = DocumentService()
|
||
|
||
# 准备文件信息
|
||
file_info = {
|
||
'fileId': template['id'],
|
||
'fileName': f"{template_name}.doc"
|
||
}
|
||
|
||
print(f"\n开始生成文档...")
|
||
|
||
# 生成文档
|
||
try:
|
||
result = doc_service.generate_document(
|
||
file_id=template['id'],
|
||
input_data=test_data,
|
||
file_info=file_info
|
||
)
|
||
|
||
print(f"[OK] 文档生成成功！")
|
||
print(f"\n生成结果:")
|
||
print(f" 文件路径: {result.get('filePath')}")
|
||
print(f" 文件名称: {result.get('fileName')}")
|
||
if result.get('downloadUrl'):
|
||
print(f" 下载URL: {result.get('downloadUrl')[:80]}...")
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"[ERROR] 文档生成失败: {str(e)}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
finally:
|
||
conn.close()
|
||
|
||
|
||
def main():
|
||
"""主函数"""
|
||
print("=" * 80)
|
||
print("验证文档生成功能")
|
||
print("=" * 80)
|
||
print()
|
||
|
||
# 测试数据
|
||
test_data = [
|
||
{"fieldCode": "target_name", "fieldValue": "张三"},
|
||
{"fieldCode": "target_gender", "fieldValue": "男"},
|
||
{"fieldCode": "target_age", "fieldValue": "44"},
|
||
{"fieldCode": "target_date_of_birth", "fieldValue": "198005"},
|
||
{"fieldCode": "target_organization_and_position", "fieldValue": "某公司总经理"},
|
||
{"fieldCode": "target_organization", "fieldValue": "某公司"},
|
||
{"fieldCode": "target_position", "fieldValue": "总经理"},
|
||
{"fieldCode": "target_education_level", "fieldValue": "本科"},
|
||
{"fieldCode": "target_political_status", "fieldValue": "中共党员"},
|
||
{"fieldCode": "target_professional_rank", "fieldValue": "正处级"},
|
||
{"fieldCode": "clue_source", "fieldValue": "群众举报"},
|
||
{"fieldCode": "target_issue_description", "fieldValue": "违反国家计划生育有关政策规定，于2010年10月生育二胎。"},
|
||
{"fieldCode": "department_opinion", "fieldValue": "建议进行初步核实"},
|
||
{"fieldCode": "filler_name", "fieldValue": "李四"},
|
||
{"fieldCode": "target_id_number", "fieldValue": "110101198005011234"},
|
||
{"fieldCode": "target_contact", "fieldValue": "13800138000"},
|
||
{"fieldCode": "target_work_basic_info", "fieldValue": "在某公司工作10年，担任总经理职务"},
|
||
{"fieldCode": "target_family_situation", "fieldValue": "已婚，有一子一女"},
|
||
{"fieldCode": "target_social_relations", "fieldValue": "社会关系简单"},
|
||
{"fieldCode": "investigation_unit_name", "fieldValue": "某市纪委监委"},
|
||
{"fieldCode": "investigation_team_leader_name", "fieldValue": "王五"},
|
||
{"fieldCode": "investigation_team_member_names", "fieldValue": "赵六、钱七"},
|
||
{"fieldCode": "investigation_team_code", "fieldValue": "DC2024001"},
|
||
{"fieldCode": "investigation_location", "fieldValue": "某市纪委监委谈话室"},
|
||
{"fieldCode": "appointment_time", "fieldValue": "2024年12月10日上午9:00"},
|
||
{"fieldCode": "appointment_location", "fieldValue": "某市纪委监委谈话室"},
|
||
{"fieldCode": "approval_time", "fieldValue": "2024年12月9日"},
|
||
{"fieldCode": "handling_department", "fieldValue": "某市纪委监委第一监督检查室"},
|
||
{"fieldCode": "handler_name", "fieldValue": "王五"},
|
||
]
|
||
|
||
# 测试几个关键模板
|
||
test_templates = [
|
||
"初步核实审批表",
|
||
"请示报告卡",
|
||
"谈话通知书第一联",
|
||
"谈话前安全风险评估表"
|
||
]
|
||
|
||
success_count = 0
|
||
failed_count = 0
|
||
|
||
for template_name in test_templates:
|
||
print()
|
||
success = test_document_generation(template_name, test_data)
|
||
if success:
|
||
success_count += 1
|
||
else:
|
||
failed_count += 1
|
||
print()
|
||
|
||
# 打印汇总
|
||
print("=" * 80)
|
||
print("测试汇总")
|
||
print("=" * 80)
|
||
print(f"总测试数: {len(test_templates)}")
|
||
print(f"成功: {success_count}")
|
||
print(f"失败: {failed_count}")
|
||
print("=" * 80)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|
||
|