#!/usr/bin/env python3
"""
小红书内页渲染引擎 v8 — HTML/CSS + Playwright + 底板背景图
详见 xiaohongshu-automation SKILL.md 中「HTML/CSS + Playwright 渲染方案」章节
"""
import os, base64
from pathlib import Path
from datetime import datetime, timezone, timedelta
CST = timezone(timedelta(hours=8)); TODAY = datetime.now(CST)
BASE = Path(os.path.expanduser("~/project/brand"))
FONT_T = BASE / "YouSheBiaoTiHei-2.ttf"
FONT_B = Path("/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc")
INNER_BG = BASE / "内页.png"; COVER_BG = BASE / "小红书封面.jpeg"
BG_B64 = base64.b64encode(open(INNER_BG,"rb").read()).decode()
TB64 = base64.b64encode(open(FONT_T,"rb").read()).decode()
BB64 = base64.b64encode(open(FONT_B,"rb").read()).decode()
C = {"dark":"#285A37","orange":"#E67E22","text":"#1E3C28","mid":"#649B73","white":"#FFFFFF"}
def page(title, blocks, pn, tp):
badge = f'{pn}/{tp}
' if tp > 1 else ''
return f''
def render(pages_data, output_dir):
"""pages_data: list of (title, blocks_html) tuples. Renders all pages in one Playwright shot."""
from playwright.sync_api import sync_playwright
from PIL import Image
os.makedirs(output_dir, exist_ok=True)
all_pages = ''.join(page(title, blocks, i+1, len(pages_data)) for i, (title, blocks) in enumerate(pages_data))
html = f"""{all_pages}"""
with open("/tmp/v8_render.html", "w", encoding="utf-8") as f:
f.write(html)
with sync_playwright() as p:
browser = p.chromium.launch(headless=True)
page = browser.new_page(viewport={"width": 1242*len(pages_data), "height": 1660})
page.goto("file:///tmp/v8_render.html")
page.screenshot(path="/tmp/v8_full.png")
browser.close()
full = Image.open("/tmp/v8_full.png").convert("RGB")
for i in range(len(pages_data)):
x = i * 1242
full.crop((x, 0, x+1242, 1660)).save(os.path.join(output_dir, f"inner-{i+1:02d}.png"), "PNG", quality=90)
print(f"📄 内页{i+1}/{len(pages_data)}: {(Path(output_dir)/f'inner-{i+1:02d}.png').stat().st_size//1024}KB")
print(f"✅ {output_dir}/")