在实际办公和开发场景中,我们经常会遇到需要向大量PDF文件添加统一封面、插入特定说明页的需求,如果这些PDF分散在多层嵌套的目录中,手动逐个处理会耗费大量时间。借助Python的os模块遍历目录,配合PyMuPDF库的页面操作能力,可以高效完成这个批量处理任务。

环境准备
首先需要安装PyMuPDF库,它是fitz模块的上层封装,提供了丰富的PDF操作接口。在命令行中执行以下命令完成安装:
pip install pymupdf
核心实现思路
整个功能的实现分为三个核心步骤:
- 遍历多层目录,筛选出所有PDF文件路径
- 读取需要插入或追加的源页面PDF文件
- 对每个目标PDF执行页面插入或追加操作并保存
目录遍历实现
使用os模块的walk方法可以递归遍历多层目录,获取所有PDF文件的完整路径:
import os
def get_all_pdf_paths(root_dir):
"""递归获取多层目录下所有PDF文件路径"""
pdf_paths = []
for root, dirs, files in os.walk(root_dir):
for file in files:
if file.lower().endswith('.pdf'):
full_path = os.path.join(root, file)
pdf_paths.append(full_path)
return pdf_paths
PDF页面操作实现
PyMuPDF中通过fitz.open打开PDF文件,使用insert_pdf方法可以实现页面插入和追加操作,该方法支持指定插入位置、插入的页面范围等参数。
import fitz
def process_pdf(target_pdf_path, source_pdf_path, insert_pos=-1):
"""
处理单个PDF文件,插入或追加指定页面
:param target_pdf_path: 目标PDF路径
:param source_pdf_path: 源页面PDF路径
:param insert_pos: 插入位置,默认-1表示追加到末尾
"""
try:
# 打开目标PDF和源页面PDF
target_doc = fitz.open(target_pdf_path)
source_doc = fitz.open(source_pdf_path)
# 插入源PDF的所有页面到目标PDF的指定位置
target_doc.insert_pdf(source_doc, start_at=insert_pos)
# 保存处理后的PDF,覆盖原文件
target_doc.save(target_pdf_path)
target_doc.close()
source_doc.close()
print(f"处理成功: {target_pdf_path}")
except Exception as e:
print(f"处理失败 {target_pdf_path}: {str(e)}")
完整批量处理代码
将目录遍历和PDF处理逻辑结合,实现完整的批量处理功能:
import os
import fitz
def get_all_pdf_paths(root_dir):
"""递归获取多层目录下所有PDF文件路径"""
pdf_paths = []
for root, dirs, files in os.walk(root_dir):
for file in files:
if file.lower().endswith('.pdf'):
full_path = os.path.join(root, file)
pdf_paths.append(full_path)
return pdf_paths
def process_pdf(target_pdf_path, source_pdf_path, insert_pos=-1):
"""
处理单个PDF文件,插入或追加指定页面
:param target_pdf_path: 目标PDF路径
:param source_pdf_path: 源页面PDF路径
:param insert_pos: 插入位置,默认-1表示追加到末尾
"""
try:
target_doc = fitz.open(target_pdf_path)
source_doc = fitz.open(source_pdf_path)
target_doc.insert_pdf(source_doc, start_at=insert_pos)
target_doc.save(target_pdf_path)
target_doc.close()
source_doc.close()
print(f"处理成功: {target_pdf_path}")
except Exception as e:
print(f"处理失败 {target_pdf_path}: {str(e)}")
def batch_process_pdfs(root_dir, source_pdf_path, insert_pos=-1):
"""
批量处理多层目录下的PDF文件
:param root_dir: 根目录路径
:param source_pdf_path: 源页面PDF路径
:param insert_pos: 插入位置
"""
pdf_paths = get_all_pdf_paths(root_dir)
print(f"共找到 {len(pdf_paths)} 个PDF文件,开始处理...")
for pdf_path in pdf_paths:
process_pdf(pdf_path, source_pdf_path, insert_pos)
if __name__ == "__main__":
# 配置参数
ROOT_DIR = "./pdf_files" # 存放PDF文件的根目录
SOURCE_PDF = "./insert_page.pdf" # 要插入的页面PDF
INSERT_POS = 0 # 插入位置,0表示插入到第一页,默认-1表示追加到末尾
batch_process_pdfs(ROOT_DIR, SOURCE_PDF, INSERT_POS)
参数说明与注意事项
代码中的关键参数说明:
ROOT_DIR:需要遍历的根目录,程序会自动处理该目录下所有子文件夹中的PDFSOURCE_PDF:包含要插入页面的PDF文件,支持单页或多页,多页会全部插入INSERT_POS:页面插入位置,0表示插入到第一页前,1表示插入到第一页后,依次类推,-1表示追加到文件末尾
注意事项:
- 处理前建议备份原始PDF文件,避免操作失误导致文件损坏
- 如果目标PDF有密码保护,需要先解密才能处理
- insert_pdf方法默认插入源PDF的所有页面,如果需要插入指定页面,可以添加
from_page和to_page参数指定范围
常见问题解决
如果遇到PDF处理失败的情况,可以先检查以下几点:
- 确认PyMuPDF版本是否为最新,旧版本可能存在兼容性问题
- 检查目标PDF是否被其他程序占用,关闭占用程序后重试
- 如果插入的页面样式异常,可以检查源PDF的页面尺寸是否与目标PDF匹配,必要时可以先调整源页面尺寸