Fixes ponyorm#432: flask can trigger teardown_request without real re…

…quest.
shishironline · Mar 24, 2019 · 5618cce · 5618cce
1 parent 0b9667b
commit 5618cce
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pony/flask/__init__.py b/pony/flask/__init__.py
@@ -8,9 +8,8 @@ def _enter_session():
 
 def _exit_session(exception):
     session = getattr(request, 'pony_session', None)
-    if session is None:
-        raise RuntimeError('Request object lost db_session')
-    session.__exit__(exc=exception)
+    if session is not None:
+        session.__exit__(exc=exception)
 
 class Pony(object):
     def __init__(self, app=None):