283 lines
9.0 KiB
Python
283 lines
9.0 KiB
Python
"""
|
|
This module implements a transaction manager that can be used to define
|
|
transaction handling in a request or view function. It is used by transaction
|
|
control middleware and decorators.
|
|
|
|
The transaction manager can be in managed or in auto state. Auto state means the
|
|
system is using a commit-on-save strategy (actually it's more like
|
|
commit-on-change). As soon as the .save() or .delete() (or related) methods are
|
|
called, a commit is made.
|
|
|
|
Managed transactions don't do those commits, but will need some kind of manual
|
|
or implicit commits or rollbacks.
|
|
"""
|
|
|
|
import warnings
|
|
|
|
from functools import wraps
|
|
|
|
from django.db import connections, DEFAULT_DB_ALIAS
|
|
|
|
|
|
class TransactionManagementError(Exception):
|
|
"""
|
|
This exception is thrown when something bad happens with transaction
|
|
management.
|
|
"""
|
|
pass
|
|
|
|
################
|
|
# Private APIs #
|
|
################
|
|
|
|
def get_connection(using=None):
|
|
"""
|
|
Get a database connection by name, or the default database connection
|
|
if no name is provided.
|
|
"""
|
|
if using is None:
|
|
using = DEFAULT_DB_ALIAS
|
|
return connections[using]
|
|
|
|
def get_autocommit(using=None):
|
|
"""
|
|
Get the autocommit status of the connection.
|
|
"""
|
|
return get_connection(using).autocommit
|
|
|
|
def set_autocommit(using=None, autocommit=True):
|
|
"""
|
|
Set the autocommit status of the connection.
|
|
"""
|
|
return get_connection(using).set_autocommit(autocommit)
|
|
|
|
def abort(using=None):
|
|
"""
|
|
Roll back any ongoing transactions and clean the transaction management
|
|
state of the connection.
|
|
|
|
This method is to be used only in cases where using balanced
|
|
leave_transaction_management() calls isn't possible. For example after a
|
|
request has finished, the transaction state isn't known, yet the connection
|
|
must be cleaned up for the next request.
|
|
"""
|
|
get_connection(using).abort()
|
|
|
|
def enter_transaction_management(managed=True, using=None, forced=False):
|
|
"""
|
|
Enters transaction management for a running thread. It must be balanced with
|
|
the appropriate leave_transaction_management call, since the actual state is
|
|
managed as a stack.
|
|
|
|
The state and dirty flag are carried over from the surrounding block or
|
|
from the settings, if there is no surrounding block (dirty is always false
|
|
when no current block is running).
|
|
"""
|
|
get_connection(using).enter_transaction_management(managed, forced)
|
|
|
|
def leave_transaction_management(using=None):
|
|
"""
|
|
Leaves transaction management for a running thread. A dirty flag is carried
|
|
over to the surrounding block, as a commit will commit all changes, even
|
|
those from outside. (Commits are on connection level.)
|
|
"""
|
|
get_connection(using).leave_transaction_management()
|
|
|
|
def is_dirty(using=None):
|
|
"""
|
|
Returns True if the current transaction requires a commit for changes to
|
|
happen.
|
|
"""
|
|
return get_connection(using).is_dirty()
|
|
|
|
def set_dirty(using=None):
|
|
"""
|
|
Sets a dirty flag for the current thread and code streak. This can be used
|
|
to decide in a managed block of code to decide whether there are open
|
|
changes waiting for commit.
|
|
"""
|
|
get_connection(using).set_dirty()
|
|
|
|
def set_clean(using=None):
|
|
"""
|
|
Resets a dirty flag for the current thread and code streak. This can be used
|
|
to decide in a managed block of code to decide whether a commit or rollback
|
|
should happen.
|
|
"""
|
|
get_connection(using).set_clean()
|
|
|
|
def clean_savepoints(using=None):
|
|
"""
|
|
Resets the counter used to generate unique savepoint ids in this thread.
|
|
"""
|
|
get_connection(using).clean_savepoints()
|
|
|
|
def is_managed(using=None):
|
|
"""
|
|
Checks whether the transaction manager is in manual or in auto state.
|
|
"""
|
|
return get_connection(using).is_managed()
|
|
|
|
def managed(flag=True, using=None):
|
|
warnings.warn("'managed' no longer serves a purpose.",
|
|
PendingDeprecationWarning, stacklevel=2)
|
|
|
|
def commit_unless_managed(using=None):
|
|
"""
|
|
Commits changes if the system is not in managed transaction mode.
|
|
"""
|
|
get_connection(using).commit_unless_managed()
|
|
|
|
def rollback_unless_managed(using=None):
|
|
"""
|
|
Rolls back changes if the system is not in managed transaction mode.
|
|
"""
|
|
get_connection(using).rollback_unless_managed()
|
|
|
|
###############
|
|
# Public APIs #
|
|
###############
|
|
|
|
def commit(using=None):
|
|
"""
|
|
Does the commit itself and resets the dirty flag.
|
|
"""
|
|
get_connection(using).commit()
|
|
|
|
def rollback(using=None):
|
|
"""
|
|
This function does the rollback itself and resets the dirty flag.
|
|
"""
|
|
get_connection(using).rollback()
|
|
|
|
def savepoint(using=None):
|
|
"""
|
|
Creates a savepoint (if supported and required by the backend) inside the
|
|
current transaction. Returns an identifier for the savepoint that will be
|
|
used for the subsequent rollback or commit.
|
|
"""
|
|
return get_connection(using).savepoint()
|
|
|
|
def savepoint_rollback(sid, using=None):
|
|
"""
|
|
Rolls back the most recent savepoint (if one exists). Does nothing if
|
|
savepoints are not supported.
|
|
"""
|
|
get_connection(using).savepoint_rollback(sid)
|
|
|
|
def savepoint_commit(sid, using=None):
|
|
"""
|
|
Commits the most recent savepoint (if one exists). Does nothing if
|
|
savepoints are not supported.
|
|
"""
|
|
get_connection(using).savepoint_commit(sid)
|
|
|
|
##############
|
|
# DECORATORS #
|
|
##############
|
|
|
|
class Transaction(object):
|
|
"""
|
|
Acts as either a decorator, or a context manager. If it's a decorator it
|
|
takes a function and returns a wrapped function. If it's a contextmanager
|
|
it's used with the ``with`` statement. In either event entering/exiting
|
|
are called before and after, respectively, the function/block is executed.
|
|
|
|
autocommit, commit_on_success, and commit_manually contain the
|
|
implementations of entering and exiting.
|
|
"""
|
|
def __init__(self, entering, exiting, using):
|
|
self.entering = entering
|
|
self.exiting = exiting
|
|
self.using = using
|
|
|
|
def __enter__(self):
|
|
self.entering(self.using)
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback):
|
|
self.exiting(exc_value, self.using)
|
|
|
|
def __call__(self, func):
|
|
@wraps(func)
|
|
def inner(*args, **kwargs):
|
|
with self:
|
|
return func(*args, **kwargs)
|
|
return inner
|
|
|
|
def _transaction_func(entering, exiting, using):
|
|
"""
|
|
Takes 3 things, an entering function (what to do to start this block of
|
|
transaction management), an exiting function (what to do to end it, on both
|
|
success and failure, and using which can be: None, indiciating using is
|
|
DEFAULT_DB_ALIAS, a callable, indicating that using is DEFAULT_DB_ALIAS and
|
|
to return the function already wrapped.
|
|
|
|
Returns either a Transaction objects, which is both a decorator and a
|
|
context manager, or a wrapped function, if using is a callable.
|
|
"""
|
|
# Note that although the first argument is *called* `using`, it
|
|
# may actually be a function; @autocommit and @autocommit('foo')
|
|
# are both allowed forms.
|
|
if using is None:
|
|
using = DEFAULT_DB_ALIAS
|
|
if callable(using):
|
|
return Transaction(entering, exiting, DEFAULT_DB_ALIAS)(using)
|
|
return Transaction(entering, exiting, using)
|
|
|
|
|
|
def autocommit(using=None):
|
|
"""
|
|
Decorator that activates commit on save. This is Django's default behavior;
|
|
this decorator is useful if you globally activated transaction management in
|
|
your settings file and want the default behavior in some view functions.
|
|
"""
|
|
def entering(using):
|
|
enter_transaction_management(managed=False, using=using)
|
|
|
|
def exiting(exc_value, using):
|
|
leave_transaction_management(using=using)
|
|
|
|
return _transaction_func(entering, exiting, using)
|
|
|
|
def commit_on_success(using=None):
|
|
"""
|
|
This decorator activates commit on response. This way, if the view function
|
|
runs successfully, a commit is made; if the viewfunc produces an exception,
|
|
a rollback is made. This is one of the most common ways to do transaction
|
|
control in Web apps.
|
|
"""
|
|
def entering(using):
|
|
enter_transaction_management(using=using)
|
|
|
|
def exiting(exc_value, using):
|
|
try:
|
|
if exc_value is not None:
|
|
if is_dirty(using=using):
|
|
rollback(using=using)
|
|
else:
|
|
if is_dirty(using=using):
|
|
try:
|
|
commit(using=using)
|
|
except:
|
|
rollback(using=using)
|
|
raise
|
|
finally:
|
|
leave_transaction_management(using=using)
|
|
|
|
return _transaction_func(entering, exiting, using)
|
|
|
|
def commit_manually(using=None):
|
|
"""
|
|
Decorator that activates manual transaction control. It just disables
|
|
automatic transaction control and doesn't do any commit/rollback of its
|
|
own -- it's up to the user to call the commit and rollback functions
|
|
themselves.
|
|
"""
|
|
def entering(using):
|
|
enter_transaction_management(using=using)
|
|
|
|
def exiting(exc_value, using):
|
|
leave_transaction_management(using=using)
|
|
|
|
return _transaction_func(entering, exiting, using)
|