在django bulk_create中获取IntegrityError的键

时间:2018-06-15 04:09:19

标签: python django bulkinsert

我正在尝试在django中执行Model.objects.bulk_create(bulk_objects),我有时会期待IntegrityError,但我想获取导致异常的对象详细信息。

try:
        # bulk_objects contains ~70k items
        Auction.objects.bulk_create(bulk_objects)
except IntegrityError as e:
        print(str(e.__cause__))
        msg = str(e.__cause__)
        match = re.search('(?!\()\d+(?=\))', msg)
        print(match.group(0)) # prints 123865 in this case

这会引发异常

insert or update on table "auction_table" violates foreign key constraint"my_constraint"
DETAIL:  Key (item_id)=(123865) is not present in table "item_table".

这很好,我希望这个,我想得到导致异常的item_id,在这种情况下是item_id为123865。

有没有一种方法可以在不对字符串执行某些正则表达式或迭代bulk_objects的情况下执行此操作?

只是希望直接从错误中获取它,否则我将继续使用正则表达式方法

1 个答案:

答案 0 :(得分:0)

当数据库的关系完整性受到影响时引发异常,例如:外键检查失败,重复密钥等... 因此,请检查是否存在任何重复条目,或者数据库的任何外键约束是否失败。

    def bulk_create(self, objs, batch_size=None):
    """
    Inserts each of the instances into the database. This does *not* call
    save() on each of the instances, does not send any pre/post save
    signals, and does not set the primary key attribute if it is an
    autoincrement field (except if features.can_return_ids_from_bulk_insert=True).
    Multi-table models are not supported.
    """

根据文档bulk_create()不支持多表模型。因此,您必须显式创建批处理,然后将该批处理发送到bulk_create()。在创建该批处理时,您可以验证这些内容。 您可以查看如何创建批处理here