如何用PyMongo实现MongoDB嵌套对象更新及数组新增字典？

需求说明 #

初始文档结构：

{
    _id: 'd2f5',
    url: 'https://exam.pl/b-4',
    creation_time: '2023.07.07 16:28:45'
}

• 首次添加check时，需创建嵌套的checks结构，同时初始化温度历史数组：

{
    _id: 'd2f5',
    url: 'https://exam.pl/b-4',
    creation_time: '2023.07.07 16:28:45',
    checks: {
        '10.08.2023-11.08.2023': {
            start_date: '10.08.2023',
            current_temp_forecast: '29.5',
            temp_forecast_history: [
                {
                    '2023.07.07 18:08:21': '29.5'
                }
            ]
        }
    }
}

• 已有check时，需将新温度值插入历史数组的开头，并更新当前温度：

{
    _id: 'd2f5',
    url: 'https://exam.pl/b-4',
    creation_time: '2023.07.07 16:28:45',
    checks: {
        '10.08.2023-11.08.2023': {
            start_date: '10.08.2023',
            current_temp_forecast: '28.2',
            temp_forecast_history: [
                {
                    '2023.07.08 19:39:52': '28.2'
                },
                {
                    '2023.07.07 18:08:21': '29.5'
                }
            ]
        }
    }
}

已尝试代码及报错 #

尝试的更新代码 #

update_query = {
    "$set": {
        f"checks.{check}.start_date": start_date,
        f"checks.{check}.current_temp_forecast": current_temp_forecast,
    },
    "$addToSet": { # 之前用过$push，同样报错
        f"checks.{check}.temp_forecast_history": {
            "$each": [
                {
                    timestamp: current_temp_forecast
                }
            ],
            "$position": 0
        }
    }
}

collection.update_one({"_id": document_id}, update_query)

报错信息（翻译后） #

pymongo.errors.WriteError: 更新文档中不允许出现以$为前缀的字段'$addToSet'。请考虑使用聚合管道配合$replaceWith。完整错误信息: {'index': 0, 'code': 52, 'errmsg': "更新文档的上下文环境中，不允许$addToSet里出现以$为前缀的'$addToSet'字段。请考虑使用聚合管道配合$replaceWith。"}

问题原因 #

1. $addToSet不支持$position修饰符：$position是$push操作的专属修饰符，$addToSet仅支持$each，无法指定元素插入位置。
2. 嵌套字段不存在时的更新冲突：首次更新时checks字段尚未创建，常规更新操作无法同时完成嵌套结构创建和数组前置插入，需用聚合管道处理复杂的条件更新逻辑。

解决方案 #

方案一：聚合管道一次性处理 #

使用聚合管道作为更新操作，可以同时处理字段存在/不存在的情况，并且正确实现数组前置插入：

# 假设变量已定义：document_id, check, start_date, current_temp_forecast, timestamp

update_pipeline = [
    {
        "$set": {
            "checks": {
                "$mergeObjects": [
                    {"$ifNull": ["$checks", {}]},
                    {
                        check: {
                            "$mergeObjects": [
                                {"$ifNull": [f"$checks.{check}", {}]},
                                {
                                    "start_date": {"$ifNull": [f"$checks.{check}.start_date", start_date]},
                                    "current_temp_forecast": current_temp_forecast,
                                    "temp_forecast_history": {
                                        "$concatArrays": [
                                            [{timestamp: current_temp_forecast}],
                                            {"$ifNull": [f"$checks.{check}.temp_forecast_history", []]}
                                        ]
                                    }
                                }
                            ]
                        }
                    }
                ]
            }
        }
    }
]

collection.update_one({"_id": document_id}, update_pipeline)

代码说明

• $mergeObjects：合并现有字段和新字段，避免覆盖原有结构。首次更新时$checks为null，则创建空对象后合并新的check结构；已有check时，合并原有check内容和新的温度数据。
• $ifNull：处理字段不存在的情况，比如start_date仅在首次创建时设置，后续更新保留原有值；温度历史数组不存在时初始化为空数组。
• $concatArrays：将新的温度记录（单个元素数组）和原有历史数组合并，实现新元素插入数组开头的效果。

方案二：分两次更新（更直观） #

如果不想用聚合管道，可分两步操作，先确保嵌套结构存在，再执行数组插入：

# 第一步：确保checks及子字段存在，设置基础值
collection.update_one(
    {"_id": document_id},
    {
        "$set": {
            f"checks.{check}.start_date": {"$ifNull": [f"$checks.{check}.start_date", start_date]},
            f"checks.{check}.current_temp_forecast": current_temp_forecast
        }
    },
    upsert=False
)

# 第二步：往历史数组开头插入新记录
collection.update_one(
    {"_id": document_id},
    {
        "$push": {
            f"checks.{check}.temp_forecast_history": {
                "$each": [{timestamp: current_temp_forecast}],
                "$position": 0
            }
        }
    }
)

代码说明

• 第一步用$ifNull保证start_date仅在首次创建时设置，同时更新当前温度。
• 第二步用$push配合$position:0实现新元素插入数组开头，此时字段已存在，不会触发之前的报错。

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