避免向 Django 模型添加重复条目的有效方法

Question

我已经阅读了很多关于如何在 Django 中使数据库/模型独一无二的文章，而且这些文章似乎都有效。但是，我没有看到任何帖子讨论避免向数据库添加重复条目的有效方法。

我的模型如下所示：

# from app.models
class TestModel(models.Model):
    text = models.TextField(unique=True, null=True)
    fixed_field = models.TextField()

我目前避免添加重复条目而不会出错的方法如下。

# from app.views
posts = ["one", "two", "three"]
fixed_field = "test"

for post in posts:
    try:
        TestModel(text=post, fixed_field = fixed_field).save()
    except IntegrityError:
        pass

如果我不这样做，我会得到一个 IntegrityError。有什么办法可以提高效率吗？

Answer 1

如果您要批量添加项目，您可以首先尝试阻止添加这些项目，方法是获取需要唯一的文本，然后制作一个不引入重复的TestModels 列表：

used_text = set(TestModel.objects.values_list('text', flat=True))

posts = ['one', 'two', 'three']
fixed_field = "test"

test_models = []
for post in posts:
    if post not in used_text:
        used_text.add(post)
        test_models.add(TestModel(text=post, fixed_field = fixed_field))

test_models.objects.bulk_create(test_models)

.bulk_create(..) [Django-doc] 然后批量创建所有记录，通常在一个查询中。如果多个查询元素数量巨大，但每次查询都会插入大量记录。

但由于竞争条件，上述操作仍然会失败，因为在获取数据库中的texts 和添加新的之间，其他查询可以更新数据库的状态，所以尽管可能性不大，但您可能应该使用重试机制，再次从列表中过滤掉 TestModels，并尝试重新插入这些内容。