Creo que la respuesta de @ AnujGupta es correcta. Sin embargo, la reversión puede generar una excepción que debería atrapar y manejar:
from django.db import transaction, DatabaseError
try:
a.save()
except DatabaseError:
try:
transaction.rollback()
except transaction.TransactionManagementError:
# Log or handle otherwise
Si encuentra que está reescribiendo este código en varias save()
ubicaciones, puede extraer el método:
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
Finalmente, puede embellecerlo usando un decorador que protege los métodos que usan 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()
# ...
Incluso si implementa el decorador anterior, es conveniente mantenerlo try_rolling_back()
como un método extraído en caso de que necesite usarlo manualmente para casos en los que se requiera un manejo específico, y el manejo genérico del decorador no sea suficiente.