pymongo $在子文档数组上设置

时间:2016-03-31 00:32:19

标签: python mongodb pymongo

我有以下形式的pymongo集合:

{
    "_id" : "R_123456789",
    "supplier_ids" : [
        {
                "id" : "S_987654321",
                "file_version" : ISODate("2016-03-15T00:00:00Z"),
                "latest" : false
        },
        {
                "id" : "S_101010101",
                "file_version" : ISODate("2016-03-29T00:00:00Z"),
                "latest" : true
        }
    ]
}

当我收到新的供应商数据时,如果供应商ID已更改,我想通过将最新的'最新'设置为False并将$推送新记录来捕获该数据。

$ set无效,因为我正在尝试使用它(在'else'之后注释掉代码):

import pymongo
from dateutil.parser import parse

new_id = 'S_323232323'
new_date = parse('20160331')

with pymongo.MongoClient() as client:
    db = client.transactions
    collection_ids = db.ids

    try:
        collection_ids.insert_one({"_id": "R_123456789",
                                   "supplier_ids": ({"id": "S_987654321",
                                                     "file_version": parse('20160315'),
                                                     "latest": False},
                                                    {"id": "S_101010101",
                                                     "file_version": parse('20160329'),
                                                     "latest": True})})
    except pymongo.errors.DuplicateKeyError:
        print('record already exists')

    record = collection_ids.find_one({'_id':'R_123456789'})

    for supplier_id in record['supplier_ids']:
        print(supplier_id)
        if supplier_id['latest']:
            print(supplier_id['id'], 'is the latest')

            if supplier_id['id'] == new_id:
                print(new_id, ' is already the latest version')
            else:
                # print('setting', supplier_id['id'], 'latest flag to False')
                # <<< THIS FAILS >>>
                # collection_ids.update_one({'_id':record['_id']},
                #                           {'$set':{'supplier_ids.latest':False}})
                print('appending', new_id)
                data_to_append = {"id" : new_id,
                                  "file_version": new_date,
                                  "latest": True}
                collection_ids.update_one({'_id':record['_id']},
                                          {'$push':{'supplier_ids':data_to_append}})

非常感谢任何和所有帮助。

这整个过程看起来很不自然 - 我应该采用更简化的方法吗?

谢谢!

1 个答案:

答案 0 :(得分:2)

您可以尝试使用位置运算符。

collection_ids.update_one(
    {'_id':record['_id'], "supplier_ids.latest": true},
    {'$set':{'supplier_ids.$.latest': false}}
)

此查询将更新supplier_ids.latest = false,如果它在文档中是真的并且符合其他条件。

要抓住的是你必须包括字段数组作为条件的一部分。

有关详细信息,请参阅Update