#!/usr/bin/env python3 """ wechat_formatter.py — Markdown → 公众号排版 HTML 两种模式: 1. 基础模式（免费）：mistune 转 HTML，公众号适配样式 2. AI 模式（需 API key）：AI 智能优化排版结构、段落节奏、视觉装饰 Usage: # 基础模式 python3 wechat_formatter.py input.md -o output.html # AI 模式（需要 OPENAI_BASE_URL + OPENAI_API_KEY） python3 wechat_formatter.py input.md -o output.html --ai # 指定样式主题 python3 wechat_formatter.py input.md -o output.html --theme clean """ import argparse import json import os import re import sys import urllib.request from pathlib import Path try: import mistune except ImportError: print("请先安装 mistune: pip3 install mistune") sys.exit(1) # mistune 插件 def _plugins(): return ["table", "strikethrough"] # ── 主题样式 ── THEMES = { "default": { "wrap": "max-width:100%;padding:0;margin:0;font-family:-apple-system,BlinkMacSystemFont,'Segoe UI','PingFang SC','Hiragino Sans GB','Microsoft YaHei',sans-serif;", "p": "font-size:16px;line-height:1.75;color:#333;margin:0 0 1em 0;letter-spacing:0.5px;", "h1": "font-size:22px;font-weight:bold;color:#1a1a1a;margin:1.2em 0 0.6em 0;line-height:1.4;", "h2": "font-size:20px;font-weight:bold;color:#1a1a1a;margin:1em 0 0.5em 0;line-height:1.4;border-bottom:2px solid #eee;padding-bottom:0.3em;", "h3": "font-size:18px;font-weight:bold;color:#1a1a1a;margin:0.8em 0 0.4em 0;line-height:1.4;", "blockquote": "border-left:4px solid #ddd;padding:8px 16px;margin:1em 0;color:#666;font-size:15px;background:#f9f9f9;", "code_inline": "background:#f5f5f5;padding:2px 6px;border-radius:3px;font-family:Menlo,Consolas,monospace;font-size:14px;color:#c7254e;", "code_block": "background:#f5f5f5;padding:16px;border-radius:6px;margin:1em 0;font-family:Menlo,Consolas,monospace;font-size:13px;line-height:1.6;overflow-x:auto;white-space:pre-wrap;word-wrap:break-word;color:#333;", "img": "max-width:100%;height:auto;display:block;margin:1em auto;border-radius:4px;", "link": "color:#576b95;text-decoration:none;", "ul": "font-size:16px;line-height:1.75;color:#333;margin:0 0 1em 0;padding-left:2em;", "li": "margin:0.3em 0;", "hr": "border:none;border-top:1px solid #ddd;margin:2em 0;", "strong": "color:#1a1a1a;", "em": "color:#555;font-style:italic;", }, "clean": { "wrap": "max-width:677px;padding:20px;margin:0 auto;font-family:-apple-system,BlinkMacSystemFont,'Segoe UI','PingFang SC','Hiragino Sans GB','Microsoft YaHei',sans-serif;background:#fff;", "p": "font-size:15px;line-height:2;color:#3f3f3f;margin:0 0 1.2em 0;letter-spacing:0.5px;", "h1": "font-size:20px;font-weight:bold;color:#111;margin:1.5em 0 0.8em 0;text-align:center;", "h2": "font-size:17px;font-weight:bold;color:#111;margin:1.5em 0 0.6em 0;", "h3": "font-size:16px;font-weight:bold;color:#111;margin:1.2em 0 0.5em 0;", "blockquote": "border-left:3px solid #ccc;padding:10px 16px;margin:1em 0;color:#777;font-size:14px;background:#fafafa;border-radius:0 4px 4px 0;", "code_inline": "background:#f0f0f0;padding:2px 6px;border-radius:3px;font-family:Menlo,Consolas,monospace;font-size:13px;color:#d14;", "code_block": "background:#f8f8f8;padding:16px;border-radius:6px;margin:1em 0;font-family:Menlo,Consolas,monospace;font-size:13px;line-height:1.6;overflow-x:auto;white-space:pre-wrap;word-wrap:break-word;color:#333;", "img": "max-width:100%;height:auto;display:block;margin:1.5em auto;border-radius:4px;", "link": "color:#576b95;text-decoration:none;", "ul": "font-size:15px;line-height:2;color:#3f3f3f;margin:0 0 1.2em 0;padding-left:1.5em;", "li": "margin:0.3em 0;", "hr": "border:none;height:1px;background:linear-gradient(to right,transparent,#ddd,transparent);margin:2em 0;", "strong": "color:#111;font-weight:600;", "em": "color:#777;font-style:italic;", }, "warm": { "wrap": "max-width:100%;padding:0;margin:0;font-family:'Georgia','Songti SC','PingFang SC',serif;", "p": "font-size:16px;line-height:2;color:#4a4a4a;margin:0 0 1.2em 0;letter-spacing:1px;", "h1": "font-size:22px;font-weight:bold;color:#2c2c2c;margin:1.2em 0 0.6em 0;line-height:1.4;text-align:center;", "h2": "font-size:19px;font-weight:bold;color:#2c2c2c;margin:1.2em 0 0.5em 0;", "h3": "font-size:17px;font-weight:bold;color:#2c2c2c;margin:1em 0 0.4em 0;", "blockquote": "border-left:4px solid #e8d5b7;padding:10px 16px;margin:1em 0;color:#8b7355;font-size:15px;background:#fdf8f0;border-radius:0 4px 4px 0;", "code_inline": "background:#fdf8f0;padding:2px 6px;border-radius:3px;font-family:Menlo,Consolas,monospace;font-size:14px;color:#8b6914;", "code_block": "background:#fdf8f0;padding:16px;border-radius:6px;margin:1em 0;font-family:Menlo,Consolas,monospace;font-size:13px;line-height:1.6;overflow-x:auto;white-space:pre-wrap;word-wrap:break-word;color:#4a4a4a;", "img": "max-width:100%;height:auto;display:block;margin:1.5em auto;border-radius:4px;", "link": "color:#8b6914;text-decoration:none;border-bottom:1px solid #e8d5b7;", "ul": "font-size:16px;line-height:2;color:#4a4a4a;margin:0 0 1.2em 0;padding-left:1.5em;", "li": "margin:0.3em 0;", "hr": "border:none;text-align:center;margin:2em 0;color:#e8d5b7;font-size:14px;letter-spacing:8px;", "strong": "color:#2c2c2c;", "em": "color:#8b7355;font-style:italic;", }, } # ── 基础渲染器 ── class WeChatRenderer(mistune.renderers.html.HTMLRenderer): NAME = "wechat" t = THEMES["default"] def paragraph(self, text): return f'

