Creio que a resposta de @ AnujGupta está correta. No entanto, a reversão pode gerar uma exceção que você deve capturar e manipular:
from django.db import transaction, DatabaseError
try:
a.save()
except DatabaseError:
try:
transaction.rollback()
except transaction.TransactionManagementError:
# Log or handle otherwise
Se você estiver reescrevendo esse código em vários save()
locais, poderá extrair o 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
Por fim, você pode pré-modificá-lo usando um decorador que protege métodos que usam 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()
# ...
Mesmo se você implementar o decorador acima, ainda é conveniente mantê-lo try_rolling_back()
como um método extraído, caso você precise usá-lo manualmente nos casos em que é necessário um manuseio específico, e o manuseio do decorador genérico não é suficiente.