使用PyMongo批量更新：合并Python字典与MongoDB集合，能否用聚合管道实现？

问题背景 #

内存中的Python字典列表：

dict_list = [
  {
    'id': 1,
    'from_value_key': 'x'
  },
  {
    'id': 2,
    'from_value_key': 'y'
  },
  {
    'id': 4,
    'from_value_key': 'z'
  }
]

原MongoDB集合collection内容：

{ id: 1, /* other_key_and_values */ },
{ id: 2, /* other_key_and_values */ },
{ id: 3, /* other_key_and_values */ }

需要实现的更新效果：

• 匹配id的文档添加to_value_key字段，值对应from_value_key
• 未在字典列表中的文档（如id=3）不修改
• 仅存在于字典列表中的文档（如id=4）忽略，不新增到集合

解决方案：使用聚合管道+$merge实现 #

完全可以用$documents、$lookup、$unwind、$merge的组合实现批量更新，无需bulk_write。

具体聚合管道代码（MongoDB Shell 4.4+） #

db.collection.aggregate([
  // 1. 引入内存中的字典列表作为临时数据集
  {
    $documents: [
      { id: 1, from_value_key: 'x' },
      { id: 2, from_value_key: 'y' },
      { id: 4, from_value_key: 'z' }
    ]
  },
  // 2. 关联原集合的文档，匹配id字段
  {
    $lookup: {
      from: 'collection',
      localField: 'id',
      foreignField: 'id',
      as: 'original_doc'
    }
  },
  // 3. 展开关联结果，保留未匹配到原集合的文档（后续会被过滤）
  {
    $unwind: {
      path: '$original_doc',
      preserveNullAndEmptyArrays: true
    }
  },
  // 4. 过滤掉仅存在于字典列表、原集合没有的文档（如id=4）
  {
    $match: {
      original_doc: { $ne: null }
    }
  },
  // 5. 构造更新后的文档：保留原字段，添加to_value_key
  {
    $replaceRoot: {
      newRoot: {
        $mergeObjects: [
          '$original_doc',
          { to_value_key: '$from_value_key' }
        ]
      }
    }
  },
  // 6. 将结果合并回原集合，按id匹配，仅更新差异字段
  {
    $merge: {
      into: 'collection',
      on: 'id',
      whenMatched: 'merge', // 仅新增/修改指定字段，不会覆盖原文档其他内容
      whenNotMatched: 'discard' // 忽略原集合不存在的文档
    }
  }
])

关键说明 #

• $documents阶段：MongoDB 4.4及以上版本支持，直接在聚合管道中引入内存数据，无需创建临时集合
• $merge的whenMatched: 'merge'：仅更新新增的to_value_key字段，原文档的其他字段完全保留，比replace模式更安全
• 自动忽略id=4的文档：$match阶段过滤掉了未关联到原集合的记录，whenNotMatched: 'discard'也确保不会向原集合新增文档
• id=3的文档：未进入聚合管道的处理流程，原集合中的该文档保持原样

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