malue-ai PDF 工具箱
帮助用户完成各种 PDF 操作：合并、拆分、加密、加水印、提取文本、转换格式。
使用场景
- 用户说「把这 5 个 PDF 合并成一个」「拆分这个 PDF 的第 3-10 页」
- 用户说「给 PDF 加个密码」「加个水印」
- 用户说「提取 PDF 里的文字」「PDF 转 Word」
依赖安装
首次使用时自动安装：
pip install pypdf reportlab
执行方式
通过 Python 脚本使用 pypdf 和 reportlab 处理 PDF。
合并 PDF
from pypdf import PdfWriter
writer = PdfWriter()
for pdf_path in ["/path/to/1.pdf", "/path/to/2.pdf", "/path/to/3.pdf"]:
writer.append(pdf_path)
writer.write("/path/to/merged.pdf")
writer.close()
print("合并完成")
拆分 PDF
from pypdf import PdfReader, PdfWriter
reader = PdfReader("/path/to/input.pdf")
writer = PdfWriter()
# 提取第 3-10 页（0-indexed）
for page_num in range(2, 10):
writer.add_page(reader.pages[page_num])
writer.write("/path/to/extracted.pdf")
writer.close()
加密 PDF
from pypdf import PdfReader, PdfWriter
reader = PdfReader("/path/to/input.pdf")
writer = PdfWriter()
for page in reader.pages:
writer.add_page(page)
writer.encrypt("user_password")
writer.write("/path/to/encrypted.pdf")
writer.close()
添加文字水印
from pypdf import PdfReader, PdfWriter
from reportlab.pdfgen import canvas
from reportlab.lib.pagesizes import A4
import io
# 创建水印 PDF
packet = io.BytesIO()
c = canvas.Canvas(packet, pagesize=A4)
c.setFont("Helvetica", 40)
c.setFillAlpha(0.3)
c.saveState()
c.translate(300, 400)
c.rotate(45)
c.drawCentredString(0, 0, "CONFIDENTIAL")
c.restoreState()
c.save()
packet.seek(0)
# 叠加水印
watermark = PdfReader(packet)
reader = PdfReader("/path/to/input.pdf")
writer = PdfWriter()
for page in reader.pages:
page.merge_page(watermark.pages[0])
writer.add_page(page)
writer.write("/path/to/watermarked.pdf")
提取文本
from pypdf import PdfReader
reader = PdfReader("/path/to/input.pdf")
text = ""
for page in reader.pages:
text += page.extract_text() + "\n"
print(f"共 {len(reader.pages)} 页，提取文本 {len(text)} 字符")
安全规则
- 加密操作确认密码：提醒用户牢记密码，PDF 加密后无法恢复
- 合并前确认顺序：列出文件顺序让用户确认
- 不覆盖原文件：输出到新文件，保留原始 PDF
输出规范
- 操作完成后报告：文件路径、页数、文件大小
- 合并/拆分前展示文件列表和页码范围
- 加密后提醒用户记住密码