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

Question

在我的 django 应用程序中，我有一个处理用户上传的 XLS 文件的 celery 任务。 它被用来做诸如“大量导入”数据之类的事情。

在 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

正如交易文档所述，您可以明确控制它们。 在您的情况下，您可以使用上下文管理器包装它：

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