Django: Как откат (@ transaction.atomic) без повышения исключения?

Question

Я использую команду Django для выполнения некоторых задач, связанных с базой данных манипуляций:

class SomeCommand(BaseCommand): 
    @transaction.atomic 
    def handle(self, *args, **options): 
     # Some stuff on the database 

Если исключение во время выполнения моей программы, @transaction.atomic гарантирует откатить. Могу ли я заставить это поведение исключать исключение? Что-то вроде:

# Doing some stuff, changing objects 

if some_condition: 
    # ABANDON ALL CHANGES AND RETURN 

Answer 1

transaction.set_rollback может сделать это.

class SomeCommand(BaseCommand): 
    @transaction.atomic 
    def handle(self, *args, **options): 
     # Doing some stuff, changing objects 
     if some_condition: 
      # Return, rolling back transaction when atomic block exits 
      transaction.set_rollback(True) 
      return 

Цитируя the docs:

Установка флага отката True вынуждает откат при выходе из самого внутреннего атомного блока. Это может быть полезно для запуска отката без привлечения исключения.

Answer 2

Просто позвоните transaction.rollback().

Вызов transaction.rollback() возвращает всю транзакцию. Любые операции с незафиксированными базами данных будут потеряны.

Вы можете увидеть пример в docs.

Answer 3

Вы можете управлять выполнением кода, чтобы поднять или не исключение: