From 033a8aaaef4bf0b719e8598027e9df12eb533feb Mon Sep 17 00:00:00 2001 From: thomasabishop Date: Thu, 7 Nov 2024 17:28:02 +0000 Subject: [PATCH] feature: tidy up controller --- src/controllers/controller.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/src/controllers/controller.py b/src/controllers/controller.py index 038886f..110cce3 100644 --- a/src/controllers/controller.py +++ b/src/controllers/controller.py @@ -13,8 +13,16 @@ class Controller: def populate_database(self): connection = self.database_service.connect() - if connection is None: - raise Exception("Failed to establish database connection") + try: + 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) - sqlite_service.create_tables() + except Exception as e: + raise Exception(e) + + finally: + if connection is not None: + self.database_service.disconnect()