for file in os.listdir(folder_path): if file.endswith(".enc"): enc_path = os.path.join(folder_path, file) with open(enc_path, "rb") as f: encrypted = f.read() decrypted = cipher.decrypt(encrypted) orig_path = enc_path[:-4] with open(orig_path, "wb") as f: f.write(decrypted) os.remove(enc_path)
with open(os.path.join(folder_path, ".salt"), "wb") as f: f.write(salt) with open(LOCKED_FLAG, "w") as f: f.write(folder_path) folder lock github
def lock_folder(folder_path, password): salt = os.urandom(16) key = derive_key(password, salt) cipher = Fernet(key) for file in os
