Django Model Transactions with Context Managers: 使用上下文管理器处理模型事务

from django.db import transaction

class TransactionManager():
    """
    使用上下文管理器处理模型事务
    使用方式:
    with TransactionManager():
        # 在此进行数据库操作，如果有异常则会回滚
    """
    def __enter__(self):
        transaction.set_autocommit(False)  # 设置为手动提交事务
        return self

    def __exit__(self, exc_type, exc_value, traceback):
        try:
            if exc_type is None:
                transaction.commit()  # 提交事务
            else:
                transaction.rollback()  # 回滚事务
        finally:
            transaction.set_autocommit(True)  # 恢复为自动提交事务

    def __call__(self, func):
        """
        可以将该类实例作为装饰器使用
        @TransactionManager()
        def my_func():
            pass
        """
        def inner_func(*args, **kwargs):
            with self:
                return func(*args, **kwargs)
        return inner_func

# 示例代码
from myapp.models import MyModel

@TransactionManager()
def insert_data():
    # 在事务中插入数据
    m = MyModel.objects.create(name="pidancode.com", age=18)
    print(m.id)

insert_data()