django - 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会在退出最内部的原子块时强制进行回滚。这对于触发回滚而不引发异常很有用。