Fixed drop_tables function

This commit is contained in:
Zvonimir Sabljic
2023-08-02 10:41:46 +02:00
parent 885792bd5e
commit 84346163a3

View File

@@ -143,19 +143,10 @@ def create_tables():
def drop_tables():
with database:
database.drop_tables([
User,
App,
ProjectDescription,
UserStories,
UserTasks,
Architecture,
DevelopmentPlanning,
DevelopmentSteps,
EnvironmentSetup,
Development
])
with database.atomic():
for table in [User, App, ProjectDescription, UserStories, UserTasks, Architecture, DevelopmentPlanning, DevelopmentSteps, EnvironmentSetup, Development]:
database.execute_sql(f'DROP TABLE IF EXISTS "{table._meta.table_name}" CASCADE')
if __name__ == "__main__":