changed the session maker to reuse engines

StampyAI · Aug 18, 2023 · b5c0457 · b5c0457
1 parent 8ac789f
commit b5c0457
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/align_data/db/session.py b/align_data/db/session.py
@@ -10,11 +10,12 @@
 
 logger = logging.getLogger(__name__)
 
+ENGINE = create_engine(DB_CONNECTION_URI, echo=False)
+
 
 @contextmanager
 def make_session(auto_commit=False):
-    engine = create_engine(DB_CONNECTION_URI, echo=False)
-    with Session(engine).no_autoflush as session:
+    with Session(ENGINE).no_autoflush as session:
         yield session
         if auto_commit:
             session.commit()