bugfix: get_session() contextmgr usage in consume_signals() (#4888)

Netflix · Sep 25, 2024 · 7631468 · 7631468
1 parent 9c14cf3
commit 7631468
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/dispatch/cli.py b/src/dispatch/cli.py
@@ -924,8 +924,8 @@ def consume_signals():
     from dispatch.database.core import get_session, get_organization_session
 
     install_plugins()
-    db_session = get_session()
-    organizations = get_all_organizations(db_session=db_session)
+    with get_session() as session:
+        organizations = get_all_organizations(db_session=session)
 
     log = logging.getLogger(__name__)