如何使用Python中的PdfFileWriter()添加书签到PDF文件

使用Python中的PdfFileWriter()添加书签到PDF文件需要使用PyPDF2库。下面是一个简单的使用例子，本文将提到相关的代码：

from PyPDF2 import PdfFileWriter, PdfFileReader
from PyPDF2.generic import Destination
from PyPDF2.utils import PdfReadError

def add_bookmark(pdf_path, output_path, page_number, bookmark_title, parent=None):
    bookmark = Destination()
    bookmark['/Title'] = bookmark_title
    bookmark['/Page'] = f'/{page_number} 0 R'
    bookmark['/Type'] = '/XYZ'
    bookmark['/Left'] = '0'
    bookmark['/Top'] = '100%'
    
    try:
        pdf = PdfFileReader(open(pdf_path, 'rb'))
    except PdfReadError:
        print(f"Failed to open {pdf_path}")
        return
    
    output_pdf = PdfFileWriter()
    
    for i in range(pdf.getNumPages()):
        page = pdf.getPage(i)
        output_pdf.addPage(page)
        
    if parent is not None:
        parent.addBookmark(bookmark, output_pdf, parent.parent)
    else:
        output_pdf.addBookmark(bookmark)
    
    with open(output_path, 'wb') as output_file:
        output_pdf.write(output_file)
    print(f"Bookmarks added to {output_path}")

在上面的代码中，我们定义了一个 add_bookmark() 函数，它接受以下参数：

- pdf_path：原始PDF文件的路径。

- output_path：要保存修改后的PDF文件的路径。

- page_number：要添加书签的页面编号（从0开始）。

- bookmark_title：新书签的标题。

- parent：新书签的父书签（如果有的话）。默认为 None。

在函数内部，我们首先尝试打开原始的PDF文件。如果打开失败，我们将输出错误消息并返回。接着，我们创建一个PdfFileWriter对象 output_pdf，该对象用于保存添加了书签的PDF文件。

然后，我们遍历原始PDF中的每一页，并将其添加到 output_pdf 中。如果存在父书签，则通过 parent.addBookmark() 方法将新书签添加到父书签下，否则直接使用 output_pdf.addBookmark() 方法添加新书签。

最后，我们使用 output_path 参数将修改后的 output_pdf 对象保存到硬盘上，并输出成功消息。

现在，我们可以使用 add_bookmark() 函数来添加书签到PDF文件。以下是一个示例：

pdf_path = 'example.pdf'
output_path = 'output.pdf'
page_number = 3
bookmark_title = 'Chapter 1'
add_bookmark(pdf_path, output_path, page_number, bookmark_title)

在上面的示例中，我们将在第4页（索引为3）添加一个名为 "Chapter 1" 的书签，然后将修改后的PDF文件保存在 "output.pdf" 中。

请注意，这里使用的示例代码假设你已经安装了PyPDF2库。如果没有安装，可以通过以下命令安装：

pip install PyPDF2

希望这个示例对你有所帮助！