快捷導(dǎo)航

Python給文件夾加解密的實現(xiàn)

更新時間：2023年11月15日 11:25:19 作者：很酷的站長

數(shù)據(jù)泄露已經(jīng)成為一個嚴(yán)重的問題,為了保護用戶和公司的隱私,給文件夾加密已經(jīng)成為一個必要的步驟,本文主要介紹了Python給文件夾加解密的實現(xiàn),感興趣的可以了解一下

一、生成密鑰

加密過程需要密鑰來進行加密和解密?？梢允褂肞ython內(nèi)置的hashlib模塊來生成一個隨機的密鑰。

import hashlib

key = hashlib.sha256(b'mysecretkey').digest()

這里使用sha256算法生成一個256位的密鑰。用戶可以自己設(shè)置密鑰，也可以使用random模塊生成隨機的密鑰。

二、加密文件夾

在有了密鑰之后，就可以對文件夾進行加密。加密的過程可以通過遞歸的方式對所有的文件進行加密。由于文件加密需要時間，因此需要使用tqdm模塊來顯示進度條。加密的過程如下：

import os
from cryptography.fernet import Fernet
from tqdm import tqdm

def encrypt_folder(path, key):
    f = Fernet(key)
    for root, _, files in os.walk(path):
        for file in tqdm(files):
            filepath = os.path.join(root, file)
            with open(filepath, 'rb+') as f1:
                data = f1.read()
                encrypted_data = f.encrypt(data)
                f1.seek(0)
                f1.write(encrypted_data)
                f1.truncate()

這里使用了cryptography.fernet模塊來進行加密。遞歸遍歷文件夾中的所有文件，對每個文件進行加密并寫回原文件。

三、解密文件夾

如果需要對文件夾進行解密，可以使用下面的代碼：

def decrypt_folder(path, key):
    f = Fernet(key)
    for root, _, files in os.walk(path):
        for file in tqdm(files):
            filepath = os.path.join(root, file)
            with open(filepath, 'rb+') as f1:
                data = f1.read()
                decrypted_data = f.decrypt(data)
                f1.seek(0)
                f1.write(decrypted_data)
                f1.truncate()

這里使用了和加密同樣的方式遞歸遍歷文件夾中的所有文件，對每個文件進行解密并寫回原文件。

四、完整代碼示例

下面給出完整的代碼示例：

import os
import hashlib
from cryptography.fernet import Fernet
from tqdm import tqdm

def generate_key():
    key = hashlib.sha256(b'mysecretkey').digest()
    return key

def encrypt_folder(path, key):
    f = Fernet(key)
    for root, _, files in os.walk(path):
        for file in tqdm(files):
            filepath = os.path.join(root, file)
            with open(filepath, 'rb+') as f1:
                data = f1.read()
                encrypted_data = f.encrypt(data)
                f1.seek(0)
                f1.write(encrypted_data)
                f1.truncate()

def decrypt_folder(path, key):
    f = Fernet(key)
    for root, _, files in os.walk(path):
        for file in tqdm(files):
            filepath = os.path.join(root, file)
            with open(filepath, 'rb+') as f1:
                data = f1.read()
                decrypted_data = f.decrypt(data)
                f1.seek(0)
                f1.write(decrypted_data)
                f1.truncate()

key = generate_key()
encrypt_folder('/path/to/folder', key)
decrypt_folder('/path/to/folder', key)

到此這篇關(guān)于Python給文件夾加解密的實現(xiàn)的文章就介紹到這了,更多相關(guān)Python 文件夾加解密內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: