使用Python开发的PDF拆分工具

pip install pypdf

# -*- coding: utf-8 -*-
import re
from pathlib import path
from pypdf import pdfreader, pdfwriter

import tkinter as tk
from tkinter import filedialog, messagebox

# ================= 配置区域 =================

# 章节标题匹配规则（适合：第1章 / 第 1 章 / 第一章 / 第十四章 等）
chapter_pattern = re.compile(
    r"第\s*[一二三四五六七八九十百千0-9]+\s*章[^\n\r]*"
)

# 是否额外按“页”拆成单页 pdf
export_single_pages = true

# ==========================================

# gui 全局对象占位
root = none
log_text = none

def sanitize_filename(name: str) -> str:
    """
    去掉 windows 不支持的文件名字符。
    """
    return re.sub(r'[\\/:*?"<>|]', "_", name)


# ---------- 方案一：优先从 pdf 书签(outline) 中找章节 ----------

def find_chapters_from_outline(reader: pdfreader):
    """
    从 pdf 书签(outline) 中找出章节：
      - 遍历所有书签
      - 标题里匹配 chapter_pattern（第x章……）
      - 获取对应页码
    返回: [{title: '第1章 xxx', start: 0}, ...]
    """
    chapters = []

    # 兼容不同版本的 pypdf：有的叫 outline，有的叫 outlines
    try:
        outlines = reader.outline
    except exception:
        try:
            outlines = reader.outlines
        except exception:
            outlines = none

    if not outlines:
        return []

    def walk(items):
        for item in items:
            # 子列表：继续递归
            if isinstance(item, list):
                walk(item)
            else:
                # 尝试拿书签标题
                try:
                    title = item.title
                except attributeerror:
                    title = str(item)
                if not isinstance(title, str):
                    title = str(title)

                # 标题里不含“第x章”就跳过
                if not chapter_pattern.search(title):
                    continue

                # 拿到书签指向的页码
                try:
                    page_num = reader.get_destination_page_number(item)
                except exception:
                    continue

                chapters.append({"title": title.strip(), "start": page_num})

    walk(outlines)

    # 去重、排序（同一页只保留一个章节）
    unique = {}
    for ch in chapters:
        if ch["start"] not in unique:
            unique[ch["start"]] = ch

    chapters = sorted(unique.values(), key=lambda c: c["start"])
    return chapters


# ---------- 方案二：从正文文本中猜章节（备用） ----------

def find_chapters_from_text(reader: pdfreader):
    """
    扫描整个 pdf 正文，猜每一章的“起始页”以及章节标题。
    规则大致是：
      - 排除“目录/contents”页面
      - 一页内允许有多个“第x章”，逐个判断
      - 只要某个匹配出现在页面较前面，且所在行不像目录行(标题 + ...... + 页码) 就认为是章节开始
    返回: [{title: '第1章 xxx', start: 0}, ...]
    """
    chapters = []
    num_pages = len(reader.pages)

    for i in range(num_pages):
        page = reader.pages[i]
        text = page.extract_text() or ""
        if not text.strip():
            continue

        # 跳过目录页（粗略判断即可）
        head = text[:100]
        if "目录" in head or "contents" in head or "contents" in head:
            continue

        # 遍历此页所有匹配 "第x章"
        for m in chapter_pattern.finditer(text):
            # 要求标题出现在页面比较靠前的位置
            if m.start() > 400:   # 这个阈值可以按需要微调
                continue

            # 找到这一行的文本内容
            lines = text.splitlines()
            line_of_match = ""
            char_pos = 0
            for line in lines:
                next_pos = char_pos + len(line) + 1  # 粗略算上换行
                if m.start() < next_pos:
                    line_of_match = line
                    break
                char_pos = next_pos

            # 目录行一般是：标题 + 一串点 + 页码
            # 例如：第1章 人际关系的构成..................1
            if re.search(r"[\.·…]{3,}\s*\d+\s*$", line_of_match):
                # 像目录的行，忽略
                continue

            title = m.group(0).strip()

            # 同一页只认一个章节起点
            if not any(ch["start"] == i for ch in chapters):
                chapters.append({"title": title, "start": i})
                break  # 当前页已经找到一个章节了，后面不再找

    return chapters


