快捷導(dǎo)航

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

更新時(shí)間：2023年11月15日 11:25:19 作者：很酷的站長(zhǎng)

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

隨著互聯(lián)網(wǎng)和科技的不斷發(fā)展，數(shù)據(jù)泄露已經(jīng)成為一個(gè)嚴(yán)重的問題。為了保護(hù)用戶和公司的隱私，給文件夾加密已經(jīng)成為一個(gè)必要的步驟。本文將介紹如何利用Python給文件夾加密。

一、生成密鑰

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

import hashlib

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

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

二、加密文件夾

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

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模塊來(lái)進(jìn)行加密。遞歸遍歷文件夾中的所有文件，對(duì)每個(gè)文件進(jìn)行加密并寫回原文件。

三、解密文件夾

如果需要對(duì)文件夾進(jìn)行解密，可以使用下面的代碼：

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()

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

四、完整代碼示例

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

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給文件夾加解密的實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)Python 文件夾加解密內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: