使用Python实现对PDF文件进行密码保护

pip install PyPDF2

import PyPDF2

# 打开要保护的PDF文件
with open('example.pdf', 'rb') as pdf_file:
    # 创建PDF阅读对象
    pdf_reader = PyPDF2.PdfFileReader(pdf_file)
    
    # 创建PDF写入对象
    pdf_writer = PyPDF2.PdfFileWriter()

    # 设置密码
    pdf_writer.addPage(pdf_reader.getPage(0))
    pdf_writer.encrypt('mypassword')

    # 保存带有密码保护的PDF文件
    with open('protected_example.pdf', 'wb') as protected_pdf_file:
        pdf_writer.write(protected_pdf_file)

import PyPDF2

# 输入密码
password = input("请输入密码: ")

# 打开受保护的PDF文件
with open('protected_example.pdf', 'rb') as protected_pdf_file:
    # 创建PDF阅读对象
    pdf_reader = PyPDF2.PdfFileReader(protected_pdf_file)
    
    # 解锁PDF文件
    if pdf_reader.decrypt(password):
        # 打印PDF文件的总页数
        print(f"总页数: {pdf_reader.numPages}")
    else:
        print("密码错误，无法打开文件。")

import PyPDF2

# 输入原始密码和新密码
original_password = input("请输入原始密码: ")
new_password = input("请输入新密码（按回车跳过）: ")

# 打开原始受保护的PDF文件
with open('protected_example.pdf', 'rb') as protected_pdf_file:
    # 创建PDF阅读对象
    pdf_reader = PyPDF2.PdfFileReader(protected_pdf_file)
    
    # 解锁PDF文件
    if pdf_reader.decrypt(original_password):
        # 创建PDF写入对象
        pdf_writer = PyPDF2.PdfFileWriter()
        
        # 复制页面到新文件
        for page_num in range(pdf_reader.numPages):
            page = pdf_reader.getPage(page_num)
            pdf_writer.addPage(page)
        
        # 设置新密码（如果提供了新密码）
        if new_password:
            pdf_writer.encrypt(new_password)
        
        # 保存新的PDF文件
        with open('modified_example.pdf', 'wb') as modified_pdf_file:
            pdf_writer.write(modified_pdf_file)
        
        print("PDF文件密码已修改并保存为'modified_example.pdf'。")
    else:
        print("原始密码错误，无法修改文件密码。")

import PyPDF2

# 输入密码
password = input("请输入密码: ")

# 打开受保护的PDF文件
with open('protected_example.pdf', 'rb') as protected_pdf_file:
    # 创建PDF阅读对象
    pdf_reader = PyPDF2.PdfFileReader(protected_pdf_file)
    
    # 解锁PDF文件
    if pdf_reader.decrypt(password):
        # 创建PDF写入对象
        pdf_writer = PyPDF2.PdfFileWriter()
        
        # 复制页面到新文件
        for page_num in range(pdf_reader.numPages):
            page = pdf_reader.getPage(page_num)
            pdf_writer.addPage(page)
        
        # 保存新的PDF文件（不设置密码）
        with open('unprotected_example.pdf', 'wb') as unprotected_pdf_file:
            pdf_writer.write(unprotected_pdf_file)
        
        print("密码已删除，PDF文件保存为'unprotected_example.pdf'。")
    else:
        print("密码错误，无法删除密码。")

import PyPDF2
import os

# 指定PDF文件所在文件夹路径
pdf_folder = 'pdf_files'

# 输入密码（或原始密码和新密码）
password = input("请输入密码（或原始密码和新密码，以逗号分隔）: ")

# 将输入的密码拆分为原始密码和新密码
passwords = password.split(',')

# 获取PDF文件夹中的所有PDF文件
pdf_files = [f for f in os.listdir(pdf_folder) if f.endswith('.pdf')]

for pdf_file in pdf_files:
    pdf_path = os.path.join(pdf_folder, pdf_file)
    
    # 打开PDF文件
    with open(pdf_path, 'rb') as pdf_file:
        # 创建PDF阅读对象
        pdf_reader = PyPDF2.PdfFileReader(pdf_file)
        
        # 解锁PDF文件
        if pdf_reader.decrypt(passwords[0]):
            # 创建PDF写入对象
            pdf_writer = PyPDF2.PdfFileWriter()
            
            # 复制页面到新文件
            for page_num in range(pdf_reader.numPages):
                page = pdf_reader.getPage(page_num)
                pdf_writer.addPage(page)
            
            # 设置新密码（如果提供了新密码）
            if len(passwords) > 1:
                pdf_writer.encrypt(passwords[1])
            
            # 保存新的PDF文件
            output_path = os.path.join(pdf_folder, f'unlocked_{pdf_file}')
            with open(output_path, 'wb') as output_pdf_file:
                pdf_writer.write(output_pdf_file)
            
            print(f"已处理文件：{pdf_file}，保存为：{output_path}")
        else:
            print(f"密码错误，无法处理文件：{pdf_file}")