使用Python的PdfFileWriter()对PDF文件进行加密

在Python中，我们可以使用PyPDF2库中的PdfFileWriter()来对PDF文件进行加密。PdfFileWriter()类允许我们创建新的PDF文件，并对其进行各种操作，包括加密。

首先，我们需要安装PyPDF2库，可以使用以下命令在终端中安装：

pip install PyPDF2

安装完成后，我们可以开始使用PdfFileWriter()来加密PDF文件。以下是一个使用PdfFileWriter()加密PDF文件的示例代码：

from PyPDF2 import PdfFileWriter, PdfFileReader

def encrypt_pdf(input_path, output_path, password):
    # 创建一个PdfFileWriter对象
    pdf_writer = PdfFileWriter()

    # 读取原始PDF文件
    pdf_reader = PdfFileReader(input_path)

    # 逐页复制原始PDF文件的内容到PdfFileWriter对象中
    for page_num in range(pdf_reader.numPages):
        page = pdf_reader.getPage(page_num)
        pdf_writer.addPage(page)

    # 设置加密密码
    pdf_writer.encrypt(password)

    # 将加密的PDF内容写入输出文件
    with open(output_path, 'wb') as output_file:
        pdf_writer.write(output_file)

if __name__ == '__main__':
    # 输入PDF文件路径
    input_path = 'example.pdf'

    # 输出PDF文件路径
    output_path = 'encrypted_example.pdf'

    # 设置加密密码
    password = 'mypassword'

    # 加密PDF文件
    encrypt_pdf(input_path, output_path, password)

在上面的示例代码中，我们定义了一个encrypt_pdf()函数，它接受输入PDF文件路径、输出PDF文件路径和加密密码作为参数。首先，我们创建一个PdfFileWriter对象，并使用PdfFileReader()读取原始PDF文件的内容。然后，我们逐页复制原始PDF文件的内容到PdfFileWriter对象中。接下来，我们使用encrypt()方法设置加密密码。最后，我们使用write()方法将加密的PDF内容写入输出文件。

在运行代码时，需要将示例中的example.pdf替换为要加密的真实PDF文件路径。生成的加密PDF文件将保存为encrypted_example.pdf，使用的加密密码可以根据需要进行修改。

需要注意的是，加密PDF文件后，如果忘记了密码，将无法打开或编辑该文件。因此，在使用加密功能时，请务必牢记密码并进行备份以防丢失。

希望以上例子对你有所帮助！