# ---------- 包一层：带 logger 的 find_chapters ----------

def find_chapters(reader: pdfreader, logger=print):
    chapters = find_chapters_from_outline(reader)
    if chapters:
        logger("✅ 使用 pdf 书签识别章节")
        return chapters

    logger("⚠️ 此 pdf 没有可用书签，改用正文文本识别章节")
    chapters = find_chapters_from_text(reader)
    return chapters


def fill_chapter_ranges(chapters, num_pages):
    """
    根据 start 页自动计算每章的 end 页。
    修改 chapters 列表，增加 end 字段。
    """
    for idx, ch in enumerate(chapters):
        start = ch["start"]
        if idx < len(chapters) - 1:
            end = chapters[idx + 1]["start"] - 1
        else:
            end = num_pages - 1
        ch["end"] = end
    return chapters


def split_pdf_by_chapters(pdf_path, output_root, logger=none):
    """
    真正拆分 pdf 的函数，所有信息通过 logger 输出到日志区域
    """
    if logger is none:
        logger = print

    pdf_path = path(pdf_path)
    output_root = path(output_root)

    if not pdf_path.exists():
        msg = f"pdf 文件不存在: {pdf_path}"
        logger(msg)
        raise filenotfounderror(msg)

    reader = pdfreader(str(pdf_path))
    num_pages = len(reader.pages)

    book_name = pdf_path.name
    logger(f"开始处理：{book_name}")
    logger(f"总页数：{num_pages}")

    # 1. 找章节
    chapters = find_chapters(reader, logger=logger)
    if not chapters:
        msg = "未识别到任何章节标题，请检查：pdf 是否有书签/正文是否能提取文字/正则是否合适。"
        logger(msg)
        raise valueerror(msg)

    logger(f"共识别到 {len(chapters)} 章：")
    for idx, ch in enumerate(chapters, start=1):
        logger(f"  第{idx}章 → {ch['title']}（起始页：{ch['start'] + 1}）")

    # 2. 填充每章的结束页
    chapters = fill_chapter_ranges(chapters, num_pages)

    # 3. 创建输出根目录
    output_root.mkdir(parents=true, exist_ok=true)
    logger(f"输出目录：{output_root}")

    # 4. 按章节导出
    for idx, ch in enumerate(chapters, start=1):
        title = ch["title"]
        start_page = ch["start"]         # 0-based
        end_page = ch["end"]             # 0-based
        page_count = end_page - start_page + 1

        safe_title = sanitize_filename(title)
        chapter_dir = output_root / f"{idx:02d}_{safe_title}"
        chapter_dir.mkdir(parents=true, exist_ok=true)

        logger("")
        logger(f"==== 处理章节 {idx}: {title} ====")
        logger(f"页码范围: {start_page + 1} - {end_page + 1}（共 {page_count} 页）")
        logger(f"章节输出目录: {chapter_dir}")

        # 4.1 导出“整章一个 pdf”
        chapter_writer = pdfwriter()
        for p in range(start_page, end_page + 1):
            chapter_writer.add_page(reader.pages[p])

        chapter_pdf_path = chapter_dir / f"{idx:02d}_{safe_title}.pdf"
        with open(chapter_pdf_path, "wb") as f:
            chapter_writer.write(f)

        logger(f"  ✅ 已生成整章 pdf: {chapter_pdf_path.name}")

        # 4.2 可选：每一页单独导出
        if export_single_pages:
            for p in range(start_page, end_page + 1):
                writer = pdfwriter()
                writer.add_page(reader.pages[p])

                # 页码用 1 开始，且补零对齐，例如 p0001.pdf
                page_label = f"p{p + 1:04d}.pdf"
                single_page_path = chapter_dir / page_label
                with open(single_page_path, "wb") as f:
                    writer.write(f)
            logger("  ✅ 已生成单页 pdf 文件（按页命名）")

    logger("")
    logger("🎉 拆分完成！")


