feature: tidy up controller
This commit is contained in:
parent
a954aafbcc
commit
033a8aaaef
1 changed files with 12 additions and 4 deletions
|
@ -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()
|
||||
|
|
Loading…
Add table
Reference in a new issue