from Crypto.Random import get_random_bytes
from crypt import *
from compress import *

if not os.path.exists("key"):
    key = get_random_bytes(32)  # 32 bytes * 8 = 256 bits (1 byte = 8 bits)
    open("key", 'wb').write(key)
    print("Saved")
else:
    key = open("key", 'rb').read()
    print("Recovered")

print(key)
file = "original_file.txt"
cfile = file + '.enc'

encrypt_file(key, file, cfile)

decrypt_file(key, cfile)

rootdir = "test"
recurse(rootdir, key)

decrypt_file(key, "test/DressLens4.mp4.gz.enc", "test/DressLens4.mp4.gz")