thomasabishop/eolas-db
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feature: tidy up controller

Browse source
This commit is contained in:
thomasabishop 2024-11-07 17:28:02 +00:00
parent a954aafbcc
commit 033a8aaaef
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/controllers/controller.py
View file

@ -13,8 +13,16 @@ class Controller:
def populate_database(self): def populate_database(self):
connection = self.database_service.connect() connection = self.database_service.connect()
if connection is None: try:
raise Exception("Failed to establish database connection") if connection is None:
raise Exception("Failed to establish database connection")
sqlite_service = SqliteService(connection)
sqlite_service.truncate_tables()
sqlite_service.create_tables()
sqlite_service = SqliteService(connection) except Exception as e:
sqlite_service.create_tables() raise Exception(e)
finally:
if connection is not None:
self.database_service.disconnect()