def encrypt(backup: str, password: str):
return crypt.encrypt(backup, password)
arquivos = []
encriptografados = []
for i in range(0, 10):
arquivo_atual = 'dados_teste' + str(i) + '.txt'
arquivos.append(arquivo_atual)
encriptografados.append(arquivo_atual + '.enc')
with open(arquivo_atual, 'wb') as f:
for j in range(0, 10*(i+1)):
f.write(''.join(random.choices(string.ascii_uppercase + string.digits, k=10*(i + 1))).encode('utf-8'))
f.write('\n'.encode('utf-8'))
# Criptografa todos os arquivos gerados
i = 0
for arq in arquivos:
print('Encriptografando arquivo {0}'.format(arq))
encrypt(arq, str(i))
i += 1
i = 0
# Descriptografa os arquivos encriptografados
descriptografados = []
for arq in encriptografados:
descriptografados.append('descripto_' + arquivos[i])
print('Descriptografando arquivo {0}'.format(arq))
decrypt(arq, 'descripto_', 'chave' + str(i) + '.bin')
i += 1
# Compara os arquivos descriptogrados com os originais
for i in range(0, 10):
arquivo_original = arquivos[i]
arquivo_descriptografado = descriptografados[i]