Puoi farcela controllando gli errori prodotti con BulkWriteError
. Questo è in realtà un "oggetto" che ha diverse proprietà. Le parti interessanti sono in details
:
import pymongo
from bson.json_util import dumps
from pymongo import MongoClient
client = MongoClient()
db = client.test
collection = db.duptest
docs = [{ '_id': 1 }, { '_id': 1 },{ '_id': 2 }]
try:
result = collection.insert_many(docs,ordered=False)
except pymongo.errors.BulkWriteError as e:
print e.details['writeErrors']
Alla prima esecuzione, questo fornirà l'elenco degli errori in e.details['writeErrors']
:
[
{
'index': 1,
'code': 11000,
'errmsg': u'E11000 duplicate key error collection: test.duptest index: _id_ dup key: { : 1 }',
'op': {'_id': 1}
}
]
In una seconda esecuzione, vengono visualizzati tre errori perché esistevano tutti gli elementi:
[
{
"index": 0,
"code": 11000,
"errmsg": "E11000 duplicate key error collection: test.duptest index: _id_ dup key: { : 1 }",
"op": {"_id": 1}
},
{
"index": 1,
"code": 11000,
"errmsg": "E11000 duplicate key error collection: test.duptest index: _id_ dup key: { : 1 }",
"op": {"_id": 1}
},
{
"index": 2,
"code": 11000,
"errmsg": "E11000 duplicate key error collection: test.duptest index: _id_ dup key: { : 2 }",
"op": {"_id": 2}
}
]
Quindi tutto ciò che devi fare è filtrare l'array per le voci con "code": 11000
e poi solo "panico" quando c'è qualcos'altro dentro
panic = filter(lambda x: x['code'] != 11000, e.details['writeErrors'])
if len(panic) > 0:
print "really panic"
Questo ti dà un meccanismo per ignorare gli errori di chiave duplicata ma ovviamente prestare attenzione a qualcosa che in realtà è un problema.