本文档旨在提供一个使用PyMuPDF库，根据指定的页码范围分割PDF文件，并保留或重建分割后PDF文件的目录（Table of Contents, TOC）的详细教程。我们将深入探讨PyMuPDF库提供的get_toc()和set_toc()方法，并提供相应的代码示例，帮助读者理解如何正确处理和更新PDF目录结构，以确保分割后的PDF文件仍然具有完整的导航功能。
PDF分割与目录保留：PyMuPDF实战指南
PyMuPDF（也称为fitz）是一个强大的Python库，用于处理PDF文档。它提供了丰富的功能，包括读取、创建、修改和提取PDF内容。本文将重点介绍如何使用PyMuPDF分割PDF文档，并确保分割后的文档保留原有的目录结构。
获取PDF目录
首先，我们需要从原始PDF文档中提取目录信息。PyMuPDF提供了Document.get_toc()方法，可以轻松获取PDF的目录。
import fitzdef get_pdf_toc(input_pdf_path): """ 从PDF文件中提取目录信息。 Args: input_pdf_path (str): 输入PDF文件的路径。 Returns: list: 目录列表，每个条目是一个包含层级、标题和页码的子列表。 """ pdf_document = fitz.open(input_pdf_path) toc = pdf_document.get_toc() pdf_document.close() return toc# 示例用法input_pdf_path = "input.pdf"toc = get_pdf_toc(input_pdf_path)print(toc)
get_toc()方法返回一个列表，其中每个元素都是一个包含目录条目信息的子列表。每个子列表的格式为[level, title, page]，其中：
level：目录条目的层级，表示目录的嵌套关系。title：目录条目的标题。page：目录条目对应的页码（注意：页码从1开始计数）。
分割PDF并更新目录
在分割PDF文档时，我们需要根据分割的页码范围，调整目录条目的页码，并将调整后的目录信息添加到新的PDF文档中。
import fitzdef split_pdf_with_toc(input_pdf_path, output_folder, page_ranges): """ 分割PDF文件，并为每个分割后的文件添加对应的目录。 Args: input_pdf_path (str): 输入PDF文件的路径。 output_folder (str): 输出文件夹的路径。 page_ranges (list): 一个包含页码范围元组的列表，例如 [(1, 10), (11, 20)]。 """ pdf_document = fitz.open(input_pdf_path) original_toc = pdf_document.get_toc() for i, (start_page, end_page) in enumerate(page_ranges, start=1): # 创建一个新的PDF文档 chapter_pdf = fitz.open() chapter_pdf.insert_pdf(pdf_document, from_page=start_page - 1, to_page=end_page - 1) # PyMuPDF 页码从0开始 # 调整目录条目的页码 adjusted_toc = [] for level, title, page in original_toc: if start_page <= page <= end_page: adjusted_page = page - start_page + 1 # 调整后的页码 adjusted_toc.append([level, title, adjusted_page]) # 确保目录结构符合PyMuPDF的要求 if adjusted_toc: if adjusted_toc[0][0] != 1: # 添加虚拟目录条目 first_level = adjusted_toc[0][0] for j in range(1, first_level): adjusted_toc.insert(0,[j, "dummy", 1]) # 设置新的目录 chapter_pdf.set_toc(adjusted_toc) # 保存新的PDF文档 output_pdf_path = f"{output_folder}/Chapter_{i}.pdf" chapter_pdf.save(output_pdf_path) chapter_pdf.close() pdf_document.close()# 示例用法input_pdf_path = "input.pdf"output_folder = "output"page_ranges = [(1, 10), (11, 20), (21,30)]split_pdf_with_toc(input_pdf_path, output_folder, page_ranges)
代码解释：
获取原始目录： 使用pdf_document.get_toc()获取原始PDF文档的目录。循环处理每个页码范围： 遍历page_ranges列表，为每个页码范围创建一个新的PDF文档。插入页面： 使用chapter_pdf.insert_pdf()将指定范围的页面从原始PDF复制到新的PDF文档中。注意： PyMuPDF的页码从0开始计数，所以from_page和to_page需要减1。调整目录页码： 遍历原始目录，只保留在新页码范围内的条目，并调整它们的页码。处理目录结构： 检查调整后的目录的第一个条目的层级是否为1。如果不是，则插入虚拟目录条目，以满足PyMuPDF对目录结构的强制要求。设置新目录： 使用chapter_pdf.set_toc()将调整后的目录设置到新的PDF文档中。保存新文档： 将新的PDF文档保存到指定的输出文件夹中。
目录结构注意事项
PyMuPDF对目录结构有严格的要求：
第一个目录条目的层级必须为1。后续条目的层级只能与前一个条目相同，或者比前一个条目高1级，或者小于前一个条目。
如果目录结构不符合这些要求，set_toc()方法可能会失败，或者导致目录显示不正确。因此，在设置目录之前，务必检查并调整目录结构。
总结
本文介绍了如何使用PyMuPDF库分割PDF文档，并保留或重建分割后PDF文件的目录。通过get_toc()和set_toc()方法，我们可以轻松地提取、修改和设置PDF目录。在实际应用中，需要根据具体情况调整代码，例如处理更复杂的目录结构，或者优化分割算法。希望本文能够帮助读者更好地理解和使用PyMuPDF库。
以上就是基于PyMuPDF实现PDF按页码范围分割并保留目录的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1369385.html
微信扫一扫 
支付宝扫一扫 
