fixed backup

This commit is contained in:
2022-10-07 23:23:57 +02:00
parent 87229a654e
commit 155a86ab06
2 changed files with 21 additions and 21 deletions

View File

@@ -1,5 +1,6 @@
import os
from zipfile import ZipFile
from models.database import Database
@@ -9,13 +10,22 @@ class Utils:
os.system('cls' if os.name == 'nt' else 'clear')
@staticmethod
def export_db(file):
with open(f"./{file}", 'w') as f:
def export_db():
with open(f"./backup.sql", 'w') as f:
for line in Database.connection.iterdump():
f.write('%s\n' % line)
with ZipFile("backup.zip", "w") as zip:
zip.write("backup.sql")
os.remove("./backup.sql")
@staticmethod
def import_db(file):
with open(f"./{file}", 'r') as f:
str = f.read()
Database.connection.executescript(str)
def import_db():
with ZipFile("backup.zip", 'r') as zip:
zip.extractall("./")
with open(f"./backup.sql", 'r') as f:
str = f.read()
Database.connection.executescript(str)
os.remove("./backup.sql")