Update to simplify the connection management.

dev
KKlochko 4 months ago
parent ccede52883
commit b10726796d

@ -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)

Loading…
Cancel
Save