Enable "future" SQLAlchemy API

smsearcy · Jun 23, 2023 · cebfd78 · cebfd78
1 parent 6c768df
commit cebfd78
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/meshinfo/models/__init__.py b/meshinfo/models/__init__.py
@@ -22,12 +22,11 @@
 
 
 def get_engine(settings):
-    return create_engine(**attr.asdict(settings))
+    return create_engine(future=True, **attr.asdict(settings))
 
 
 def get_session_factory(engine) -> sessionmaker:
-    factory = sessionmaker()
-    factory.configure(bind=engine)
+    factory = sessionmaker(engine, future=True)
     return factory
 
 

diff --git a/meshinfo/models/meta.py b/meshinfo/models/meta.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import pendulum
-from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import declarative_base
 from sqlalchemy.schema import MetaData
 from sqlalchemy.types import TIMESTAMP, TypeDecorator