86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
测试 report_generator.py 的保存逻辑
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import sys
|
|||
|
|
import os
|
|||
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|||
|
|
|
|||
|
|
from lib.report_generator import save_report_history
|
|||
|
|
from lib.db import execute_query
|
|||
|
|
|
|||
|
|
def test_save_logic():
|
|||
|
|
"""测试保存逻辑"""
|
|||
|
|
print("="*60)
|
|||
|
|
print("测试 report_generator 的保存逻辑")
|
|||
|
|
print("="*60)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 1. 保存测试记录
|
|||
|
|
print("\n1. 调用 save_report_history...")
|
|||
|
|
test_file_path = '/reports/test_from_generator.xlsx'
|
|||
|
|
|
|||
|
|
print(f" 准备保存:")
|
|||
|
|
print(f" - file_path: '{test_file_path}'")
|
|||
|
|
print(f" - status: 1")
|
|||
|
|
|
|||
|
|
history_id = save_report_history(
|
|||
|
|
report_date='2025-01-01',
|
|||
|
|
start_time='2025-01-01 08:00:00',
|
|||
|
|
end_time='2025-01-02 08:00:00',
|
|||
|
|
split_type='test',
|
|||
|
|
split_value='test_value',
|
|||
|
|
split_name='测试名称',
|
|||
|
|
config_id=999999,
|
|||
|
|
total_orders=10,
|
|||
|
|
total_amount=100.50,
|
|||
|
|
sum_results={'test_field': 50.0},
|
|||
|
|
file_path=test_file_path,
|
|||
|
|
status=1
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
print(f" ✓ 保存成功，ID: {history_id}")
|
|||
|
|
|
|||
|
|
# 2. 查询记录
|
|||
|
|
print("\n2. 查询刚保存的记录...")
|
|||
|
|
query_sql = "SELECT * FROM t_daily_report_history WHERE id = %s"
|
|||
|
|
result = execute_query(query_sql, (history_id,))
|
|||
|
|
|
|||
|
|
if result and len(result) > 0:
|
|||
|
|
record = result[0]
|
|||
|
|
print(" ✓ 查询成功")
|
|||
|
|
print(f" - ID: {record.get('id')}")
|
|||
|
|
print(f" - file_path: '{record.get('file_path')}'")
|
|||
|
|
print(f" - status: {record.get('status')}")
|
|||
|
|
|
|||
|
|
if record.get('file_path') == test_file_path:
|
|||
|
|
print("\n✓✓✓ save_report_history 工作正常！")
|
|||
|
|
else:
|
|||
|
|
print(f"\n❌❌❌ file_path 不匹配！")
|
|||
|
|
print(f" 期望: '{test_file_path}'")
|
|||
|
|
print(f" 实际: '{record.get('file_path')}'")
|
|||
|
|
else:
|
|||
|
|
print(" ❌ 查询失败，没有找到数据")
|
|||
|
|
|
|||
|
|
# 3. 清理测试数据
|
|||
|
|
print("\n3. 清理测试数据...")
|
|||
|
|
from lib.db import execute_update
|
|||
|
|
execute_update("DELETE FROM t_daily_report_history WHERE id = %s", (history_id,))
|
|||
|
|
print(" ✓ 已清理")
|
|||
|
|
|
|||
|
|
print("\n" + "="*60)
|
|||
|
|
print("测试完成")
|
|||
|
|
print("="*60)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"\n❌ 测试失败: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
|
|||
|
|
input("\n按回车键退出...")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
test_save_logic()
|