Fixed Bug - Fail to Export as PDF

2025-11-20 00:34:01 +08:00
parent cad25b63c1
commit a89bf7de89
2 changed files with 76 additions and 3 deletions
@@ -0,0 +1,70 @@
+#!/usr/bin/env python
+"""
+PDF导出脚本
+"""
+import json
+import os
+import sys
+from datetime import datetime
+from pathlib import Path
+
+# 添加项目路径到sys.path
+sys.path.insert(0, '/Users/mayiding/Desktop/GitMy/BettaFish')
+
+def export_pdf(ir_file_path):
+    """导出PDF"""
+    try:
+        # 读取IR文件
+        print(f"正在读取报告文件: {ir_file_path}")
+        with open(ir_file_path, 'r', encoding='utf-8') as f:
+            document_ir = json.load(f)
+
+        # 导入PDF渲染器
+        from ReportEngine.renderers.pdf_renderer import PDFRenderer
+
+        # 创建PDF渲染器
+        print("正在初始化PDF渲染器...")
+        renderer = PDFRenderer()
+
+        # 生成PDF
+        print("正在生成PDF...")
+        pdf_bytes = renderer.render_to_bytes(document_ir, optimize_layout=True)
+
+        # 确定输出文件名
+        topic = document_ir.get('metadata', {}).get('topic', 'report')
+        output_dir = Path('/Users/mayiding/Desktop/GitMy/BettaFish/final_reports/pdf')
+        output_dir.mkdir(parents=True, exist_ok=True)
+
+        pdf_filename = f"report_{topic}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf"
+        output_path = output_dir / pdf_filename
+
+        # 保存PDF文件
+        print(f"正在保存PDF到: {output_path}")
+        with open(output_path, 'wb') as f:
+            f.write(pdf_bytes)
+
+        print(f"✅ PDF导出成功！")
+        print(f"文件位置: {output_path}")
+        print(f"文件大小: {len(pdf_bytes) / 1024 / 1024:.2f} MB")
+
+        return str(output_path)
+
+    except Exception as e:
+        print(f"❌ PDF导出失败: {str(e)}")
+        import traceback
+        traceback.print_exc()
+        return None
+
+if __name__ == "__main__":
+    # 使用最新的报告文件
+    latest_report = "/Users/mayiding/Desktop/GitMy/BettaFish/final_reports/ir/report_ir_人工智能行情发展走势_20251119_235407.json"
+
+    if os.path.exists(latest_report):
+        print("="*50)
+        print("开始导出PDF")
+        print("="*50)
+        result = export_pdf(latest_report)
+        if result:
+            print(f"\n📄 PDF文件已生成: {result}")
+    else:
+        print(f"❌ 报告文件不存在: {latest_report}")