From b10726796d5079285529c1c24987ac9d6d77f309 Mon Sep 17 00:00:00 2001 From: KKlochko Date: Tue, 7 Jan 2025 21:50:22 +0200 Subject: [PATCH] Update to simplify the connection management. --- tui_rsync/infrastructure/orm/__init__.py | 3 ++- tui_rsync/infrastructure/orm/database_manager.py | 3 +-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tui_rsync/infrastructure/orm/__init__.py b/tui_rsync/infrastructure/orm/__init__.py index 2bd190d..38df868 100644 --- a/tui_rsync/infrastructure/orm/__init__.py +++ b/tui_rsync/infrastructure/orm/__init__.py @@ -1,4 +1,5 @@ from .database_manager import DatabaseManager from .in_memory_database_manager import InMemoryDatabaseManager +from .sqlite_database_manager import SqliteDatabaseManager -__all__ = ['DatabaseManager', 'InMemoryDatabaseManager'] +__all__ = ['DatabaseManager', 'InMemoryDatabaseManager', 'SqliteDatabaseManager'] diff --git a/tui_rsync/infrastructure/orm/database_manager.py b/tui_rsync/infrastructure/orm/database_manager.py index ca36e5a..af7a8c7 100644 --- a/tui_rsync/infrastructure/orm/database_manager.py +++ b/tui_rsync/infrastructure/orm/database_manager.py @@ -23,5 +23,4 @@ class DatabaseManager(DatabaseManagerPort): return self.db def create_tables(self): - with self.db: - self.db.create_tables(self.models, safe=True) + self.db.create_tables(self.models, safe=True)