diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2007-02-25 22:44:52 +0000 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2007-02-25 22:44:52 +0000 |
| commit | 962c22c9eda7d2ab7dc0b41bd1c7a52cf0c9d008 (patch) | |
| tree | f0ab113c7947c80dfea42d4a1bef52217bf6ed96 /lib/sqlalchemy/engine/threadlocal.py | |
| parent | 8fa3becd5fac57bb898a0090bafaac377b60f070 (diff) | |
| download | sqlalchemy-962c22c9eda7d2ab7dc0b41bd1c7a52cf0c9d008.tar.gz | |
migrated (most) docstrings to pep-257 format, docstring generator using straight <pre> + trim() func
for now. applies most of [ticket:214], compliemnts of Lele Gaifax
Diffstat (limited to 'lib/sqlalchemy/engine/threadlocal.py')
| -rw-r--r-- | lib/sqlalchemy/engine/threadlocal.py | 69 |
1 files changed, 54 insertions, 15 deletions
diff --git a/lib/sqlalchemy/engine/threadlocal.py b/lib/sqlalchemy/engine/threadlocal.py index beac3ee3f..2bbb1ed43 100644 --- a/lib/sqlalchemy/engine/threadlocal.py +++ b/lib/sqlalchemy/engine/threadlocal.py @@ -2,19 +2,24 @@ from sqlalchemy import schema, exceptions, util, sql, types import StringIO, sys, re from sqlalchemy.engine import base, default -"""provides a thread-local transactional wrapper around the basic ComposedSQLEngine. multiple calls to engine.connect() -will return the same connection for the same thread. also provides begin/commit methods on the engine itself -which correspond to a thread-local transaction.""" +"""Provide a thread-local transactional wrapper around the basic ComposedSQLEngine. + +Multiple calls to engine.connect() will return the same connection for +the same thread. also provides begin/commit methods on the engine +itself which correspond to a thread-local transaction. +""" class TLSession(object): def __init__(self, engine): self.engine = engine self.__tcount = 0 + def get_connection(self, close_with_result=False): try: return self.__transaction._increment_connect() except AttributeError: return TLConnection(self, close_with_result=close_with_result) + def reset(self): try: self.__transaction._force_close() @@ -23,20 +28,24 @@ class TLSession(object): except AttributeError: pass self.__tcount = 0 + def in_transaction(self): return self.__tcount > 0 + def begin(self): if self.__tcount == 0: self.__transaction = self.get_connection() self.__trans = self.__transaction._begin() self.__tcount += 1 return self.__trans + def rollback(self): if self.__tcount > 0: try: self.__trans._rollback_impl() finally: self.reset() + def commit(self): if self.__tcount == 1: try: @@ -45,6 +54,7 @@ class TLSession(object): self.reset() elif self.__tcount > 1: self.__tcount -= 1 + def is_begun(self): return self.__tcount > 0 @@ -53,67 +63,96 @@ class TLConnection(base.Connection): base.Connection.__init__(self, session.engine, close_with_result=close_with_result) self.__session = session self.__opencount = 1 + session = property(lambda s:s.__session) + def _increment_connect(self): self.__opencount += 1 return self + def _create_transaction(self, parent): return TLTransaction(self, parent) + def _begin(self): return base.Connection.begin(self) + def in_transaction(self): return self.session.in_transaction() + def begin(self): return self.session.begin() + def close(self): if self.__opencount == 1: base.Connection.close(self) self.__opencount -= 1 + def _force_close(self): self.__opencount = 0 base.Connection.close(self) - + class TLTransaction(base.Transaction): def _commit_impl(self): base.Transaction.commit(self) + def _rollback_impl(self): base.Transaction.rollback(self) + def commit(self): self.connection.session.commit() + def rollback(self): self.connection.session.rollback() - + class TLEngine(base.Engine): - """an Engine that includes support for thread-local managed transactions. This engine - is better suited to be used with threadlocal Pool object.""" + """An Engine that includes support for thread-local managed transactions. + + This engine is better suited to be used with threadlocal Pool + object. + """ + def __init__(self, *args, **kwargs): - """the TLEngine relies upon the ConnectionProvider having "threadlocal" behavior, - so that once a connection is checked out for the current thread, you get that same connection - repeatedly.""" + """The TLEngine relies upon the ConnectionProvider having + "threadlocal" behavior, so that once a connection is checked out + for the current thread, you get that same connection + repeatedly. + """ + super(TLEngine, self).__init__(*args, **kwargs) self.context = util.ThreadLocal() + def raw_connection(self): - """returns a DBAPI connection.""" + """Return a DBAPI connection.""" + return self.connection_provider.get_connection() + def connect(self, **kwargs): - """returns a Connection that is not thread-locally scoped. this is the equilvalent to calling - "connect()" on a ComposedSQLEngine.""" + """Return a Connection that is not thread-locally scoped. + + This is the equivalent to calling ``connect()`` on a + ComposedSQLEngine. + """ + return base.Connection(self, self.connection_provider.unique_connection()) def _session(self): if not hasattr(self.context, 'session'): self.context.session = TLSession(self) return self.context.session + session = property(_session, doc="returns the current thread's TLSession") def contextual_connect(self, **kwargs): - """returns a TLConnection which is thread-locally scoped.""" + """Return a TLConnection which is thread-locally scoped.""" + return self.session.get_connection(**kwargs) - + def begin(self): return self.session.begin() + def commit(self): self.session.commit() + def rollback(self): self.session.rollback() |
