如何用python给文件夹设置密码

如何用Python给文件夹设置密码

在Python中给文件夹设置密码的常见方法有：加密文件夹内容、使用第三方库、创建图形用户界面（GUI）。其中，加密文件夹内容是一种较为常见且简单的方法。具体来说，可以通过PyCrypto库对文件夹中的文件进行加密和解密，从而实现文件夹的密码保护。

加密文件夹内容：利用PyCrypto库对文件夹中的文件进行加密和解密操作，是实现文件夹密码保护的常见方法之一。

一、安装和准备工作

安装PyCrypto库

在进行加密和解密操作之前，需要安装PyCrypto库。可以使用以下命令进行安装：

pip install pycryptodome

导入必要的库

在Python代码中导入必要的库，包括os、hashlib、getpass和AES等。

import os
import hashlib
from getpass import getpass
from Crypto.Cipher import AES
from Crypto.Util.Padding import pad, unpad

二、加密文件夹内容

生成密钥和初始向量

生成一个密钥和初始向量（IV），用于加密和解密操作。密钥可以通过用户输入的密码生成。

def generate_key(password):
    return hashlib.sha256(password.encode()).digest()
def generate_iv():
    return os.urandom(16)

加密文件

对文件夹中的文件进行遍历，并逐个进行加密操作。

def encrypt_file(file_path, key, iv):
    cipher = AES.new(key, AES.MODE_CBC, iv)
    with open(file_path, 'rb') as f:
        plaintext = f.read()
    ciphertext = cipher.encrypt(pad(plaintext, AES.block_size))
    with open(file_path, 'wb') as f:
        f.write(iv + ciphertext)

加密文件夹中的所有文件

编写一个函数，遍历文件夹中的所有文件，并对每个文件进行加密。

def encrypt_folder(folder_path, password):
    key = generate_key(password)
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            file_path = os.path.join(root, file)
            iv = generate_iv()
            encrypt_file(file_path, key, iv)

三、解密文件夹内容

解密文件

对文件夹中的文件进行遍历，并逐个进行解密操作。

def decrypt_file(file_path, key):
    with open(file_path, 'rb') as f:
        iv = f.read(16)
        ciphertext = f.read()
    cipher = AES.new(key, AES.MODE_CBC, iv)
    plaintext = unpad(cipher.decrypt(ciphertext), AES.block_size)
    with open(file_path, 'wb') as f:
        f.write(plaintext)

解密文件夹中的所有文件

编写一个函数，遍历文件夹中的所有文件，并对每个文件进行解密。

def decrypt_folder(folder_path, password):
    key = generate_key(password)
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            file_path = os.path.join(root, file)
            decrypt_file(file_path, key)

四、创建图形用户界面（GUI）

安装tkinter库

可以使用tkinter库创建一个简单的图形用户界面，方便用户输入密码和选择文件夹。

pip install tk

编写GUI代码

编写一个简单的GUI代码，用于输入密码和选择需要加密或解密的文件夹。

import tkinter as tk
from tkinter import filedialog
def encrypt_action():
    folder_path = filedialog.askdirectory()
    password = getpass("Enter password: ")
    encrypt_folder(folder_path, password)
    print("Folder encrypted successfully.")
def decrypt_action():
    folder_path = filedialog.askdirectory()
    password = getpass("Enter password: ")
    decrypt_folder(folder_path, password)
    print("Folder decrypted successfully.")
root = tk.Tk()
root.title("Folder Encryption Tool")
encrypt_button = tk.Button(root, text="Encrypt Folder", command=encrypt_action)
encrypt_button.pack(pady=10)
decrypt_button = tk.Button(root, text="Decrypt Folder", command=decrypt_action)
decrypt_button.pack(pady=10)
root.mainloop()