使用pymongo更新MongoDB多层嵌套文档：为指定国家新增城市数据

你当前的写法确实会覆盖同大洲下所有其他国家数据，原因是你在更新时直接将countries字段设置为仅包含当前国家的数组，会直接替换掉原文档中整个countries数组的内容，法国等其他国家的数据会被直接清除。另外你原有生成cities数组的代码存在笔误，重复定义了两次city_id字段，漏掉了city_name，需要修正。

MongoDB支持通过**数组位置操作符$**定位到嵌套数组中匹配查询条件的元素，可以完全实现你需要的where(continent=Europe && country=England)层级定位效果，不会影响其他国家的数据。

场景1：直接替换英格兰的完整城市列表 #

如果你已经拿到了英格兰最新的全量城市列表（比如[London, Bath, Manchester]），想要直接覆盖旧的城市数组，可以用$set操作符配合位置定位：

from pymongo import MongoClient

client = MongoClient('你的MongoDB连接串')
db = client['你的库名']
# 集合名不要和变量名冲突，建议自定义集合名
geo_collection = db['你的集合名']

def update_country_cities(continent, country, new_cities):
    # new_cities格式示例：[{"city_name": "Manchester", "city_id": "203"}, ...]
    # 查询条件同时匹配大洲和对应国家
    query = {
        "continent_id": continent["id"],
        "countries.country_id": country["id"]
    }
    update = {
        "$set": {
            # $会自动定位到query中匹配到的那个国家元素
            "countries.$.cities": new_cities
        }
    }
    # 执行更新
    result = geo_collection.update_one(query, update)
    # 兼容新增大洲/国家的场景
    if result.matched_count == 0:
        # 对应大洲不存在，直接插入新大洲文档
        geo_collection.insert_one({
            "continent_name": continent["name"],
            "continent_id": continent["id"],
            "countries": [
                {
                    "country_name": country["name"],
                    "country_id": country["id"],
                    "cities": new_cities
                }
            ]
        })

场景2：仅追加新增城市、避免重复插入 #

如果你只需要新增曼彻斯特，不需要覆盖已有城市，也不需要每次传全量列表，可以用$addToSet+$each操作符，自动过滤已存在的城市：

def add_cities_to_country(continent, country, add_cities):
    # add_cities为要新增的城市列表，比如[{"city_name": "Manchester", "city_id": "203"}]
    query = {
        "continent_id": continent["id"],
        "countries.country_id": country["id"]
    }
    update = {
        "$addToSet": {
            "countries.$.cities": {
                "$each": add_cities
            }
        }
    }
    geo_collection.update_one(query, update)

# 定义参数
continent = {"id": "001", "name": "Europe"}
country = {"id": "012", "name": "England"}

# 场景1调用：传全量最新城市列表
new_full_cities = [
    {"city_name": "London", "city_id": "201"},
    {"city_name": "Bath", "city_id": "202"},
    {"city_name": "Manchester", "city_id": "203"}
]
update_country_cities(continent, country, new_full_cities)

# 场景2调用：仅传新增的城市
add_cities = [{"city_name": "Manchester", "city_id": "203"}]
add_cities_to_country(continent, country, add_cities)

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