使用Python的PdfFileWriter()库生成带密码保护的PDF文件

生成带密码保护的PDF文件可以使用Python中的PyPDF2库，该库提供了PdfFileWriter()类，可用于创建和修改PDF文件。以下是一个使用PyPDF2生成带密码保护的PDF文件的例子：

from PyPDF2 import PdfWriter, PdfFileReader

def add_password(input_pdf, output_pdf, password):
    # 创建PdfFileReader对象打开要加密的PDF文件
    with open(input_pdf, 'rb') as file:
        pdf = PdfFileReader(file)
        # 创建PdfFileWriter对象
        writer = PdfWriter()
        # 将PDF内容复制到writer对象
        for i in range(pdf.getNumPages()):
            writer.addPage(pdf.getPage(i))
        
        # 使用密码保护整个PDF文件
        writer.encrypt(password)
        
        # 将writer对象的内容写入到输出PDF文件
        with open(output_pdf, 'wb') as file:
            writer.write(file)

# 调用add_password函数生成带密码保护的PDF文件
input_pdf = 'input.pdf'
output_pdf = 'output.pdf'
password = 'mypassword'
add_password(input_pdf, output_pdf, password)

以上代码中，add_password函数接收三个参数：输入PDF文件的路径，输出PDF文件的路径和密码。函数首先使用PdfFileReader打开输入PDF文件并创建PdfFileWriter对象。然后将输入PDF文件的内容复制到writer对象中。接下来，通过调用writer对象的encrypt方法给整个PDF文件添加密码保护。最后，将writer对象的内容写入到输出PDF文件中。

要使用该函数，需要将要加密的PDF文件的路径（input_pdf），生成的带密码保护的PDF文件的路径（output_pdf）和设置的密码（password）作为参数传递给add_password函数。

上述例子生成的output.pdf文件将会是带有密码保护的PDF文件。在打开output.pdf文件时，需要输入设置的密码才能查看和编辑文件内容。