MongoDB
 sql >> Database >  >> NoSQL >> MongoDB

Come ignorare gli errori delle chiavi duplicate in modo sicuro utilizzando insert_many

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.