From 99ed3997ce5b32a4ae2efd9a94cf23cf1b3b53ca Mon Sep 17 00:00:00 2001
From: LPOCHOLLE <>
Date: Tue, 12 Oct 2021 10:10:33 +0200
Subject: [PATCH] Optimize import

---
 Backup.py | 10 +++++++---
 main.py   |  2 +-
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/Backup.py b/Backup.py
index 034c8b5..ff57850 100644
--- a/Backup.py
+++ b/Backup.py
@@ -1,10 +1,14 @@
+import io
+import os
 import re
+import tempfile
 import mgzip as gzip
 import tarfile
 import sqlite3
-from crypt import *
 import pathlib
 
+import crypt
+
 
 class Backup:
     key = None
@@ -86,13 +90,13 @@ def uncompress(file):
 
 def crypt(file, key):
     encrypted_file = tempfile.SpooledTemporaryFile()
-    encrypt_file(key, file, encrypted_file)
+    crypt.encrypt_file(key, file, encrypted_file)
     return encrypted_file
 
 
 def uncrypt(file, key):
     decrypted_file = tempfile.SpooledTemporaryFile()
-    decrypt_file(key, file, decrypted_file)
+    crypt.decrypt_file(key, file, decrypted_file)
     return decrypted_file
 
 
diff --git a/main.py b/main.py
index d9e853a..2a01316 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,6 @@
 from Crypto.Random import get_random_bytes
 import os
-from Backup import *
+from Backup import Backup
 
 if not os.path.exists("key"):
     key = get_random_bytes(32)  # 32 bytes * 8 = 256 bits (1 byte = 8 bits)