如何使用PyMongo仅更新文档中的部分嵌套字段？

假设我们有如下MongoDB文档：

{ '_id': 1, 'Name': 'Document1', 'Data': { 'Color': 'Green', 'Size': 5 } }

现在想用PyMongo通过一个字典来更新该文档，比如更新用的字典是：

dict1 = { 'Name': 'Document1', 'Data': { 'Size': 5 } }

如果明确知道dict1['Data']里的具体字段，我们可以直接指定嵌套路径来更新：

collection.update_one( { 'Name': dict1['Name'] }, { '$set': { 'Data.Size': dict1['Data']['Size'] } })

但如果是通用场景，不清楚dict1['Data']里包含哪些字段，直接用下面的写法会覆盖整个Data对象，导致原文档中存在但dict1没有的字段（比如Color）被删除：

collection.update_one( { 'Name': dict1['Name'] }, { '$set': { 'Data': dict1['Data'] } })

解决方案：动态构建嵌套字段的$set参数 #

我们可以写一个工具函数，把嵌套字典的结构转换成MongoDB支持的点分隔路径，这样就能让$set仅更新指定的嵌套字段，同时保留其他原有字段：

def build_nested_set(data_dict, parent_key=""):
    set_dict = {}
    for key, value in data_dict.items():
        # 如果值还是字典，递归处理嵌套结构
        if isinstance(value, dict):
            nested_items = build_nested_set(value, f"{parent_key}{key}.")
            set_dict.update(nested_items)
        else:
            # 拼接成MongoDB的点路径格式
            set_dict[f"{parent_key}{key}"] = value
    return set_dict

# 针对dict1的Data字段构建更新参数
update_fields = build_nested_set(dict1['Data'], "Data.")

# 执行更新操作
collection.update_one(
    {'Name': dict1['Name']},
    {'$set': update_fields}
)

效果验证 #

执行上述代码后，原文档的Data.Color会被保留为Green，而Data.Size会被更新为dict1中的值（如果dict1里的Size有变化的话），不会删除任何原有字段。

内容的提问来源于stack exchange，提问作者espogian