From 9bb60868d6ff08d5c754fdbda0e20bb5bf65cdb9 Mon Sep 17 00:00:00 2001
From: 732642856 <732642856@qq.com>
Date: Sat, 13 Jun 2026 10:48:16 +0800
Subject: [PATCH] =?UTF-8?q?v6.9.8:=20PDF=20=E6=8A=A5=E5=91=8A=E7=9B=B4?=
 =?UTF-8?q?=E6=8E=A5=E8=BE=93=E5=87=BA=20=E2=80=94=20report=5Fbuilder.py?=
 =?UTF-8?q?=20=E6=94=AF=E6=8C=81=20--pdf=20=E6=A0=87=E5=BF=97?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

feat(pdf): report_builder.py v2.0 — 新增 --pdf 标志，Playwright 渲染 HTML→PDF
feat(pdf): 羊皮纸主题 A4 排版，中英双语封面+目录+分页章节
chore: 页脚版本号更新 v3.5→v6.9.7

用法: python3 scripts/report_builder.py ./report_dir --name 张三 --lagna 狮子座 --pdf
依赖: pip install playwright && playwright install chromium
---
 scripts/report_builder.py | 58 +++++++++++++++++++++++++++++++++------
 1 file changed, 50 insertions(+), 8 deletions(-)

diff --git a/scripts/report_builder.py b/scripts/report_builder.py
index 072e955e..687f1ec9 100644
--- a/scripts/report_builder.py
+++ b/scripts/report_builder.py
@@ -1,16 +1,19 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 """
-Jyotish Report Builder v1.0
-将 Markdown 分析报告转换为精美 HTML（羊皮纸风格）
+Jyotish Report Builder v2.0
+将 Markdown 分析报告转换为精美 HTML + PDF（羊皮纸风格）
 
 基于 CNWU16/vedic-astro-skills 的 report_builder.py 改编
 适配我们的 jyotish-vedic-astrology 引擎输出格式
 
 用法:
-  python3 report_builder.py <folder> --name "名字" --lagna "上升" --lang cn
+  # HTML 输出
   python3 report_builder.py ./report_dir --name "张三" --lagna "狮子座" --lang cn
-  python3 report_builder.py ./report_dir --name "Obama" --lagna "Leo" --lang en
+  # PDF 直接输出（需要 playwright 已安装）
+  python3 report_builder.py ./report_dir --name "张三" --lagna "狮子座" --lang cn --pdf
+  # 英文
+  python3 report_builder.py ./report_dir --name "Obama" --lagna "Leo" --lang en --pdf
 
 功能:
   - 自动扫描目录下的 MD 文件，按章节注册表排序
@@ -18,7 +21,7 @@ Jyotish Report Builder v1.0
   - 自动生成目录
   - 每个章节独立分页（A4打印优化）
   - 羊皮纸主题CSS，支持中英文双语
-  - 可直接浏览器打开后 Ctrl+P → Save as PDF
+  - --pdf: 使用 Playwright 直接生成 PDF（无需手动打印）
 """
 
 import os
@@ -345,6 +348,33 @@ def build_section(num, title, md_text):
 </div>"""
 
 
+def _html_to_pdf(html_path, pdf_path):
+    """Convert HTML to PDF using Playwright headless Chromium."""
+    try:
+        from playwright.sync_api import sync_playwright
+    except ImportError:
+        print("  [WARN] playwright not installed. Install with: pip install playwright && playwright install chromium")
+        print("  [INFO] HTML report available at:", html_path)
+        return False
+
+    print("  Launching headless Chromium...")
+    with sync_playwright() as p:
+        browser = p.chromium.launch(headless=True)
+        page = browser.new_page()
+        page.goto(f"file://{os.path.abspath(html_path)}", wait_until="networkidle")
+        page.pdf(
+            path=pdf_path,
+            format="A4",
+            print_background=True,
+            margin={"top": "22mm", "bottom": "24mm", "left": "20mm", "right": "20mm"},
+        )
+        browser.close()
+
+    size_kb = os.path.getsize(pdf_path) / 1024
+    print(f"  [OK] PDF: {pdf_path} ({size_kb:.0f} KB)")
+    return True
+
+
 def main():
     parser = argparse.ArgumentParser(
         description="Jyotish Report Builder — MD → HTML",
@@ -361,6 +391,7 @@ Examples:
     parser.add_argument("--status", default="—", help="Current status")
     parser.add_argument("--lang", default="cn", choices=["cn", "en"], help="Language (default: cn)")
     parser.add_argument("--output", default=None, help="Output HTML path")
+    parser.add_argument("--pdf", action="store_true", help="Generate PDF directly (requires playwright)")
     args = parser.parse_args()
 
     folder = args.folder.rstrip("/\\")
@@ -406,11 +437,11 @@ Examples:
     footer_cn = """<div class="footer-note">
   本报告基于传统吠陀占星方法（Parashari Jyotish | KN Rao School）。<br>
   每项结论均有量化行星指标支撑。仅供自我反思与战略思考参考。<br>
-  Powered by Jyotish Engine v3.5 &amp; Swiss Ephemeris</div>"""
+  Powered by Jyotish Engine v6.9.7 &amp; Swiss Ephemeris</div>"""
     footer_en = """<div class="footer-note">
   Generated using traditional Vedic astrological methods (Parashari Jyotish | KN Rao School).<br>
   Every claim backed by quantified planetary metrics. For self-reflection purposes only.<br>
-  Powered by Jyotish Engine v3.5 &amp; Swiss Ephemeris</div>"""
+  Powered by Jyotish Engine v6.9.7 &amp; Swiss Ephemeris</div>"""
     footer = footer_cn if lang == "cn" else footer_en
 
     html_lang = "zh-CN" if lang == "cn" else "en"
@@ -427,7 +458,18 @@ Examples:
 
     size = os.path.getsize(out) / 1024
     print(f"\n  [OK] Output: {out} ({size:.0f} KB)")
-    print(f"  -> Open in browser -> Ctrl+P -> Save as PDF")
+
+    # PDF generation
+    if args.pdf:
+        pdf_out = os.path.splitext(out)[0] + ".pdf"
+        ok = _html_to_pdf(out, pdf_out)
+        if ok:
+            print(f"  -> PDF ready: {pdf_out}")
+        else:
+            print(f"  -> HTML report: {out} (open in browser → Ctrl+P → Save as PDF)")
+    else:
+        print(f"  -> Open in browser -> Ctrl+P -> Save as PDF")
+        print(f"  -> Or re-run with --pdf for direct PDF generation")
 
 
 if __name__ == "__main__":