Я считаю, что ответ @ AnujGupta правильный. Однако откат сам может вызвать исключение, которое вы должны перехватить и обработать:
from django.db import transaction, DatabaseError
try:
a.save()
except DatabaseError:
try:
transaction.rollback()
except transaction.TransactionManagementError:
# Log or handle otherwise
Если вы обнаружите, что переписываете этот код в разных save()
местах, вы можете извлечь метод:
import traceback
def try_rolling_back():
try:
transaction.rollback()
log.warning('rolled back') # example handling
except transaction.TransactionManagementError:
log.exception(traceback.format_exc()) # example handling
Наконец, вы можете предварительно защитить его, используя декоратор, который защищает методы, которые используют save()
:
from functools import wraps
def try_rolling_back_on_exception(fn):
@wraps(fn)
def wrapped(*args, **kwargs):
try:
return fn(*args, **kwargs)
except:
traceback.print_exc()
try_rolling_back()
return wrapped
@try_rolling_back_on_exception
def some_saving_method():
# ...
model.save()
# ...
Даже если вы реализуете описанный выше декоратор, его все равно удобно использовать try_rolling_back()
в качестве извлеченного метода на тот случай, если вам понадобится использовать его вручную в тех случаях, когда требуется особая обработка, а общая обработка декоратора недостаточна.