Added functionality for saving and restoring project files

This commit is contained in:
Zvonimir Sabljic
2023-08-02 16:22:42 +02:00
parent 6cf8de709c
commit cdb6f83c61
4 changed files with 77 additions and 2 deletions

View File

@@ -13,6 +13,7 @@ from database.models.development_planning import DevelopmentPlanning
from database.models.development_steps import DevelopmentSteps
from database.models.environment_setup import EnvironmentSetup
from database.models.development import Development
from database.models.file_snapshot import FileSnapshot
def save_user(user_id, email="email", password="password"):
@@ -139,13 +140,14 @@ def create_tables():
DevelopmentPlanning,
DevelopmentSteps,
EnvironmentSetup,
Development
Development,
FileSnapshot,
])
def drop_tables():
with database.atomic():
for table in [User, App, ProjectDescription, UserStories, UserTasks, Architecture, DevelopmentPlanning, DevelopmentSteps, EnvironmentSetup, Development]:
for table in [User, App, ProjectDescription, UserStories, UserTasks, Architecture, DevelopmentPlanning, DevelopmentSteps, EnvironmentSetup, Development, FileSnapshot]:
database.execute_sql(f'DROP TABLE IF EXISTS "{table._meta.table_name}" CASCADE')