django多线程环境事务

Question

问题：python,django多线程读取ecel并导入到数据库，主线程事务有效，子线程事务失效
代码截图

问题简化版：

    @action(methods=['post'], detail=False)
    @transaction.atomic
    def uploadTest(self, request):
        data = {'code': 0, 'msg': '', 'data': []}
        connection.cursor().execute("INSERT INTO bby_test (id, name,age) VALUES (%s, %s, %s)", (1, 'name', 1))
        for i in range(0, 5):
            # 1.多线程调用 X
             pool.submit(Test,i)
            # 2.方法调用 √
             Test(i)
            # 问题：如何操作使多线程达到普通方法调用的效果，使一次子线程失败，所有子线程及主线程保持一致,不去操作数据库
        return JsonResponse(data)

def Test(i):
    print(i)
    if i % 2 == 0:
        a = 1/0
    connection.cursor().execute("INSERT INTO bby_test (id, name,age) VALUES (%s, %s, %s)", (2, 'name2', 2))

Answer 1

基于Monster 组和GPT的调写：
在你的情况下，主线程中的事务是有效的，因为它是由Django框架自动创建和管理的。但是在子线程中，你需要手动创建事务，并手动提交或回滚，否则事务将不会生效。

你可以通过使用django.db.connections来在子线程中手动创建和管理数据库连接和事务。以下是一个示例代码：

from django.db import connections

def import_data():
    # 从excel中读取数据
    # ...

    # 获取数据库连接
    connection = connections['default']

    # 开始事务
    with connection.cursor() as cursor:
        cursor.execute("BEGIN")

    try:
        # 将数据导入到数据库中
        # ...

        # 提交事务
        with connection.cursor() as cursor:
            cursor.execute("COMMIT")
    except:
        # 回滚事务
        with connection.cursor() as cursor:
            cursor.execute("ROLLBACK")

在子线程中，你需要使用connections来获取数据库连接，并使用cursor来手动执行SQL语句。在执行SQL语句之前，你需要使用BEGIN命令来开启一个事务，在执行完SQL语句后，使用COMMIT命令提交事务，如果出现异常，则需要使用ROLLBACK命令回滚事务