python如何批量文档加密

一、PYTHON批量文档加密的方法概述

Python批量文档加密可以通过使用PyPDF2库对PDF文档进行加密、使用Fernet模块对文本文件进行加密、结合os库批量处理文件等方法实现。其中，使用PyPDF2库是处理PDF文档的一种常见方式，而Fernet模块则适用于加密文本文件。以下将详细介绍使用Fernet模块对文本文件进行加密的方法。

Fernet是Python中cryptography库的一个对称加密模块，它提供了一种简单易用的加密和解密方法。使用Fernet进行加密的步骤包括：生成密钥、读取文件内容、加密文件内容、保存加密后的内容。首先，我们需要生成一个密钥，该密钥将用于文件内容的加密和解密。然后，读取目标文件的内容，并使用生成的密钥对内容进行加密。最后，将加密后的内容保存回文件中或另存为新文件。

二、使用PyPDF2库对PDF文档加密

1、安装和导入PyPDF2库

在开始使用PyPDF2库对PDF文档进行加密之前，您需要确保已经安装了该库。可以使用pip命令进行安装：

pip install PyPDF2

在代码中导入PyPDF2模块：

import PyPDF2

2、读取和加密PDF文件

使用PyPDF2库加密PDF文件主要涉及读取原始文件和设置密码。以下是一个简单的示例代码：

def encrypt_pdf(input_file, output_file, password):
    # 打开原始PDF文件
    with open(input_file, 'rb') as infile:
        reader = PyPDF2.PdfReader(infile)
        writer = PyPDF2.PdfWriter()
        # 将所有页面添加到新PDF对象中
        for page in range(len(reader.pages)):
            writer.add_page(reader.pages[page])
        # 设置密码
        writer.encrypt(password)
        # 写入加密后的PDF文件
        with open(output_file, 'wb') as outfile:
            writer.write(outfile)

在这个函数中，我们读取了输入文件并创建了一个新的PDF写入对象。通过遍历原始PDF中的所有页面并将其添加到新PDF中，然后使用encrypt()方法设置密码，最后将加密后的内容写入输出文件。

3、批量加密PDF文件

为了批量加密多个PDF文件，可以使用os库遍历目标目录下的所有PDF文件，并调用上面的函数进行加密：

import os
def batch_encrypt_pdfs(directory, password):
    for filename in os.listdir(directory):
        if filename.endswith('.pdf'):
            input_file = os.path.join(directory, filename)
            output_file = os.path.join(directory, f'encrypted_{filename}')
            encrypt_pdf(input_file, output_file, password)
使用示例
batch_encrypt_pdfs('/path/to/pdf/files', 'yourpassword')

这里的batch_encrypt_pdfs函数接收目录路径和密码作为参数，并对该目录下的所有PDF文件进行加密处理。

三、使用Fernet模块对文本文件进行加密

1、安装和导入cryptography库

首先，确保您已经安装了cryptography库。可以使用以下命令进行安装：

pip install cryptography

在代码中导入Fernet模块：

from cryptography.fernet import Fernet

2、生成密钥并加密文件

在加密文件之前，您需要生成一个Fernet密钥。以下是生成密钥并对文件进行加密的示例代码：

def generate_key():
    return Fernet.generate_key()
def encrypt_file(input_file, output_file, key):
    # 创建Fernet对象
    fernet = Fernet(key)
    # 读取文件内容
    with open(input_file, 'rb') as file:
        file_data = file.read()
    # 加密文件内容
    encrypted_data = fernet.encrypt(file_data)
    # 写入加密后的内容
    with open(output_file, 'wb') as file:
        file.write(encrypted_data)

在这个例子中，generate_key()函数用于生成一个新的Fernet密钥。encrypt_file()函数则使用该密钥对文件内容进行加密，并将加密后的内容写入输出文件。

3、批量加密文本文件

您可以使用os库遍历目标目录下的所有文本文件，并调用上面的函数进行加密：

def batch_encrypt_files(directory, key):
    for filename in os.listdir(directory):
        if filename.endswith('.txt'):
            input_file = os.path.join(directory, filename)
            output_file = os.path.join(directory, f'encrypted_{filename}')
            encrypt_file(input_file, output_file, key)
使用示例
key = generate_key()
batch_encrypt_files('/path/to/text/files', key)

这里的batch_encrypt_files函数接收目录路径和密钥作为参数，并对该目录下的所有文本文件进行加密处理。

四、结合os库进行文件遍历和处理

1、获取目录下的所有文件

使用os库可以轻松获取目录下的所有文件。以下是一个简单的获取文件列表的示例：

import os
def list_files(directory):
    return [os.path.join(directory, f) for f in os.listdir(directory) if os.path.isfile(os.path.join(directory, f))]

2、根据文件类型选择不同的加密方法

在批量处理文件时，可能需要根据文件类型选择不同的加密方法。例如，对于PDF文件使用PyPDF2，而对于文本文件使用Fernet。以下是一个示例代码：

def batch_encrypt(directory, pdf_password, fernet_key):
    for filepath in list_files(directory):
        if filepath.endswith('.pdf'):
            output_file = os.path.join(directory, f'encrypted_{os.path.basename(filepath)}')
            encrypt_pdf(filepath, output_file, pdf_password)
        elif filepath.endswith('.txt'):
            output_file = os.path.join(directory, f'encrypted_{os.path.basename(filepath)}')
            encrypt_file(filepath, output_file, fernet_key)
使用示例
pdf_password = 'yourpdfpassword'
fernet_key = generate_key()
batch_encrypt('/path/to/files', pdf_password, fernet_key)

通过这种方式，可以根据文件类型选择合适的加密方法，从而实现批量文档加密。

五、加密后的文件管理和解密

1、保存和管理加密后的文件

在批量加密过程中，您可能需要将加密后的文件保存到特定的目录中，以便于管理。可以使用os.makedirs()函数创建新的目录，并将加密后的文件移动到该目录中。

import shutil
def save_encrypted_files(directory, encrypted_directory):
    if not os.path.exists(encrypted_directory):
        os.makedirs(encrypted_directory)
    for filename in os.listdir(directory):
        if filename.startswith('encrypted_'):
            shutil.move(os.path.join(directory, filename), encrypted_directory)

2、解密文件

为了能够访问加密后的文件内容，您需要提供解密功能。以下是使用Fernet对文本文件进行解密的示例：

def decrypt_file(input_file, output_file, key):
    # 创建Fernet对象
    fernet = Fernet(key)
    # 读取加密文件内容
    with open(input_file, 'rb') as file:
        encrypted_data = file.read()
    # 解密内容
    decrypted_data = fernet.decrypt(encrypted_data)
    # 写入解密后的内容
    with open(output_file, 'wb') as file:
        file.write(decrypted_data)
使用示例
decrypt_file('/path/to/encrypted_file.txt', '/path/to/decrypted_file.txt', fernet_key)