如何在目录中加密嵌套文件夹

Question

我正在尝试制作一个可以加密目录的程序。但是，我目前只能在目录中加密文件，而不得对嵌套的文件夹进行加密。一个例子是，如果您想加密“图片”文件夹，它也将加密“保存的图片”和“相机滚动”以及“图片”中的文件。任何帮助将不胜感激

from cryptography.fernet import Fernet
import os

'''
def write_key():
    key = Fernet.generate_key()
    with open('key.key', "wb") as key_file:
        key_file.write(key)
'''


def load_key():
    file = open('key.key', 'rb')
    key_ = file.read()
    file.close()
    return key_


key = load_key()
fer = Fernet(key)

os.chdir(r"C:\Users\{user}\encrypt_test".format(user=os.getlogin()))
files = os.listdir()
print(files)

for i in range(len(files)):

    with open(files[i], 'rb') as f:
        data = f.read()

    with open(files[i], 'w') as f:
        print(len(data))
        f.write(fer.encrypt(data).decode())
        print('done')

Answer 1

您要做的是递归浏览目录以查找所有文件并对其采取一些操作。 os.listdir 您已经看到的方法只给您是直接目录中的条目。虽然您可以使用该方法编写递归方法（例如检查给定的条目是否是目录，如果是的，则在其上运行os.listdir，Else Encrypt，重复），但必须更易于使用 os.walk ，它将处理递归下方的递归对于您来说，以及将文件与每个级别的目录分开。这里要注意的一件事是files os.walk返回的数组只是该级别的文件的名称，因此您需要将其与<<代码> root 值以获取完整的路径。

重写您的代码以使用该渲染：

from cryptography.fernet import Fernet
import os
import path

'''
def write_key():
    key = Fernet.generate_key()
    with open('key.key', "wb") as key_file:
        key_file.write(key)
'''


def load_key():
    file = open('key.key', 'rb')
    key_ = file.read()
    file.close()
    return key_


key = load_key()
fer = Fernet(key)

os.chdir(r"C:\Users\{user}\encrypt_test".format(user=os.getlogin()))

for root, dirs, files in os.walk('.'):
    print(files)
    for filename in files:
        filepath = path.join(root, filename)

        with open(filepath, 'rb') as f:
            data = f.read()

        with open(filepath, 'w') as f:
            print(len(data))
            f.write(fer.encrypt(data).decode())
            print('done')