使用Python的PdfFileWriter()库生成带目录的PDF文件

要生成带目录的PDF文件，可以使用Python中的PdfFileWriter库。这个库可以用于创建、修改和合并PDF文件。下面是一个使用PdfFileWriter库生成带目录的PDF文件的示例代码。

首先，我们需要安装PyPDF2库，可以使用以下命令来安装：

pip install PyPDF2

接下来，我们可以开始编写代码。下面是一个生成带目录的PDF文件的示例代码：

from PyPDF2 import PdfWriter, PdfFileReader


def add_bookmark(writer, title, page_number):
    bookmark = writer.addBookmark(title, page_number)
    return bookmark


def generate_pdf_with_bookmarks(input_pdf_path, output_pdf_path, bookmarks):
    input_pdf = PdfFileReader(open(input_pdf_path, 'rb'))
    output_pdf = PdfWriter()

    # 添加目录页
    output_pdf.addPage(input_pdf.getPage(0))

    # 添加每个目录项和对应页面
    for title, page_number in bookmarks:
        bookmark = add_bookmark(output_pdf, title, page_number)
        output_pdf.addPage(input_pdf.getPage(page_number))

    # 将生成的PDF保存到文件中
    with open(output_pdf_path, 'wb') as output_file:
        output_pdf.write(output_file)


# 示例调用
input_pdf_path = 'input.pdf'
output_pdf_path = 'output.pdf'

bookmarks = [('Chapter 1', 1), ('Chapter 2', 3), ('Chapter 3', 5)]

generate_pdf_with_bookmarks(input_pdf_path, output_pdf_path, bookmarks)

上面的代码通过调用generate_pdf_with_bookmarks函数来生成带有目录的PDF文件。这个函数接受三个参数：输入PDF文件路径、输出PDF文件路径和目录项列表。目录项列表是一个包含目录项和对应页面编号的元组列表。

在函数中，我们首先使用PdfFileReader类打开输入的PDF文件，并创建一个新的PdfWriter实例来保存生成的PDF文件。然后，我们将输入PDF的 页添加到输出PDF中作为目录页。

接下来，我们遍历目录项列表，并分别将目录项和对应页面添加到输出PDF中。为了在PDF中创建目录项，我们调用了addBookmark方法，并将返回的书签对象保存到变量中。

最后，我们将生成的PDF保存到输出文件中。

使用上面的代码，你可以生成一个带有目录的PDF文件。在这个示例中，我们添加了三个目录项，分别是"Chapter 1"、"Chapter 2"和"Chapter 3"，并指定了每个目录项对应的页面编号。

希望本文对你有所帮助！