Pymongo：INSERT_MANY+唯一索引 [英] Pymongo : insert_many + unique index

问题描述

我希望insert_many()个文档位于我的集合中。其中一些可能与集合内的现有文档具有相同的键/值对(在我的示例中为screen_name)。我在这个键上设置了唯一的索引集，因此我收到一个错误。

my_collection.create_index("screen_name", unique = True)

my_collection.insert_one({"screen_name":"user1", "foobar":"lalala"})
# no problem

to_insert = [
    {"screen_name":"user1", "foobar":"foo"}, 
    {"screen_name":"user2", "foobar":"bar"}
]
my_collection.insert_many(to_insert)

# error : 
# File "C:Program FilesPythonAnaconda3libsite-packagespymongoulk.py", line 331, in execute_command 
# raise BulkWriteError(full_result)
# 
# BulkWriteError: batch op errors occurred

我想：

1. 未收到错误
2. 不更改已存在的文档(此处{"screen_name":"user1", "foobar":"lalala"})
3. 插入所有不存在的文档(此处为{"screen_name":"user2", "foobar":"bar"})

---

编辑：正如有人在评论中所说，"这个问题是在询问如何执行大容量插入并忽略唯一索引错误，同时仍然插入成功的记录。因此，它与问题"How do I do Bulk Insert"不是重复的。"请重新打开它。

推荐答案

一种解决方案是使用insert_many的ordered参数，并将其设置为False(默认为True)：

my_collection.insert_many(to_insert, ordered=False)

From the PyMongo documentation:

已排序(可选)：如果将插入True(默认值)文档 按所提供的顺序在服务器上连续运行。如果出现错误，则所有 剩余的镶件将中止。如果False，则文档将插入到 服务器顺序任意(可能并行)和所有文档 将尝试插入。

但是，当无法插入所有文档时，您仍然必须处理异常。

根据您的用例，您可以决定pass、记录警告或检查异常。

这篇关于Pymongo：INSERT_MANY+唯一索引的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！