利用Python与PyMuPDF库批量向多层目录下的PDF文件追加或插入指定页面

pip install PyMuPDF

import fitz  # PyMuPDF库
import os
import pathlib

# 定义目标根目录和要追加的页面路径
target_folder = r"C:\Folders"
page_to_add_path = r"C:\page2.pdf"

# 收集所有PDF文件路径
pdffiles = []
for root, dirs, files in os.walk(target_folder):
    for name in files:
        if name.endswith(".pdf"):
            pdffiles.append(os.path.join(root, name))

# 打开要追加的源PDF页面
# 注意：这里只打开一次源PDF，避免重复IO操作
try:
    src_page_doc = fitz.open(page_to_add_path)
except fitz.FileNotFoundError:
    print(f"错误：未找到源PDF文件 '{page_to_add_path}'。请检查路径。")
    exit()
except Exception as e:
    print(f"打开源PDF文件时发生错误：{e}")
    exit()

# 遍历并处理每个PDF文件
print(f"开始处理 {len(pdffiles)} 个PDF文件...")
for f_path in pdffiles:
    try:
        # 将目标PDF文件内容读取到内存中
        # 这种方式对于在服务器上处理大量小文件特别有效，避免频繁的文件句柄操作
        stream = pathlib.Path(f_path).read_bytes()
        # 从内存中打开PDF文档
        doc = fitz.open("pdf", stream)

        # 将源PDF文档（这里只有一页）追加到当前文档的末尾
        doc.insert_pdf(src_page_doc)

        # 保存修改后的PDF，覆盖原文件
        # ez_save() 方法是一个便捷的保存方式，它会尝试优化文件大小
        doc.ez_save(f_path)
        print(f"成功处理: {f_path}")
    except fitz.EmptyFileError:
        print(f"警告: 文件 {f_path} 是空的或损坏的PDF，已跳过。")
    except Exception as e:
        print(f"处理文件 {f_path} 时发生错误: {e}")
    finally:
        # 确保文档关闭，释放资源
        if 'doc' in locals() and not doc.is_closed:
            doc.close()

# 关闭源PDF文档
if not src_page_doc.is_closed:
    src_page_doc.close()

print("所有PDF文件处理完成。")

import fitz
import os
import pathlib

# 定义目标根目录和要插入的页面路径
target_folder = r"C:\Folders"
page_before_path = r"C:\Files\page1.pdf"
page_after_path = r"C:\Files\page3.pdf"

# 收集所有PDF文件路径
pdffiles = []
for root, dirs, files in os.walk(target_folder):
    for name in files:
        if name.endswith(".pdf"):
            pdffiles.append(os.path.join(root, name))

# 预先打开要插入的源PDF页面
try:
    src1 = fitz.open(page_before_path)
    src3 = fitz.open(page_after_path)
except fitz.FileNotFoundError as e:
    print(f"错误：未找到源PDF文件。请检查路径：{e}")
    exit()
except Exception as e:
    print(f"打开源PDF文件时发生错误：{e}")
    exit()

print(f"开始处理 {len(pdffiles)} 个PDF文件，插入前后页面...")
for f_path in pdffiles:
    try:
        stream = pathlib.Path(f_path).read_bytes()
        doc = fitz.open("pdf", stream)

        # 在文档的第0页（即开头）插入 src1 的所有页面
        doc.insert_pdf(src1, start_at=0)
        # 在文档的末尾追加 src3 的所有页面
        doc.insert_pdf(src3)

        doc.ez_save(f_path)
        print(f"成功处理: {f_path}")
    except fitz.EmptyFileError:
        print(f"警告: 文件 {f_path} 是空的或损坏的PDF，已跳过。")
    except Exception as e:
        print(f"处理文件 {f_path} 时发生错误: {e}")
    finally:
        if 'doc' in locals() and not doc.is_closed:
            doc.close()

# 关闭源PDF文档
if not src1.is_closed:
    src1.close()
if not src3.is_closed:
    src3.close()

print("所有PDF文件处理完成。")