Updating multiple documents with specific filters using pymongo

Question

I would like to update multiple documents in a mongoDB database using pymongo. I have this data:

data_to_be_updated = [
    {"sourceID" : 6, "source" : "test", "name" : "simon"},
    {"sourceID" : 8, "source" : "test", "name" : "greg"},
    {"sourceID" : 9, "source" : "test", "name" : "julie"},
    {"sourceID" : 10, "source" : "test", "name" : "john"}
    ]
sourceIDs = [6, 8, 9, 10]

I would like to update each of the elements in data_to_be_inserted, filtering them by their sourceID. I have tried using the update_many function, but it updates all documents matching a single filter. I could of course use a for loop like this:

for item in data_to_be_updated:
    collection.update_one({"sourceID": item["sourceID"]}, item})

The above method uses to many calls. How do i achieve the same in a single call to the database?

A. Jesse Jiryu Davis · Accepted Answer · 2017-08-14 13:26:43Z

2

Use bulk_write. Something like this, depending on exactly what fields need to be updated:

from pymongo.operations import UpdateOne

data_to_be_updated = [
    {"sourceID": 6, "source": "test", "name": "simon"},
    {"sourceID": 8, "source": "test", "name": "greg"},
    {"sourceID": 9, "source": "test", "name": "julie"},
    {"sourceID": 10, "source": "test", "name": "john"}
]

result = collection.bulk_write([
    UpdateOne(filter={'sourceID': d['sourceID']},
              update={'$set': {'name': d['name'],
                               'source': d['source']}})
    for d in data_to_be_updated])

print(result.bulk_api_result)

answered Aug 14, 2017 at 13:26

A. Jesse Jiryu Davis

24.1k5 gold badges60 silver badges72 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Updating multiple documents with specific filters using pymongo

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related