1 files changed, 7 insertions, 2 deletions

diff --git a/db/session.py b/db/session.py
index 4d83649..ec700bd 100644
--- a/db/session.py
+++ b/db/session.py
@@ -17,5 +17,10 @@ def get_session():
         yield session
 
 @contextmanager
-def get_session_context():
-    yield from get_session()
+def get_session_context(autocommit=True):
+    session = next(get_session())
+
+    yield session
+
+    if autocommit:
+        session.commit()