lionel/cryptAES
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Optimize import

Browse Source
This commit is contained in:
LPOCHOLLE 2021-10-12 10:10:33 +02:00
parent bf57b17f99
commit 99ed3997ce
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Backup.py
View File

@ -1,10 +1,14 @@
import io
import os
import re import re
import tempfile
import mgzip as gzip import mgzip as gzip
import tarfile import tarfile
import sqlite3 import sqlite3
from crypt import *
import pathlib import pathlib
import crypt
class Backup: class Backup:
key = None key = None
@ -86,13 +90,13 @@ def uncompress(file):
def crypt(file, key): def crypt(file, key):
encrypted_file = tempfile.SpooledTemporaryFile() encrypted_file = tempfile.SpooledTemporaryFile()
encrypt_file(key, file, encrypted_file) crypt.encrypt_file(key, file, encrypted_file)
return encrypted_file return encrypted_file
def uncrypt(file, key): def uncrypt(file, key):
decrypted_file = tempfile.SpooledTemporaryFile() decrypted_file = tempfile.SpooledTemporaryFile()
decrypt_file(key, file, decrypted_file) crypt.decrypt_file(key, file, decrypted_file)
return decrypted_file return decrypted_file

2
main.py
View File

@ -1,6 +1,6 @@
from Crypto.Random import get_random_bytes from Crypto.Random import get_random_bytes
import os import os
from Backup import * from Backup import Backup
if not os.path.exists("key"): if not os.path.exists("key"):
key = get_random_bytes(32) # 32 bytes * 8 = 256 bits (1 byte = 8 bits) key = get_random_bytes(32) # 32 bytes * 8 = 256 bits (1 byte = 8 bits)