可以通过try-except语句来捕获BulkCreateError或BulkUpdateError异常,并使用异常对象的属性来识别导致错误的对象及其索引。
下面是一个示例代码:
from django.db import IntegrityError
from django.core.exceptions import ObjectDoesNotExist
from django.db.models import F
try:
objs = [MyModel(name='foo'), MyModel(name='bar'), MyModel(name='baz')]
result = MyModel.objects.bulk_create(objs)
except IntegrityError as e:
# e.message的格式为:
# "duplicate key value violates unique constraint "app_mymodel_name_key"\nDETAIL: Key (name)=(...) already exists."
errors = {}
for i, obj in enumerate(objs):
if str(obj) in e.message:
# 将最后的") already exists."剔除,仅保留名称部分
name = e.message.split(')')[0].split('(')[1].strip("'")
errors[name] = i
print("以下对象无法保存:")
for name, index in errors.items():
print(f"{index}: {name}")
在上面的例子中,我们试图使用bulk_create请求创建三个MyModel对象。假设MyModel有一个唯一的名称字段,其中foo和bar已经存在于数据库中。因此,bulk_create只能成功创建一个baz对象,并触发IntegrityError异常。我们捕获了异常并使用异常消息字符串来检查哪些对象导致了错误。我们可以使用对象的字符串表示形式来识别哪些对象是错误的,并使用其索引来报告错误。本例中,输出将是:
以下对象无法保存:
0: foo
1: bar
如果我们希望通过bulk_update更新多个对象但不确定哪些对象的更新将导致主键冲突,则可以使用相同的技巧来捕获BulkUpdateError异常。在这种情况下,我们可以使用F()表达式来更新字段值,