PDF加密文件解密

这段代码使用了 PyPDF2 库来实现 PDF 文件的解密操作。用户可以输入要解密的 PDF 文件或包含 PDF 文件的文件夹路径，提供对应的密码，然后将解密后的 PDF 文件保存到指定的解密文件夹中。以下是代码的主要功能和步骤：

1. 导入了必要的模块：os 用于文件操作，PyPDF2 用于 PDF 文件操作。
2. 定义了 get_reader() 函数，用于获取 PDF 读取实例。如果文件是加密的，会要求提供密码进行解密。
3. 定义了 decrypt_pdf() 函数，用于将加密的 PDF 文件解密，并生成无需密码的新 PDF 文件。
4. 定义了 get_pdf_paths() 函数，用于获取文件夹中的所有 PDF 文件路径。
5. 主程序部分：

• 用户输入要解密的 PDF 文件或文件夹路径、密码以及解密后的文件夹路径。
• 如果输入的是文件夹路径，则获取文件夹中的所有 PDF 文件路径，并逐个进行解密。
• 如果输入的是单个 PDF 文件路径，则直接对该文件进行解密。
• 解密后的 PDF 文件将保存到指定的解密文件夹中。

注意事项：

• 在使用该代码之前，需要安装 PyPDF2 库，可以使用 pip install PyPDF2 命令进行安装。
• 代码中使用了命令行交互来获取用户输入，可以根据需要修改为图形界面输入。

这段代码实现了基本的 PDF 文件解密功能，你可以根据需要进行扩展和优化，例如添加更多的错误处理、界面美化等。同时，请注意确保使用合法的方式使用该代码，遵循相关法律法规。

import os
from PyPDF2 import PdfWriter, PdfReader
 
def get_reader(filename, password):
    try:
        old_file = open(filename, 'rb')
    except Exception as err:
        print('文件打开失败！' + str(err))
        return None
 
    # 创建读实例
    pdf_reader = PdfReader(old_file, strict=False)
 
    # 解密操作
    if pdf_reader.is_encrypted:
        if password is None:
            print('%s文件被加密，需要密码！' % filename)
            return None
        else:
            if pdf_reader.decrypt(password) != 1:
                print('%s密码不正确！' % filename)
                return None
 
    if old_file in locals():
        old_file.close()
 
    return pdf_reader
 
def decrypt_pdf(filename, password, decrypted_folder):
    """
    将加密的文件进行解密，并生成一个无需密码的pdf文件
    :param filename: 原先加密的pdf文件
    :param password: 对应的密码
    :param decrypted_folder: 解密后放置文件的文件夹路径
    :return:
    """
    # 生成一个Reader和Writer
    pdf_reader = get_reader(filename, password)
    if pdf_reader is None:
        return
 
    pdf_writer = PdfWriter()
    for page in pdf_reader.pages:
        pdf_writer.add_page(page)
 
    # 获取解密后的文件名
    file_name, _ = os.path.splitext(filename)
 
    if decrypted_folder is not None:
        decrypted_filepath = os.path.join(decrypted_folder, os.path.basename(file_name) + ".pdf")
    else:
        print('生成文件失败，请给出解密后放置文件夹')
        return
 
    # 写入新文件
    with open(decrypted_filepath, 'wb') as decrypted_file:
        pdf_writer.write(decrypted_file)
        print('解密成功' + decrypted_filepath)
 
# 获取文件夹中的pdf文件路径
def get_pdf_paths(folder_path):
    pdf_paths = []
    for root, _, files in os.walk(folder_path):
        for file in files:
            if file.endswith(".pdf"):
                pdf_path = os.path.join(root, file)
                pdf_paths.append(pdf_path)
    return pdf_paths
 
 
pdf_path = input('请输入要解密的pdf或含pdf文件夹路径：')
# pdf_path = '文件夹或文件路径'
password = input('请输入密码：')
# password = '密码'
decrypted_folder = input('请输入解密文件夹：')
# decrypted_folder = '解密文件夹'
 
if os.path.isdir(pdf_path):
    pdf_paths = get_pdf_paths(pdf_path)
    for path in pdf_paths:
        decrypt_pdf(path, password, decrypted_folder)  # 修改此处，传递解密后文件存放的文件夹路径
else:
    _, extension = os.path.splitext(pdf_path)
    if extension.lower() == '.pdf':
        decrypt_pdf(pdf_path, password, decrypted_folder)
    else:
        print('文件非pdf')