- [feature] New session events after_transaction_create

    and after_transaction_end
    allows tracking of new SessionTransaction objects.
    If the object is inspected, can be used to determine
    when a session first becomes active and when
    it deactivates.

1 files changed, 9 insertions, 1 deletions

diff --git a/lib/sqlalchemy/orm/session.py b/lib/sqlalchemy/orm/session.py
index e0f79cd8a..36ff5fc84 100644
--- a/lib/sqlalchemy/orm/session.py
+++ b/lib/sqlalchemy/orm/session.py
@@ -150,6 +150,9 @@ class SessionTransaction(object):
         if self.session._enable_transaction_accounting:
             self._take_snapshot()
 
+        if self.session.dispatch.after_transaction_create:
+            self.session.dispatch.after_transaction_create(self.session, self)
+
     @property
     def is_active(self):
         return self.session is not None and self._active
@@ -379,9 +382,14 @@ class SessionTransaction(object):
                     connection.close()
                 else:
                     transaction.close()
+
+        self._deactivate()
+        if self.session.dispatch.after_transaction_end:
+            self.session.dispatch.after_transaction_end(self.session, self)
+
+        if self._parent is None:
             if not self.session.autocommit:
                 self.session.begin()
-        self._deactivate()
         self.session = None
         self._connections = None