#!/usr/bin/env python3
"""
PDF 导出脚本 V2 - 使用 markdown + weasyprint
支持完整 Markdown 格式，更好的中文渲染
用法: python3 export_pdf_v2.py --title "标题" --content "内容" --output "输出路径"
"""
import sys
import os
import argparse
# 添加依赖路径
sys.path.insert(0, '/workspace/.local/lib/python3.11/site-packages')
import markdown
from weasyprint import HTML, CSS
def get_default_font():
"""获取系统默认中文字体"""
font_paths = [
# Windows
('C:/Windows/Fonts/msyh.ttc', 'Microsoft YaHei'),
('C:/Windows/Fonts/simhei.ttf', 'SimHei'),
('C:/Windows/Fonts/simsun.ttc', 'SimSun'),
('C:/Windows/Fonts/arialuni.ttf', 'Arial Unicode MS'),
# Linux
('/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc', 'WenQuanYi Zen Hei'),
('/usr/share/fonts/truetype/noto/NotoSansCJK-Regular.ttc', 'Noto Sans CJK'),
# macOS
('/System/Library/Fonts/PingFang.ttc', 'PingFang SC'),
('/Library/Fonts/Arial Unicode.ttf', 'Arial Unicode MS'),
]
for path, name in font_paths:
if os.path.exists(path):
return name
return 'sans-serif'
def export_pdf(title: str, content: str, output_path: str, data_source: str = "") -> str:
"""将 Markdown 内容导出为 PDF"""
# 获取中文字体
chinese_font = get_default_font()
# 获取当前时间
from datetime import datetime
generate_time = datetime.now().strftime("%Y-%m-%d %H:%M")
# Markdown 转 HTML
md = markdown.Markdown(extensions=[
'tables',
'fenced_code',
'toc',
'nl2br',
])
html_body = md.convert(content)
# 处理 H3 标题：如果已有序号（如 "1. "、"2. "），添加标记
import re
def mark_numbered_h3(match):
h3_content = match.group(1)
# 检查是否以数字+点开头
if re.match(r'^\d+\.', h3_content.strip()):
return f'{h3_content}
'
return match.group(0)
html_body = re.sub(r'(.*?)
', mark_numbered_h3, html_body, flags=re.DOTALL)
# 构建完整 HTML 文档 - 参考样式：蓝色标题、简洁专业
html_template = f'''
{title}
{title}
{html_body}
报告生成时间：{generate_time}
{f'
数据来源：{data_source}
' if data_source else '
数据来源：系统自动生成
'}
'''
# 确保输出目录存在
os.makedirs(os.path.dirname(os.path.abspath(output_path)), exist_ok=True)
# HTML 转 PDF
HTML(string=html_template).write_pdf(output_path)
return output_path
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='导出 PDF 文档（支持完整 Markdown）')
parser.add_argument('--title', required=True, help='文档标题')
parser.add_argument('--content', required=True, help='文档正文（Markdown 格式）')
parser.add_argument('--output', required=True, help='输出 .pdf 文件路径')
parser.add_argument('--data-source', default='', help='数据来源说明（可选）')
args = parser.parse_args()
output = export_pdf(args.title, args.content, args.output, args.data_source)
print(f'output_path: {output}')
print('PDF 文档导出成功')