{text}

\n' def heading(self, text, level, **kw): s = self.t.get(f"h{level}", self.t["h3"]) return f'{text}\n' def block_quote(self, text): return f'

{text}

\n' def block_code(self, code, info=None): from html import escape return f'

{escape(code)}

\n' def codespan(self, text): return f'{text}' def image(self, text, url, title=None): return f' {text}

' def link(self, text, url, title=None): return f'{text}' def list(self, text, ordered, **kw): tag = "ol" if ordered else "ul" return f'<{tag} style="{self.t["ul"]}">{text}\n' def list_item(self, text): return f'

{text}

\n' def thematic_break(self): return f'

\n' def strong(self, text): return f'{text}' def emphasis(self, text): return f'{text}' # ── AI 智能排版 ── def ai_enhance_md(md_text: str, api_base: str, api_key: str, model: str = "gpt-4o-mini") -> str: """用 AI 优化 markdown 排版结构，不改变内容只优化形式。""" system_prompt = """你是一个公众号排版优化专家。用户会给你一篇 markdown 文章，你需要做以下优化（只优化排版，不改变任何内容）： 1. 在关键转折处插入分割线（---） 2. 在金句/重点句前后加空行，让它们「呼吸」 3. 如果有超过5行的大段落，适当拆分 4. 在开头或结尾可适当加入装饰性分割线 5. 不要添加、删除或修改任何文字内容 6. 保持所有图片标记 [图片：xxx] 不变 7. 输出纯 markdown，不要加任何解释只输出优化后的 markdown。""" url = f"{api_base.rstrip('/')}/chat/completions" payload = json.dumps({ "model": model, "messages": [ {"role": "system", "content": system_prompt}, {"role": "user", "content": md_text} ], "temperature": 0.3, }).encode("utf-8") req = urllib.request.Request(url, data=payload, headers={ "Content-Type": "application/json", "Authorization": f"Bearer {api_key}", }) try: resp = urllib.request.urlopen(req, timeout=60) result = json.loads(resp.read()) return result["choices"][0]["message"]["content"] except Exception as e: print(f"AI 排版失败: {e}") return md_text # ── 主逻辑 ── def md_to_html(md_text: str, theme: str = "default") -> str: t = THEMES.get(theme, THEMES["default"]) r = WeChatRenderer() r.t = t m = mistune.create_markdown(renderer=r, plugins=_plugins()) return f'

{m(md_text)}

' REFERRAL_URL = "https://derouter.ai?ref=5E79A927" def format_article(input_path: str, output_path: str, theme: str = "default", ai: bool = False, api_base: str = "", api_key: str = "", model: str = "gpt-4o-mini"): md_text = Path(input_path).read_text("utf-8") if ai: if not api_key: api_key = os.environ.get("OPENAI_API_KEY", "") if not api_base: api_base = os.environ.get("OPENAI_BASE_URL", "") if not api_key: print(f"⚠️ AI 排版需要 API key，基础排版模式已启用") print(f"💡 获取 API key: {REFERRAL_URL}") print(f" 注册后设置: export OPENAI_API_KEY=你的key OPENAI_BASE_URL=https://api.derouter.ai/openai/v1") else: if not api_base: api_base = "https://api.openai.com/v1" print("🤖 AI 智能排版中...") md_text = ai_enhance_md(md_text, api_base, api_key, model) print("✅ AI 排版完成") html = md_to_html(md_text, theme) # 页脚 footer = '

排版由 wechat-formatter 生成

' html = html.replace("", f"{footer}") Path(output_path).write_text(html, "utf-8") print(f"📄 HTML 已保存: {output_path}") # 基础模式也提示 AI 模式 if not ai: print(f"💡 提示: 加 --ai 参数可启用 AI 智能排版，需 API key → {REFERRAL_URL}") elif not api_key: print(f"💡 想要 AI 智能排版？获取 API key: {REFERRAL_URL}") return html def main(): parser = argparse.ArgumentParser(description="Markdown → 公众号排版 HTML") parser.add_argument("input", help="输入 markdown 文件路径") parser.add_argument("-o", "--output", default="", help="输出 HTML 文件路径（默认与输入同目录）") parser.add_argument("--theme", default="default", choices=THEMES.keys(), help="排版主题: default, clean, warm") parser.add_argument("--ai", action="store_true", help="启用 AI 智能排版优化") parser.add_argument("--model", default="gpt-4o-mini", help="AI 模型名称") args = parser.parse_args() output = args.output or str(Path(args.input).with_suffix(".html")) api_base = os.environ.get("OPENAI_BASE_URL", "") api_key = os.environ.get("OPENAI_API_KEY", "") format_article(args.input, output, args.theme, args.ai, api_base, api_key, args.model) if __name__ == "__main__": main()

wechat-formatter

Install this skill

Comments