195 lines
5.7 KiB
Python
195 lines
5.7 KiB
Python
"""
|
||
使用新的SVG矢量图表功能重新生成最新报告的PDF
|
||
"""
|
||
|
||
import json
|
||
import sys
|
||
from pathlib import Path
|
||
from datetime import datetime
|
||
from loguru import logger
|
||
|
||
# 添加项目路径
|
||
sys.path.insert(0, str(Path(__file__).parent))
|
||
|
||
from ReportEngine.renderers import PDFRenderer
|
||
|
||
def find_latest_report():
|
||
"""
|
||
在 `final_reports/ir` 中查找最新的报告 IR JSON。
|
||
|
||
按修改时间倒序选择第一条，若目录或文件缺失则记录错误并返回 None。
|
||
|
||
返回:
|
||
Path | None: 最新 IR 文件路径；未找到则为 None。
|
||
"""
|
||
ir_dir = Path("final_reports/ir")
|
||
|
||
if not ir_dir.exists():
|
||
logger.error(f"报告目录不存在: {ir_dir}")
|
||
return None
|
||
|
||
# 获取所有JSON文件并按修改时间排序
|
||
json_files = sorted(ir_dir.glob("*.json"), key=lambda x: x.stat().st_mtime, reverse=True)
|
||
|
||
if not json_files:
|
||
logger.error("未找到报告文件")
|
||
return None
|
||
|
||
latest_file = json_files[0]
|
||
logger.info(f"找到最新报告: {latest_file.name}")
|
||
|
||
return latest_file
|
||
|
||
def load_document_ir(file_path):
|
||
"""
|
||
读取指定路径的 Document IR JSON，并统计章节/图表数量。
|
||
|
||
解析失败时返回 None；成功时会打印章节数与图表数，便于确认
|
||
输入报告的规模。
|
||
|
||
参数:
|
||
file_path: IR 文件路径
|
||
|
||
返回:
|
||
dict | None: 解析后的 Document IR；失败返回 None。
|
||
"""
|
||
try:
|
||
with open(file_path, 'r', encoding='utf-8') as f:
|
||
document_ir = json.load(f)
|
||
|
||
logger.info(f"成功加载报告: {file_path.name}")
|
||
|
||
# 统计图表数量
|
||
chart_count = 0
|
||
chapters = document_ir.get('chapters', [])
|
||
|
||
def count_charts(blocks):
|
||
"""递归统计 block 列表中的 Chart.js 图表数量"""
|
||
count = 0
|
||
for block in blocks:
|
||
if isinstance(block, dict):
|
||
if block.get('type') == 'widget' and block.get('widgetType', '').startswith('chart.js'):
|
||
count += 1
|
||
# 递归处理嵌套blocks
|
||
nested = block.get('blocks')
|
||
if isinstance(nested, list):
|
||
count += count_charts(nested)
|
||
return count
|
||
|
||
for chapter in chapters:
|
||
blocks = chapter.get('blocks', [])
|
||
chart_count += count_charts(blocks)
|
||
|
||
logger.info(f"报告包含 {len(chapters)} 个章节，{chart_count} 个图表")
|
||
|
||
return document_ir
|
||
|
||
except Exception as e:
|
||
logger.error(f"加载报告失败: {e}")
|
||
return None
|
||
|
||
def generate_pdf_with_vector_charts(document_ir, output_path):
|
||
"""
|
||
使用 PDFRenderer 将 Document IR 渲染为包含 SVG 矢量图表的 PDF。
|
||
|
||
启用布局优化，生成后输出文件大小与成功提示；异常时返回 None。
|
||
|
||
参数:
|
||
document_ir: 完整的 Document IR
|
||
output_path: 目标 PDF 路径
|
||
|
||
返回:
|
||
Path | None: 成功时返回生成的 PDF 路径，失败返回 None。
|
||
"""
|
||
try:
|
||
logger.info("=" * 60)
|
||
logger.info("开始生成PDF（带矢量图表）")
|
||
logger.info("=" * 60)
|
||
|
||
# 创建PDF渲染器
|
||
renderer = PDFRenderer()
|
||
|
||
# 渲染PDF
|
||
result_path = renderer.render_to_pdf(
|
||
document_ir,
|
||
output_path,
|
||
optimize_layout=True
|
||
)
|
||
|
||
logger.info("=" * 60)
|
||
logger.info(f"✓ PDF生成成功: {result_path}")
|
||
logger.info("=" * 60)
|
||
|
||
# 显示文件大小
|
||
file_size = result_path.stat().st_size
|
||
size_mb = file_size / (1024 * 1024)
|
||
logger.info(f"文件大小: {size_mb:.2f} MB")
|
||
|
||
return result_path
|
||
|
||
except Exception as e:
|
||
logger.error(f"生成PDF失败: {e}", exc_info=True)
|
||
return None
|
||
|
||
def main():
|
||
"""
|
||
主入口：重新生成最新报告的矢量 PDF。
|
||
|
||
步骤：
|
||
1) 查找最新 IR 文件；
|
||
2) 读取并统计报告结构；
|
||
3) 构造输出文件名并确保目录存在；
|
||
4) 调用渲染函数生成 PDF，输出路径与特性说明。
|
||
|
||
返回:
|
||
int: 0 表示成功，非 0 表示失败。
|
||
"""
|
||
logger.info("🚀 使用SVG矢量图表重新生成最新报告的PDF")
|
||
logger.info("")
|
||
|
||
# 1. 找到最新报告
|
||
latest_report = find_latest_report()
|
||
if not latest_report:
|
||
logger.error("未找到报告文件")
|
||
return 1
|
||
|
||
# 2. 加载报告数据
|
||
document_ir = load_document_ir(latest_report)
|
||
if not document_ir:
|
||
logger.error("加载报告失败")
|
||
return 1
|
||
|
||
# 3. 生成输出文件名
|
||
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||
report_name = latest_report.stem.replace("report_ir_", "")
|
||
output_filename = f"report_vector_{report_name}_{timestamp}.pdf"
|
||
output_path = Path("final_reports/pdf") / output_filename
|
||
|
||
# 确保输出目录存在
|
||
output_path.parent.mkdir(parents=True, exist_ok=True)
|
||
|
||
logger.info(f"输出路径: {output_path}")
|
||
logger.info("")
|
||
|
||
# 4. 生成PDF
|
||
result = generate_pdf_with_vector_charts(document_ir, output_path)
|
||
|
||
if result:
|
||
logger.info("")
|
||
logger.info("🎉 PDF生成完成！")
|
||
logger.info("")
|
||
logger.info("特性说明:")
|
||
logger.info(" ✓ 图表以SVG矢量格式渲染")
|
||
logger.info(" ✓ 支持无限缩放不失真")
|
||
logger.info(" ✓ 保留完整的图表视觉效果")
|
||
logger.info(" ✓ 折线图、柱状图、饼图等均为矢量曲线")
|
||
logger.info("")
|
||
logger.info(f"PDF文件位置: {result.absolute()}")
|
||
return 0
|
||
else:
|
||
logger.error("❌ PDF生成失败")
|
||
return 1
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|