1#!/usr/bin/env python3
2"""
3小红书帖子长图导出工具
4 
5用法:
6    python3 export-long-image.py --posts '<json>' --output output.jpg
7    python3 export-long-image.py --posts-file posts.json --output output.jpg
8 
9posts JSON 格式:
10[
11    {
12        "title": "帖子标题",
13        "author": "作者名",
14        "stats": "1.3万赞 5171收藏",
15        "desc": "正文摘要，支持\\n换行",
16        "images": ["url1", "url2", ...],
17        "per_image_text": {
18            "1": "第2张图的说明文字（0-indexed）",
19            "3": "第4张图的说明文字"
20        }
21    },
22    ...
23]
24 
25per_image_text 可选：如果原帖文字明确指向某张图，可以把说明放在对应图片上。
26未指定 per_image_text 时，所有文字放在该帖第一张图前的文字块中。
27"""
28 
29import argparse
30import json
31import os
32import sys
33import tempfile
34import urllib.request
35from PIL import Image, ImageDraw, ImageFont
36 
37# --- 配置 ---
38WIDTH = 800
39PAD = 24
40LINE_SPACE = 10
41FONT_CANDIDATES = [
42    "/System/Library/Fonts/STHeiti Medium.ttc",
43    "/System/Library/Fonts/Hiragino Sans GB.ttc",
44    "/System/Library/Fonts/Supplemental/Arial Unicode.ttf",
45    "/usr/share/fonts/truetype/noto/NotoSansCJK-Regular.ttc",
46    "/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc",
47]
48 
49 
50def find_font():
51    for path in FONT_CANDIDATES:
52        if os.path.exists(path):
53            return path
54    return None
55 
56 
57def load_font(path, size):
58    if path:
59        try:
60            return ImageFont.truetype(path, size, index=0)
61        except Exception:
62            pass
63    return ImageFont.load_default()
64 
65 
66def wrap_text(text, font, max_width, draw):
67    lines = []
68    for paragraph in text.split("\n"):
69        paragraph = paragraph.strip()
70        if not paragraph:
71            continue
72        current = ""
73        for char in paragraph:
74            test = current + char
75            bbox = draw.textbbox((0, 0), test, font=font)
76            if bbox[2] - bbox[0] > max_width:
77                if current:
78                    lines.append(current)
79                current = char
80            else:
81                current = test
82        if current:
83            lines.append(current)
84    return lines
85 
86 
87def draw_lines(draw, lines, font, x, y, fill):
88    for line in lines:
89        draw.text((x, y), line, font=font, fill=fill)
90        bbox = draw.textbbox((0, 0), line, font=font)
91        y += (bbox[3] - bbox[1]) + LINE_SPACE
92    return y
93 
94 
95def measure_lines(lines, font, draw):
96    h = 0
97    for line in lines:
98        bbox = draw.textbbox((0, 0), line if line else " ", font=font)
99        h += (bbox[3] - bbox[1]) + LINE_SPACE
100    return h
101 
102 
103def make_text_block(title, author_line, desc, font_path, width):
104    """白底黑字文字块，模仿小红书原样"""
105    title_font = load_font(font_path, 32)
106    author_font = load_font(font_path, 20)
107    body_font = load_font(font_path, 24)
108 
109    tmp = Image.new("RGB", (width, 10))
110    draw = ImageDraw.Draw(tmp)
111    max_w = width - PAD * 2
112 
113    title_lines = wrap_text(title, title_font, max_w, draw)
114    author_lines = [author_line] if author_line else []
115    desc_lines = wrap_text(desc, body_font, max_w, draw) if desc else []
116 
117    # 计算高度
118    total_h = PAD
119    total_h += measure_lines(title_lines, title_font, draw)
120    if author_lines:
121        total_h += 4
122        total_h += measure_lines(author_lines, author_font, draw)
123    if desc_lines:
124        total_h += 8
125        total_h += measure_lines(desc_lines, body_font, draw)
126    total_h += PAD
127 
128    # 绘制
129    block = Image.new("RGB", (width, total_h), (255, 255, 255))
130    draw = ImageDraw.Draw(block)
131 
132    y = PAD
133    y = draw_lines(draw, title_lines, title_font, PAD, y, (33, 33, 33))
134    if author_lines:
135        y += 4
136        y = draw_lines(draw, author_lines, author_font, PAD, y, (153, 153, 153))
137    if desc_lines:
138        y += 8
139        y = draw_lines(draw, desc_lines, body_font, PAD, y, (66, 66, 66))
140 
141    return block
142 
143 
144def make_image_caption(text, font_path, width):
145    """图片上方的小说明文字块"""
146    font = load_font(font_path, 20)
147    tmp = Image.new("RGB", (width, 10))
148    draw = ImageDraw.Draw(tmp)
149    lines = wrap_text(text, font, width - PAD * 2, draw)
150 
151    h = PAD + measure_lines(lines, font, draw) + 8
152    block = Image.new("RGB", (width, h), (245, 245, 245))
153    draw = ImageDraw.Draw(block)
154    draw_lines(draw, lines, font, PAD, PAD // 2, (100, 100, 100))
155    return block
156 
157 
158def download_image(url, tmpdir, idx):
159    """下载图片到临时目录"""
160    ext = ".webp"
161    path = os.path.join(tmpdir, f"img_{idx}{ext}")
162    try:
163        req = urllib.request.Request(url, headers={"User-Agent": "Mozilla/5.0"})
164        with urllib.request.urlopen(req, timeout=30) as resp:
165            with open(path, "wb") as f:
166                f.write(resp.read())
167        return path
168    except Exception as e:
169        print(f"  警告: 下载失败 {url[:60]}... ({e})", file=sys.stderr)
170        return None
171 
172 
173def main():
174    parser = argparse.ArgumentParser(description="小红书帖子长图导出")
175    parser.add_argument("--posts", help="Posts JSON string")
176    parser.add_argument("--posts-file", help="Posts JSON file path")
177    parser.add_argument("--output", "-o", required=True, help="Output JPG path")
178    parser.add_argument("--width", type=int, default=800, help="Image width (default 800)")
179    parser.add_argument("--quality", type=int, default=88, help="JPEG quality (default 88)")
180    args = parser.parse_args()
181 
182    global WIDTH
183    WIDTH = args.width
184 
185    # 读取 posts 数据
186    if args.posts:
187        posts = json.loads(args.posts)
188    elif args.posts_file:
189        with open(args.posts_file, "r") as f:
190            posts = json.load(f)
191    else:
192        print("错误: 需要 --posts 或 --posts-file", file=sys.stderr)
193        sys.exit(1)
194 
195    font_path = find_font()
196    if not font_path:
197        print("警告: 未找到中文字体，文字可能显示异常", file=sys.stderr)
198 
199    sep = Image.new("RGB", (WIDTH, 3), (230, 230, 230))
200    pieces = []
201 
202    with tempfile.TemporaryDirectory() as tmpdir:
203        img_counter = 0
204        for pi, post in enumerate(posts):
205            title = post.get("title", "")
206            author = post.get("author", "")
207            stats = post.get("stats", "")
208            desc = post.get("desc", "")
209            images = post.get("images", [])
210            per_image_text = post.get("per_image_text", {})
211 
212            # 作者行
213            author_line = author
214            if stats:
215                author_line = f"{author} · {stats}" if author else stats
216 
217            # 主文字块
218            text_block = make_text_block(title, author_line, desc, font_path, WIDTH)
219            pieces.append(text_block)
220 
221            # 图片
222            for i, url in enumerate(images):
223                # 是否有针对这张图的说明
224                img_key = str(i)
225                if img_key in per_image_text:
226                    caption_block = make_image_caption(per_image_text[img_key], font_path, WIDTH)
227                    pieces.append(caption_block)
228 
229                img_path = download_image(url, tmpdir, img_counter)
230                img_counter += 1
231                if img_path:
232                    try:
233                        im = Image.open(img_path).convert("RGB")
234                        ratio = WIDTH / im.width
235                        im = im.resize((WIDTH, int(im.height * ratio)), Image.LANCZOS)
236                        pieces.append(im)
237                    except Exception as e:
238                        print(f"  警告: 图片处理失败 ({e})", file=sys.stderr)
239 
240            # 帖子间分隔线
241            if pi < len(posts) - 1:
242                pieces.append(sep)
243 
244    if not pieces:
245        print("错误: 没有内容可拼接", file=sys.stderr)
246        sys.exit(1)
247 
248    total_h = sum(p.height for p in pieces)
249    long_img = Image.new("RGB", (WIDTH, total_h), (255, 255, 255))
250    y = 0
251    for p in pieces:
252        long_img.paste(p, (0, y))
253        y += p.height
254 
255    long_img.save(args.output, "JPEG", quality=args.quality)
256    print(f"完成: {args.output} ({WIDTH}x{total_h})")
257 
258 
259if __name__ == "__main__":
260    main()
261