Python文件夹加密与解密代码指南

在现代社会中，数据安全变得越来越重要。为了保护个人隐私和机密信息，文件夹加密是一种常见做法。本文将介绍如何使用Python实现在文件夹上加密和解密的简单方法。

1. 加密与解密的基本原理

文件夹加密的核心思想是使用密码对文件夹中的文件进行处理，只有输入正确的密码才能恢复文件。实际上，加密可以通过对文件内容进行编码实现，常用的加密算法包括对称加密和非对称加密。

关系图示例

在加密与解密的过程中，密码管理、文件内容以及加密后的文件是关键的组成部分。以下是这些元素之间的关系图：

erDiagram
    PASSWORDEFFECTS {
        string password
    }
    FILECONTENT {
        string content
    }
    ENCRYPTEDFILE {
        string encryptedContent
    }

    PASSWORDEFFECTS ||--o{ FILECONTENT : encrypts
    PASSWORDEFFECTS ||--o{ ENCRYPTEDFILE : generates
    FILECONTENT ||--o{ ENCRYPTEDFILE : transforms

2. Python实现加密和解密

下面是用Python对文件夹进行加密和解密的基本示例。我们将使用cryptography库提供的对称加密算法来实现这一功能。

安装所需库

在开始之前，确保你已经安装了cryptography库。可以使用以下命令进行安装：

pip install cryptography

加密代码示例

以下是一个简单的代码示例，演示如何加密文件夹中的文件。

import os
from cryptography.fernet import Fernet

def generate_key():
    return Fernet.generate_key()

def encrypt_file(file_path, key):
    with open(file_path, 'rb') as file:
        original_data = file.read()
    fernet = Fernet(key)
    encrypted_data = fernet.encrypt(original_data)
    with open(file_path, 'wb') as file:
        file.write(encrypted_data)

def encrypt_folder(folder_path, key):
    for filename in os.listdir(folder_path):
        file_path = os.path.join(folder_path, filename)
        if os.path.isfile(file_path):
            encrypt_file(file_path, key)

# 使用示例
key = generate_key()
encrypt_folder('path_to_your_folder', key)
print(f'Encryption completed. Key: {key.decode()}')

解密代码示例

为了让加密操作有效，必须正确解密文件。以下是解密文件夹中的文件的代码示例：

def decrypt_file(file_path, key):
    with open(file_path, 'rb') as file:
        encrypted_data = file.read()
    fernet = Fernet(key)
    decrypted_data = fernet.decrypt(encrypted_data)
    with open(file_path, 'wb') as file:
        file.write(decrypted_data)

def decrypt_folder(folder_path, key):
    for filename in os.listdir(folder_path):
        file_path = os.path.join(folder_path, filename)
        if os.path.isfile(file_path):
            decrypt_file(file_path, key)

# 使用示例
decrypt_folder('path_to_your_folder', key)
print('Decryption completed.')

序列图示例

加密和解密的过程涉及一系列操作，下面的序列图展示了这个流程：

sequenceDiagram
    A->>B: Generate Encryption Key
    B->>A: Return Key
    A->>B: Encrypt Files in Folder
    B->>B: Read each file
    B->>B: Encrypt file content
    B->>B: Save encrypted file
    A->>B: Decrypt Files in Folder
    B->>B: Read each file
    B->>B: Decrypt file content
    B->>B: Save decrypted file

结论

使用Python进行文件夹的加密和解密是一个相对简单而有效的方式。在本文中，我们展示了如何生成加密密钥、加密文件夹以及解密文件夹的基本示例。这不仅为个人隐私保护提供了保障，也为数据安全管理提供了技术支持。希望这篇文章对你有所帮助，让我们更好地保护自己的数字资产。