#!/usr/bin/env python3 """ 小红书帖子长图导出工具 用法: python3 export-long-image.py --posts '' --output output.jpg python3 export-long-image.py --posts-file posts.json --output output.jpg posts JSON 格式: [ { "title": "帖子标题", "author": "作者名", "stats": "1.3万赞 5171收藏", "desc": "正文摘要，支持\\n换行", "images": ["url1", "url2", ...], "per_image_text": { "1": "第2张图的说明文字（0-indexed）", "3": "第4张图的说明文字" } }, ... ] per_image_text 可选：如果原帖文字明确指向某张图，可以把说明放在对应图片上。 未指定 per_image_text 时，所有文字放在该帖第一张图前的文字块中。 """ import argparse import json import os import sys import tempfile import urllib.request from PIL import Image, ImageDraw, ImageFont # --- 配置 --- WIDTH = 800 PAD = 24 LINE_SPACE = 10 FONT_CANDIDATES = [ "/System/Library/Fonts/STHeiti Medium.ttc", "/System/Library/Fonts/Hiragino Sans GB.ttc", "/System/Library/Fonts/Supplemental/Arial Unicode.ttf", "/usr/share/fonts/truetype/noto/NotoSansCJK-Regular.ttc", "/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc", ] def find_font(): for path in FONT_CANDIDATES: if os.path.exists(path): return path return None def load_font(path, size): if path: try: return ImageFont.truetype(path, size, index=0) except Exception: pass return ImageFont.load_default() def wrap_text(text, font, max_width, draw): lines = [] for paragraph in text.split("\n"): paragraph = paragraph.strip() if not paragraph: continue current = "" for char in paragraph: test = current + char bbox = draw.textbbox((0, 0), test, font=font) if bbox[2] - bbox[0] > max_width: if current: lines.append(current) current = char else: current = test if current: lines.append(current) return lines def draw_lines(draw, lines, font, x, y, fill): for line in lines: draw.text((x, y), line, font=font, fill=fill) bbox = draw.textbbox((0, 0), line, font=font) y += (bbox[3] - bbox[1]) + LINE_SPACE return y def measure_lines(lines, font, draw): h = 0 for line in lines: bbox = draw.textbbox((0, 0), line if line else " ", font=font) h += (bbox[3] - bbox[1]) + LINE_SPACE return h def make_text_block(title, author_line, desc, font_path, width): """白底黑字文字块，模仿小红书原样""" title_font = load_font(font_path, 32) author_font = load_font(font_path, 20) body_font = load_font(font_path, 24) tmp = Image.new("RGB", (width, 10)) draw = ImageDraw.Draw(tmp) max_w = width - PAD * 2 title_lines = wrap_text(title, title_font, max_w, draw) author_lines = [author_line] if author_line else [] desc_lines = wrap_text(desc, body_font, max_w, draw) if desc else [] # 计算高度 total_h = PAD total_h += measure_lines(title_lines, title_font, draw) if author_lines: total_h += 4 total_h += measure_lines(author_lines, author_font, draw) if desc_lines: total_h += 8 total_h += measure_lines(desc_lines, body_font, draw) total_h += PAD # 绘制 block = Image.new("RGB", (width, total_h), (255, 255, 255)) draw = ImageDraw.Draw(block) y = PAD y = draw_lines(draw, title_lines, title_font, PAD, y, (33, 33, 33)) if author_lines: y += 4 y = draw_lines(draw, author_lines, author_font, PAD, y, (153, 153, 153)) if desc_lines: y += 8 y = draw_lines(draw, desc_lines, body_font, PAD, y, (66, 66, 66)) return block def make_image_caption(text, font_path, width): """图片上方的小说明文字块""" font = load_font(font_path, 20) tmp = Image.new("RGB", (width, 10)) draw = ImageDraw.Draw(tmp) lines = wrap_text(text, font, width - PAD * 2, draw) h = PAD + measure_lines(lines, font, draw) + 8 block = Image.new("RGB", (width, h), (245, 245, 245)) draw = ImageDraw.Draw(block) draw_lines(draw, lines, font, PAD, PAD // 2, (100, 100, 100)) return block def download_image(url, tmpdir, idx): """下载图片到临时目录""" ext = ".webp" path = os.path.join(tmpdir, f"img_{idx}{ext}") try: req = urllib.request.Request(url, headers={"User-Agent": "Mozilla/5.0"}) with urllib.request.urlopen(req, timeout=30) as resp: with open(path, "wb") as f: f.write(resp.read()) return path except Exception as e: print(f" 警告: 下载失败 {url[:60]}... ({e})", file=sys.stderr) return None def main(): parser = argparse.ArgumentParser(description="小红书帖子长图导出") parser.add_argument("--posts", help="Posts JSON string") parser.add_argument("--posts-file", help="Posts JSON file path") parser.add_argument("--output", "-o", required=True, help="Output JPG path") parser.add_argument("--width", type=int, default=800, help="Image width (default 800)") parser.add_argument("--quality", type=int, default=88, help="JPEG quality (default 88)") args = parser.parse_args() global WIDTH WIDTH = args.width # 读取 posts 数据 if args.posts: posts = json.loads(args.posts) elif args.posts_file: with open(args.posts_file, "r") as f: posts = json.load(f) else: print("错误: 需要 --posts 或 --posts-file", file=sys.stderr) sys.exit(1) font_path = find_font() if not font_path: print("警告: 未找到中文字体，文字可能显示异常", file=sys.stderr) sep = Image.new("RGB", (WIDTH, 3), (230, 230, 230)) pieces = [] with tempfile.TemporaryDirectory() as tmpdir: img_counter = 0 for pi, post in enumerate(posts): title = post.get("title", "") author = post.get("author", "") stats = post.get("stats", "") desc = post.get("desc", "") images = post.get("images", []) per_image_text = post.get("per_image_text", {}) # 作者行 author_line = author if stats: author_line = f"{author} · {stats}" if author else stats # 主文字块 text_block = make_text_block(title, author_line, desc, font_path, WIDTH) pieces.append(text_block) # 图片 for i, url in enumerate(images): # 是否有针对这张图的说明 img_key = str(i) if img_key in per_image_text: caption_block = make_image_caption(per_image_text[img_key], font_path, WIDTH) pieces.append(caption_block) img_path = download_image(url, tmpdir, img_counter) img_counter += 1 if img_path: try: im = Image.open(img_path).convert("RGB") ratio = WIDTH / im.width im = im.resize((WIDTH, int(im.height * ratio)), Image.LANCZOS) pieces.append(im) except Exception as e: print(f" 警告: 图片处理失败 ({e})", file=sys.stderr) # 帖子间分隔线 if pi < len(posts) - 1: pieces.append(sep) if not pieces: print("错误: 没有内容可拼接", file=sys.stderr) sys.exit(1) total_h = sum(p.height for p in pieces) long_img = Image.new("RGB", (WIDTH, total_h), (255, 255, 255)) y = 0 for p in pieces: long_img.paste(p, (0, y)) y += p.height long_img.save(args.output, "JPEG", quality=args.quality) print(f"完成: {args.output} ({WIDTH}x{total_h})") if __name__ == "__main__": main()