31 lines
790 B
Python
31 lines
790 B
Python
|
|
import os
|
|
from zipfile import ZipFile
|
|
|
|
from models.database import Database
|
|
|
|
class Utils:
|
|
@staticmethod
|
|
def clear_screen():
|
|
os.system('cls' if os.name == 'nt' else 'clear')
|
|
|
|
@staticmethod
|
|
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():
|
|
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") |