feature: tidy up controller

This commit is contained in:
thomasabishop 2024-11-07 17:28:02 +00:00
parent a954aafbcc
commit 033a8aaaef

View file

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