Я вважаю, що відповідь @ 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()
як витягнутий метод, якщо вам потрібно використовувати його вручну для тих випадків, коли потрібна конкретна обробка, а загальної обробки декораторів недостатньо.