# ========= gui 部分（两个按钮 + 日志框） =========

selected_pdf_file = ""
selected_output_dir = ""

def append_log(msg: str):
    """写日志到 text，并自动滚动"""
    if log_text is none:
        print(msg)
        return
    log_text.config(state="normal")
    log_text.insert(tk.end, msg + "\n")
    log_text.see(tk.end)
    log_text.config(state="disabled")
    # 刷新一下界面，让日志滚动更及时
    if root is not none:
        root.update_idletasks()


def choose_pdf():
    """按钮1：选择 pdf 文件"""
    global selected_pdf_file
    path = filedialog.askopenfilename(
        title="请选择 pdf 文件",
        filetypes=[("pdf 文件", "*.pdf"), ("所有文件", "*.*")]
    )
    if path:
        selected_pdf_file = path
        label_pdf.config(text=f"已选择 pdf：{path}")
        append_log(f"已选择 pdf 文件：{path}")


def choose_output_and_run():
    """按钮2：选择输出目录并开始拆分"""
    global selected_output_dir, selected_pdf_file

    if not selected_pdf_file:
        messagebox.showwarning("提示", "请先选择 pdf 文件！")
        return

    path = filedialog.askdirectory(title="请选择输出目录")
    if not path:
        return

    selected_output_dir = path
    label_output.config(text=f"输出目录：{path}")
    append_log("")
    append_log(f"输出目录设置为：{path}")
    append_log("开始拆分，请稍候...\n")

    # 清理一下旧的错误提示
    try:
        split_pdf_by_chapters(selected_pdf_file, selected_output_dir, logger=append_log)
        messagebox.showinfo("完成", "拆分完成！\n请到输出目录查看各章节文件夹。")
    except exception as e:
        append_log(f"❌ 出错：{e}")
        messagebox.showerror("错误", f"处理过程中出现错误：\n{e}")


if __name__ == "__main__":
    # 创建窗口
    root = tk.tk()
    root.title("pdf 章节拆分工具")
    root.geometry("400x400")   # 固定为 400x400 的窗口
    root.resizable(false, false)

    # 上半部分：按钮区域
    btn_frame = tk.frame(root)
    btn_frame.pack(padx=10, pady=10, fill="x")

    # 按钮1：选择 pdf
    btn_pdf = tk.button(btn_frame, text="1. 请选择你的 pdf 文件", command=choose_pdf)
    btn_pdf.pack(fill="x")

    label_pdf = tk.label(btn_frame, text="尚未选择 pdf 文件", anchor="w")
    label_pdf.pack(fill="x", pady=(5, 10))

    # 按钮2：选择输出目录 + 开始拆分
    btn_output = tk.button(btn_frame, text="2. 请选择输出目录并开始拆分", command=choose_output_and_run)
    btn_output.pack(fill="x")

    label_output = tk.label(btn_frame, text="尚未选择输出目录", anchor="w")
    label_output.pack(fill="x", pady=(5, 0))

    # 下半部分：日志输出区域（带滚动条）
    log_frame = tk.frame(root)
    log_frame.pack(padx=10, pady=10, fill="both", expand=true)

    log_text = tk.text(log_frame, state="disabled")
    log_text.pack(side="left", fill="both", expand=true)

    scrollbar = tk.scrollbar(log_frame, command=log_text.yview)
    scrollbar.pack(side="right", fill="y")

    log_text.config(yscrollcommand=scrollbar.set)

    append_log("日志初始化完成。")
    append_log("提示：先选择 pdf 文件，再选择输出目录开始拆分。")

    root.mainloop()