介绍

使用这个只有10行代码的Python应用程序，您可以为您的PDF文件设置自定义密码。只需将源代码放在与PDF文件相同的位置，并在程序中提供源PDF的名称、密码和输出PDF的名称。

这个应用程序的功能

1. 创建原始PDF的副本

2. 使用密码保护副本PDF

3. 删除原始文件

要求

- Python编程语言

编写代码

import PyPDF2

def set_pdf_password(input_file, output_file, password):
    # 打开输入PDF文件
    with open(input_file, 'rb') as file:
        pdf = PyPDF2.PdfReader(file)

        # 创建一个新的PDF写入器对象
        pdf_writer = PyPDF2.PdfWriter()

        # 遍历每一页将其加入到写入器中
        for page_num in range(pdf.numPages):
            pdf_writer.addPage(pdf.getPage(page_num))

        # 设置密码保护
        pdf_writer.encrypt(password)

        # 写入加密后的PDF文件
        with open(output_file, 'wb') as output:
            pdf_writer.write(output)

    print("密码已成功设置并保存为新文件。")

# 示例用法
input_file = 'input.pdf'  # 输入的PDF文件路径
output_file = 'protected.pdf'  # 加密后的PDF文件保存路径
password = '123456'  # 设置的密码

set_pdf_password(input_file, output_file, password)