for 循环中 Django 应用程序中的原子事务

Question

在我的 Django 应用程序中，我有一个处理用户上传的 XLS 文件的 celery 任务。它被用来做类似 "mass import" 数据的事情。

在 for 循环中，我正在处理每一行——从每一行我想创建一个模型实例对象（所以我将有多个 Model.objects.create(...) 调用——对于一行我可能将创建多个对象及其外键）。

在 django 中是否可以为整个 for 循环创建一个原子事务？例如，如果 XLS 包含 100 行 - 其中 90 行成功但 91 行不成功（因为行中的数据错误） - 是否可以回滚之前的 90（或更多）数据库保存？

我的代码如下：

workbook = xlrd.open_workbook(importobject.file.path)
worksheet = workbook.sheet_by_index(0)

processed_rows = []
omitted_rows = []

print('Mass import rows: ', worksheet.nrows)

startrange = 0

if str(worksheet.cell(0, 0).value) == 'header':
    print('Omit first row as header row')
    startrange = 1

for rowno in range(startrange, worksheet.nrows):
    logger.info('Processing row {}:'.format(rowno))

    for colno in range(worksheet.ncols):
        cell = worksheet.cell(rowno, colno)
        print('Row {}, Col {}:'.format(rowno, colno), str(cell.value))
        # process call and row
    # validate and save object here - if not valid - rollback whole mass import
    #MyModel.objects.create()

Answer 1

如交易文档所述，您可以 control them explicitly。在你的情况下，你可以用上下文管理器包装它：

from django.db import transaction

with transaction.atomic():
    for rowno in range(startrange, worksheet.nrows):
        logger.info('Processing row {}:'.format(rowno))

        for colno in range(worksheet.ncols):
            cell = worksheet.cell(rowno, colno)
            print('Row {}, Col {}:'.format(rowno, colno), str(cell.value))
        if not ... valid:
            raise IntegrityError()  # will cause the entire transaction to be rolled back

for 循环中 Django 应用程序中的原子事务

Atomic transactions in Django app in a for loop

python

django

postgresql

django-orm