Python: password使用随机密码保护PDF，并保存文件名-密码

pip install PyPDF2

import random
import string
from PyPDF2 import PdfFileWriter, PdfFileReader

def generate_random_password(length):
    characters = string.ascii_letters + string.digits + string.punctuation
    password = ''.join(random.choice(characters) for _ in range(length))
    return password

def protect_pdf_with_password(input_file, output_file):
    password = generate_random_password(10)  # 生成一个长度为10的随机密码
    pdf_writer = PdfFileWriter()
    pdf_reader = PdfFileReader(input_file)

    for page_num in range(pdf_reader.numPages):
        pdf_writer.addPage(pdf_reader.getPage(page_num))

    pdf_writer.encrypt(password)  # 使用随机密码加密PDF文件

    with open(output_file, 'wb') as output:
        pdf_writer.write(output)

    return password

input_file = 'input.pdf'  # 输入的PDF文件路径
output_file = 'output.pdf'  # 输出的加密后的PDF文件路径

password = protect_pdf_with_password(input_file, output_file)
print(f"生成的随机密码为：{password}")