Update to simplify the connection management.

This commit is contained in:
2025-01-07 21:50:22 +02:00
parent ccede52883
commit b10726796d
2 changed files with 3 additions and 3 deletions
+2 -1
View File
@@ -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']
@@ -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)