Flush the session in get_pk_values()

epandurski · Feb 23, 2019 · e32e52d · e32e52d
1 parent c0302b8
commit e32e52d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/flask_signalbus/atomic.py b/flask_signalbus/atomic.py
@@ -42,6 +42,7 @@ def get_pk_values(cls, instance_or_pk):
         """Return a primary key as a tuple when given any instance or primary key."""
 
         if isinstance(instance_or_pk, cls):
+            cls._flask_signalbus_sa.session.flush()
             instance_or_pk = inspect(cls).primary_key_from_instance(instance_or_pk)
         return instance_or_pk if isinstance(instance_or_pk, tuple) else (instance_or_pk,)