summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/engine/threadlocal.py
diff options
context:
space:
mode:
Diffstat (limited to 'lib/sqlalchemy/engine/threadlocal.py')
-rw-r--r--lib/sqlalchemy/engine/threadlocal.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/lib/sqlalchemy/engine/threadlocal.py b/lib/sqlalchemy/engine/threadlocal.py
index 505d1fadd..badfde8dd 100644
--- a/lib/sqlalchemy/engine/threadlocal.py
+++ b/lib/sqlalchemy/engine/threadlocal.py
@@ -64,6 +64,8 @@ class TLEngine(base.Engine):
self.pool.connect, connection),
**kw)
self._connections.conn = weakref.ref(connection)
+ else:
+ connection.should_close_with_result = kw.get('close_with_result', False)
return connection._increment_connect()
@@ -81,11 +83,9 @@ class TLEngine(base.Engine):
self.contextual_connect().begin_nested())
return self
- def begin(self):
- if not hasattr(self._connections, 'trans'):
- self._connections.trans = []
- self._connections.trans.append(self.contextual_connect().begin())
- return self
+ def begin(self, close_with_result=False):
+ conn = self.contextual_connect()
+ return base.Engine._trans_ctx(conn, conn.begin(), close_with_result)
def __enter__(self):
return self