229 lines
7.6 KiB
Python
229 lines
7.6 KiB
Python
"""
|
||
测试文档生成功能 - 使用虚拟数据生成初步核实审批表
|
||
"""
|
||
from services.document_service import DocumentService
|
||
from datetime import datetime, timedelta
|
||
from minio import Minio
|
||
import os
|
||
|
||
# 虚拟数据 - 初步核实审批表字段
|
||
VIRTUAL_DATA = [
|
||
{"fieldCode": "target_name", "fieldValue": "张三"},
|
||
{"fieldCode": "target_gender", "fieldValue": "男"},
|
||
{"fieldCode": "target_nationality", "fieldValue": "汉族"},
|
||
{"fieldCode": "target_date_of_birth", "fieldValue": "198005"},
|
||
{"fieldCode": "target_place_of_birth", "fieldValue": "山西太原"},
|
||
{"fieldCode": "target_political_status", "fieldValue": "中共党员"},
|
||
{"fieldCode": "target_education", "fieldValue": "本科学历"},
|
||
{"fieldCode": "target_organization_and_position", "fieldValue": "某公司总经理"},
|
||
{"fieldCode": "target_id_number", "fieldValue": "140101198005150123"},
|
||
{"fieldCode": "target_contact", "fieldValue": "13800138000"},
|
||
{"fieldCode": "target_address", "fieldValue": "山西省太原市小店区某某街道123号"},
|
||
{"fieldCode": "clue_source", "fieldValue": "群众举报"},
|
||
{"fieldCode": "clue_info", "fieldValue": "被举报人利用职务便利，违规接受他人请托，收受他人财物"},
|
||
{"fieldCode": "investigation_reason", "fieldValue": "根据群众举报，反映被核查人存在违规违纪问题"},
|
||
{"fieldCode": "investigation_basis", "fieldValue": "《中国共产党纪律检查机关监督执纪工作规则》相关规定"},
|
||
]
|
||
|
||
TEMPLATE_CODE = "PRELIMINARY_VERIFICATION_APPROVAL"
|
||
FILE_NAME = "初步核实审批表_张三.docx"
|
||
|
||
def print_section(title):
|
||
"""打印章节标题"""
|
||
print("\n" + "="*60)
|
||
print(f" {title}")
|
||
print("="*60)
|
||
|
||
def print_result(success, message):
|
||
"""打印测试结果"""
|
||
status = "✓" if success else "✗"
|
||
print(f"{status} {message}")
|
||
|
||
def test_document_generation():
|
||
"""测试文档生成"""
|
||
print_section("文档生成测试 - 初步核实审批表")
|
||
|
||
print("\n使用虚拟数据生成文档...")
|
||
print(f"模板编码: {TEMPLATE_CODE}")
|
||
print(f"文件名称: {FILE_NAME}")
|
||
print(f"\n虚拟数据字段数量: {len(VIRTUAL_DATA)}")
|
||
print("\n字段列表:")
|
||
for item in VIRTUAL_DATA[:5]: # 只显示前5个
|
||
print(f" - {item['fieldCode']}: {item['fieldValue']}")
|
||
print(f" ... 还有 {len(VIRTUAL_DATA) - 5} 个字段")
|
||
|
||
try:
|
||
# 创建文档服务实例
|
||
doc_service = DocumentService()
|
||
|
||
# 准备文件信息
|
||
file_info = {
|
||
'fileId': 1,
|
||
'fileName': FILE_NAME,
|
||
'templateCode': TEMPLATE_CODE
|
||
}
|
||
|
||
print("\n开始生成文档...")
|
||
|
||
# 生成文档
|
||
result = doc_service.generate_document(
|
||
template_code=TEMPLATE_CODE,
|
||
input_data=VIRTUAL_DATA,
|
||
file_info=file_info
|
||
)
|
||
|
||
print_result(True, "文档生成成功！")
|
||
|
||
file_path = result.get('filePath', '')
|
||
print(f"\n生成的文件路径: {file_path}")
|
||
|
||
return file_path
|
||
|
||
except Exception as e:
|
||
print_result(False, f"文档生成失败: {str(e)}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return None
|
||
|
||
def generate_download_url(file_path):
|
||
"""生成下载URL"""
|
||
print_section("生成下载URL")
|
||
|
||
if not file_path:
|
||
print("⚠ 没有文件路径，无法生成URL")
|
||
return None
|
||
|
||
try:
|
||
# 去掉开头的斜杠
|
||
object_name = file_path.lstrip('/')
|
||
|
||
# 创建MinIO客户端
|
||
minio_config = {
|
||
'endpoint': os.getenv('MINIO_ENDPOINT', 'minio.datacubeworld.com:9000'),
|
||
'access_key': os.getenv('MINIO_ACCESS_KEY', 'JOLXFXny3avFSzB0uRA5'),
|
||
'secret_key': os.getenv('MINIO_SECRET_KEY', 'G1BR8jStNfovkfH5ou39EmPl34E4l7dGrnd3Cz0I'),
|
||
'secure': os.getenv('MINIO_SECURE', 'true').lower() == 'true'
|
||
}
|
||
bucket_name = os.getenv('MINIO_BUCKET', 'finyx')
|
||
|
||
client = Minio(
|
||
minio_config['endpoint'],
|
||
access_key=minio_config['access_key'],
|
||
secret_key=minio_config['secret_key'],
|
||
secure=minio_config['secure']
|
||
)
|
||
|
||
print(f"生成预签名URL（7天有效）...")
|
||
print(f"对象名称: {object_name}")
|
||
|
||
# 生成预签名URL（7天有效期）
|
||
url = client.presigned_get_object(
|
||
bucket_name,
|
||
object_name,
|
||
expires=timedelta(days=7)
|
||
)
|
||
|
||
print_result(True, "URL生成成功！")
|
||
print(f"\n下载链接（7天有效）:")
|
||
print(f"{url}")
|
||
|
||
return url
|
||
|
||
except Exception as e:
|
||
print_result(False, f"URL生成失败: {str(e)}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return None
|
||
|
||
def verify_file_exists(file_path):
|
||
"""验证文件是否存在"""
|
||
print_section("验证文件是否存在")
|
||
|
||
if not file_path:
|
||
print("⚠ 没有文件路径")
|
||
return False
|
||
|
||
try:
|
||
# 去掉开头的斜杠
|
||
object_name = file_path.lstrip('/')
|
||
|
||
# 创建MinIO客户端
|
||
minio_config = {
|
||
'endpoint': os.getenv('MINIO_ENDPOINT', 'minio.datacubeworld.com:9000'),
|
||
'access_key': os.getenv('MINIO_ACCESS_KEY', 'JOLXFXny3avFSzB0uRA5'),
|
||
'secret_key': os.getenv('MINIO_SECRET_KEY', 'G1BR8jStNfovkfH5ou39EmPl34E4l7dGrnd3Cz0I'),
|
||
'secure': os.getenv('MINIO_SECURE', 'true').lower() == 'true'
|
||
}
|
||
bucket_name = os.getenv('MINIO_BUCKET', 'finyx')
|
||
|
||
client = Minio(
|
||
minio_config['endpoint'],
|
||
access_key=minio_config['access_key'],
|
||
secret_key=minio_config['secret_key'],
|
||
secure=minio_config['secure']
|
||
)
|
||
|
||
# 检查文件是否存在
|
||
stat = client.stat_object(bucket_name, object_name)
|
||
|
||
print_result(True, "文件存在")
|
||
print(f"\n文件信息:")
|
||
print(f" 文件大小: {stat.size} 字节")
|
||
print(f" 最后修改: {stat.last_modified}")
|
||
print(f" 内容类型: {stat.content_type}")
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
print_result(False, f"文件验证失败: {str(e)}")
|
||
return False
|
||
|
||
def main():
|
||
"""主函数"""
|
||
print("\n" + "="*60)
|
||
print(" 文档生成测试 - 初步核实审批表")
|
||
print("="*60)
|
||
|
||
# 1. 生成文档
|
||
file_path = test_document_generation()
|
||
|
||
if not file_path:
|
||
print("\n✗ 文档生成失败，无法继续")
|
||
return
|
||
|
||
# 2. 验证文件存在
|
||
verify_file_exists(file_path)
|
||
|
||
# 3. 生成下载URL
|
||
download_url = generate_download_url(file_path)
|
||
|
||
# 总结
|
||
print_section("测试总结")
|
||
|
||
results = {
|
||
'文档生成': file_path is not None,
|
||
'文件验证': verify_file_exists(file_path) if file_path else False,
|
||
'URL生成': download_url is not None
|
||
}
|
||
|
||
print("\n测试结果：")
|
||
for test_name, success in results.items():
|
||
status = "✓ 通过" if success else "✗ 失败"
|
||
print(f" {test_name}: {status}")
|
||
|
||
if download_url:
|
||
print("\n" + "="*60)
|
||
print(" 下载链接")
|
||
print("="*60)
|
||
print(f"\n{download_url}\n")
|
||
print("="*60)
|
||
print("提示：")
|
||
print(" - 此链接7天内有效")
|
||
print(" - 可以直接在浏览器中打开下载")
|
||
print(" - 文件已成功上传到MinIO")
|
||
print("="*60)
|
||
|
||
if __name__ == '__main__':
|
||
main()
|
||
|