250 lines
11 KiB
Python
250 lines
11 KiB
Python
"""生成三类测试 PDF：文本版 / 影印版(扫描) / 混合版，用于验证解析效果。"""
|
||
import os, sys
|
||
import pymupdf as fitz
|
||
from PIL import Image, ImageDraw, ImageFont
|
||
|
||
HERE = os.path.dirname(os.path.abspath(__file__))
|
||
OUT = os.path.join(HERE, "..", "tests", "pdfs")
|
||
os.makedirs(OUT, exist_ok=True)
|
||
|
||
|
||
def font_path():
|
||
for p in [
|
||
"/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc",
|
||
"/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc",
|
||
]:
|
||
if os.path.exists(p):
|
||
return p
|
||
return None
|
||
|
||
|
||
def draw_chart(path, title="产品季度销量趋势"):
|
||
"""生成一张图表图片（折线图）。"""
|
||
img = Image.new("RGB", (700, 380), "white")
|
||
d = ImageDraw.Draw(img)
|
||
f = None
|
||
try:
|
||
f = ImageFont.truetype(font_path(), 20)
|
||
except Exception:
|
||
f = ImageFont.load_default()
|
||
d.text((250, 10), title, fill="black", font=f)
|
||
d.rectangle([60, 50, 660, 320], outline="black", width=2)
|
||
# 坐标轴刻度
|
||
xs = [80, 190, 300, 410, 520, 630]
|
||
labels = ["Q1", "Q2", "Q3", "Q4", "Q5", "Q6"]
|
||
vals = [80, 150, 120, 220, 190, 300]
|
||
for x, lb in zip(xs, labels):
|
||
d.text((x - 15, 325), lb, fill="black", font=f)
|
||
d.line([x, 315, x, 55], fill="lightgray", width=1)
|
||
pts = []
|
||
for x, v in zip(xs, vals):
|
||
y = 315 - v
|
||
d.ellipse([x - 4, y - 4, x + 4, y + 4], fill="blue")
|
||
pts.append((x, y))
|
||
d.line(pts, fill="blue", width=3)
|
||
for (x, y), v in zip(pts, vals):
|
||
d.text((x - 10, y - 28), str(v), fill="black", font=f)
|
||
img.save(path)
|
||
|
||
|
||
def draw_photo(path):
|
||
"""生成一张风景照片风格的图（渐变+山形）。"""
|
||
img = Image.new("RGB", (600, 350), (135, 206, 250))
|
||
d = ImageDraw.Draw(img)
|
||
d.rectangle([0, 200, 600, 350], fill=(34, 139, 34)) # 草地
|
||
d.polygon([(150, 200), (300, 60), (450, 200)], fill=(105, 105, 105)) # 山
|
||
d.polygon([(350, 200), (520, 40), (600, 200)], fill=(139, 139, 139))
|
||
d.ellipse([80, 50, 180, 150], fill=(255, 255, 0)) # 太阳
|
||
try:
|
||
f = ImageFont.truetype(font_path(), 22)
|
||
d.text((180, 300), "山间风景图", fill="white", font=f)
|
||
except Exception:
|
||
pass
|
||
img.save(path)
|
||
|
||
|
||
def make_text_pdf(path):
|
||
"""文本版 PDF：标题/段落/列表/表格 + 内嵌图表。"""
|
||
doc = fitz.open()
|
||
page = doc.new_page(width=595, height=842) # A4
|
||
font = "china-s"
|
||
y = 60
|
||
page.insert_text((72, y), "智能营销系统产品白皮书", fontsize=22, fontname=font)
|
||
y += 40
|
||
page.insert_text((72, y), "版本 1.0 | 2025 年 8 月", fontsize=12, fontname=font)
|
||
y += 40
|
||
page.insert_text((72, y), "一、系统概述", fontsize=16, fontname=font)
|
||
y += 28
|
||
paras = [
|
||
"本白皮书介绍智能营销系统（SmartMarketing Pro）的整体架构、核心能力与商业价值。系统基于大数据与机器学习技术，为企业在获客、转化、留存三个环节提供端到端的自动化营销能力。",
|
||
"系统当前已服务超过 1200 家企业客户，覆盖零售、金融、教育、医疗等多个行业。2025 年第二季度整体营收同比增长 32%，续费率达到 86%。",
|
||
]
|
||
for p in paras:
|
||
page.insert_textbox(fitz.Rect(72, y, 523, y + 60), p, fontsize=11, fontname=font, align=fitz.TEXT_ALIGN_LEFT)
|
||
y += 65
|
||
page.insert_text((72, y), "二、核心功能模块", fontsize=16, fontname=font)
|
||
y += 28
|
||
items = [
|
||
"1. 客户画像：基于行为数据构建 360 度用户画像，支持千人千面推荐；",
|
||
"2. 智能触达：覆盖短信、邮件、站内信等多渠道，自动选择最优触达时间；",
|
||
"3. 效果分析：实时 ROI 分析，支持 A/B 实验与归因模型；",
|
||
"4. 自动化流程：可视化编排营销旅程，支持条件分支与延迟节点。",
|
||
]
|
||
for it in items:
|
||
page.insert_text((90, y), it, fontsize=11, fontname=font)
|
||
y += 22
|
||
y += 10
|
||
page.insert_text((72, y), "三、产品版本对比", fontsize=16, fontname=font)
|
||
y += 30
|
||
# 表格（用文本对齐绘制）
|
||
cols = [(72, 200), (200, 90), (290, 90), (380, 90), (470, 60)]
|
||
headers = ["版本", "用户数", "自动化", "AI 洞察", "价格"]
|
||
rows = [
|
||
["基础版", "≤500", "基础流程", "无", "¥9,999/年"],
|
||
["专业版", "≤5000", "全部流程", "基础", "¥39,999/年"],
|
||
["企业版", "不限", "全部流程", "高级", "¥99,999/年"],
|
||
]
|
||
def draw_row(cy, cells, bold=False):
|
||
for (x, w), c in zip(cols, cells):
|
||
page.insert_text((x + 5, cy + 8), c, fontsize=10, fontname=font)
|
||
return cy
|
||
# 表头下划线
|
||
page.draw_line((72, y + 20), (530, y + 20), color=(0, 0, 0))
|
||
y += 24
|
||
y = draw_row(y, headers)
|
||
page.draw_line((72, y + 8), (530, y + 8), color=(0, 0, 0))
|
||
for r in rows:
|
||
y += 22
|
||
y = draw_row(y, r)
|
||
page.draw_line((72, y + 8), (530, y + 8), color=(0.8, 0.8, 0.8))
|
||
y += 30
|
||
# 插入内嵌图表图片
|
||
chart = os.path.join(OUT, "_chart.png")
|
||
draw_chart(chart)
|
||
page.insert_image(fitz.Rect(120, y - 14, 475, y + 150), filename=chart)
|
||
y += 178
|
||
page.insert_text((72, y), "图 1：产品季度销量趋势（2024Q1-2025Q2）", fontsize=10, fontname=font)
|
||
y += 28
|
||
page.insert_text((72, y), "四、总结", fontsize=16, fontname=font)
|
||
y += 28
|
||
page.insert_textbox(fitz.Rect(72, y, 523, y + 80), "智能营销系统通过数据驱动的方式显著提升了企业的营销效率。未来将持续引入大模型能力，提供更智能的营销策略生成与自然语言交互体验。", fontsize=11, fontname=font)
|
||
doc.save(path)
|
||
print("text pdf ->", path)
|
||
|
||
|
||
def make_scanned_pdf(path, pages=3):
|
||
"""影印版 PDF：把排版好的页面渲染成图片，模拟扫描件（无文本层）。"""
|
||
# 先用文本版生成一页，再渲染成图
|
||
tmp = os.path.join(OUT, "_tmp_text.pdf")
|
||
make_text_pdf(tmp)
|
||
src = fitz.open(tmp)
|
||
page0 = src[0]
|
||
pix = page0.get_pixmap(matrix=fitz.Matrix(2.0, 2.0))
|
||
img_path = os.path.join(OUT, "_scan_src.png")
|
||
pix.save(img_path)
|
||
|
||
doc = fitz.open()
|
||
for i in range(pages):
|
||
page = doc.new_page(width=595, height=842)
|
||
if i == 0:
|
||
img = img_path
|
||
else:
|
||
# 后续页用纯文字图片
|
||
img2 = os.path.join(OUT, f"_scan_text_{i}.png")
|
||
im = Image.new("RGB", (1190, 1684), "white")
|
||
d = ImageDraw.Draw(im)
|
||
try:
|
||
f = ImageFont.truetype(font_path(), 34)
|
||
except Exception:
|
||
f = ImageFont.load_default()
|
||
lines = [
|
||
f"影印版第 {i} 页",
|
||
"",
|
||
"本页内容说明：影印版 PDF 没有可选择的文本层，",
|
||
"所有内容都以图片形式存在。需要借助 OCR 或多模态大模型",
|
||
"才能把内容准确还原为可编辑的文字。",
|
||
"",
|
||
"表格示例：",
|
||
"科目 数量 单价 金额",
|
||
"笔记本 100 30 3000",
|
||
"显示器 20 1200 24000",
|
||
"键盘 50 100 5000",
|
||
]
|
||
yy = 80
|
||
for ln in lines:
|
||
d.text((60, yy), ln, fill="black", font=f)
|
||
yy += 60
|
||
im.save(img2)
|
||
img = img2
|
||
page.insert_image(fitz.Rect(0, 0, 595, 842), filename=img)
|
||
doc.save(path)
|
||
print("scanned pdf ->", path)
|
||
|
||
|
||
def make_mixed_pdf(path):
|
||
"""混合版 PDF：文本页 + 影印图片页 + 含内嵌图的文本页。"""
|
||
doc = fitz.open()
|
||
font = "china-s"
|
||
# 第 1 页：纯文本
|
||
p = doc.new_page(width=595, height=842)
|
||
p.insert_text((72, 60), "混合文档示例：目录页", fontsize=20, fontname=font)
|
||
p.insert_text((72, 100), "本页为可复制文本。下一页为扫描图片页，再下一页为带图表页面。", fontsize=12, fontname=font)
|
||
p.insert_text((72, 130), "1. 项目背景", fontsize=12, fontname=font)
|
||
p.insert_text((72, 152), "2. 数据概览", fontsize=12, fontname=font)
|
||
p.insert_text((72, 174), "3. 实施方案", fontsize=12, fontname=font)
|
||
# 第 2 页：扫描图片页（照片 + 文字合成一张扫描图）
|
||
p2 = doc.new_page(width=595, height=842)
|
||
photo = os.path.join(OUT, "_photo.png")
|
||
draw_photo(photo)
|
||
# 合成：上部照片 + 下部说明文字 → 模拟真实扫描页
|
||
page_img = Image.new("RGB", (1190, 1684), "white")
|
||
photo_im = Image.open(photo)
|
||
photo_im = photo_im.resize((1190, 690))
|
||
page_img.paste(photo_im, (0, 0))
|
||
d = ImageDraw.Draw(page_img)
|
||
try:
|
||
f = ImageFont.truetype(font_path(), 34)
|
||
except Exception:
|
||
f = ImageFont.load_default()
|
||
d.text((80, 750), "扫描页说明：这张页面是扫描件，文字和图片融为一体，", fill="black", font=f)
|
||
d.text((80, 820), "只能通过图像识别还原内容。下方还有一段正文。", fill="black", font=f)
|
||
lines = [
|
||
"正文段落：本扫描页展示了项目现场的照片。",
|
||
"项目位于山区，风景优美，施工进度正常。",
|
||
"更多细节请参见后续章节。",
|
||
]
|
||
yy = 950
|
||
for ln in lines:
|
||
d.text((80, yy), ln, fill="black", font=f)
|
||
yy += 60
|
||
page_img_path = os.path.join(OUT, "_mix_scan.png")
|
||
page_img.save(page_img_path)
|
||
p2.insert_image(fitz.Rect(0, 0, 595, 842), filename=page_img_path)
|
||
# 第 3 页：文本 + 内嵌表格图 + 段落
|
||
p3 = doc.new_page(width=595, height=842)
|
||
p3.insert_text((72, 60), "混合文档示例：数据概览页", fontsize=18, fontname=font)
|
||
p3.insert_text((72, 100), "下表展示了各区域销售业绩，表格以图片形式嵌入：", fontsize=11, fontname=font)
|
||
table_img = os.path.join(OUT, "_table.png")
|
||
timg = Image.new("RGB", (900, 300), "white")
|
||
td = ImageDraw.Draw(timg)
|
||
try:
|
||
tf = ImageFont.truetype(font_path(), 26)
|
||
except Exception:
|
||
tf = ImageFont.load_default()
|
||
td.text((20, 20), "区域 | 销售额(万) | 目标达成率", fill="black", font=tf)
|
||
td.text((20, 80), "华东 | 5200 | 108%", fill="black", font=tf)
|
||
td.text((20, 140), "华南 | 3800 | 96%", fill="black", font=tf)
|
||
td.text((20, 200), "华北 | 4100 | 102%", fill="black", font=tf)
|
||
timg.save(table_img)
|
||
p3.insert_image(fitz.Rect(90, 130, 505, 230), filename=table_img)
|
||
p3.insert_text((72, 280), "结论：华东区表现最佳，华南区需加强跟进。", fontsize=11, fontname=font)
|
||
doc.save(path)
|
||
print("mixed pdf ->", path)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
make_text_pdf(os.path.join(OUT, "test_text.pdf"))
|
||
make_scanned_pdf(os.path.join(OUT, "test_scanned.pdf"), pages=2)
|
||
make_mixed_pdf(os.path.join(OUT, "test_mixed.pdf"))
|
||
print("ALL DONE ->", OUT)
|