#!/usr/bin/env python3 from __future__ import annotations import argparse import html import json import subprocess import time from pathlib import Path CHROME = Path("/Applications/Google Chrome.app/Contents/MacOS/Google Chrome") def main() -> int: parser = argparse.ArgumentParser(description="Render an exact local fallback infographic from a frozen SKU manifest.") parser.add_argument("--manifest", required=True, type=Path) parser.add_argument("--source-dir", required=True, type=Path) parser.add_argument("--output", required=True, type=Path) parser.add_argument("--method", default="local-controlled-fallback") parser.add_argument("--reason", default="NotebookLM infographic creation rate limit persisted after documented retries") args = parser.parse_args() if not CHROME.exists(): raise FileNotFoundError(CHROME) data = json.loads(args.manifest.read_text(encoding="utf-8")) flow = data["main_flow"] count = len(flow) cards = [] for index, step in enumerate(flow, 1): cards.append( f'

{index:02d}

' f'

{index}

{html.escape(step)}

' f'

标准主流程 · STEP {index:02d}

' ) values = "".join(f'{html.escape(value)}' for value in data["customer_values"]) page = f'''

CPT

{html.escape(data.get('release_name', data['sku'] + '-' + data['name']))}主流程

{html.escape(data['one_liner'])}

{html.escape(data.get('release_code', data['sku']))} · {count} 步标准闭环

{''.join(cards)}

''' draft = args.source_dir / "04-生成成果" / f"{data['sku']}-主流程图-本地受控渲染.html" draft.parent.mkdir(parents=True, exist_ok=True) draft.write_text(page, encoding="utf-8") args.output.parent.mkdir(parents=True, exist_ok=True) profile = args.source_dir / "04-生成成果" / "chrome-profile" process = subprocess.Popen([ str(CHROME), "--headless=new", "--disable-gpu", "--hide-scrollbars", "--no-first-run", "--disable-background-networking", "--run-all-compositor-stages-before-draw", "--virtual-time-budget=1200", f"--user-data-dir={profile}", "--window-size=2752,1536", "--force-device-scale-factor=1", f"--screenshot={args.output}", draft.resolve().as_uri(), ], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True) deadline = time.time() + 20 while time.time() < deadline: if args.output.exists() and args.output.stat().st_size > 10000: break if process.poll() is not None: break time.sleep(0.25) if process.poll() is None: process.terminate() try: process.wait(timeout=3) except subprocess.TimeoutExpired: process.kill() process.wait(timeout=3) if not args.output.exists() or args.output.stat().st_size <= 10000: stdout, stderr = process.communicate() raise RuntimeError(f"Chrome did not create a valid screenshot: {stderr or stdout}") receipt = { "sku": data["sku"], "generation_method": args.method, "reason": args.reason, "output": str(args.output), "html_source": str(draft), "main_flow": flow, } (args.source_dir / "local-fallback-result.json").write_text(json.dumps(receipt, ensure_ascii=False, indent=2), encoding="utf-8") print(json.dumps(receipt, ensure_ascii=False)) return 0 if __name__ == "__main__": raise SystemExit(main())