基于Pydantic定义MongoDB嵌入式一对多模型的字段处理问题

核心思路是利用Pydantic的字段排除和模型初始化后钩子来满足需求：

• 给子模型定义运行时需要但序列化时隐藏的字段，通过Field(exclude=True)控制序列化行为
• 借助model_post_init钩子在模型实例化完成后自动注入依赖字段，避免验证阶段因字段缺失报错
• 父模型在初始化后，将自身的关联字段注入到所有嵌套子模型实例中

代码实现示例 #

首先导入依赖模块：

from pydantic import BaseModel, Field
from typing import List

定义子模型Address：

class Address(BaseModel):
    # 需序列化到MongoDB的业务字段
    street: str
    city: str
    state: str
    zip: str

    # 运行时存在但序列化时排除的字段，设置默认值避免验证报错
    id: str = Field(default=None, exclude=True)
    name: str = Field(default=None, exclude=True)

    # 可选：确保运行时字段必存在，避免注入遗漏
    def model_post_init(self, __context) -> None:
        assert self.id is not None and self.name is not None, "id和name字段必须完成注入"

定义父模型Person：

class Person(BaseModel):
    _id: str
    name: str
    addresses: List[Address]

    def model_post_init(self, __context) -> None:
        # 给每个嵌套的Address实例注入父模型的关联字段
        for addr in self.addresses:
            addr.id = self._id
            addr.name = self.name

使用演示 #

# 构造符合MongoDB格式的原始数据
person_data = {
    "_id": "joe",
    "name": "Joe Bookreader",
    "addresses": [
        {
            "street": "123 Fake Street",
            "city": "Faketon",
            "state": "MA",
            "zip": "12345"
        },
        {
            "street": "1 Some Other Street",
            "city": "Boston",
            "state": "MA",
            "zip": "12345"
        }
    ]
}

# 实例化模型
person = Person(**person_data)

# 运行时可正常访问子模型的注入字段
print(person.addresses[0].id)  # 输出: joe
print(person.addresses[0].name)  # 输出: Joe Bookreader

# 序列化为MongoDB兼容格式（自动排除id和name）
mongo_doc = person.model_dump()
print(mongo_doc)
# 输出结果与需求示例完全一致

补充说明 #

• 更新场景适配：无论是创建新实例还是更新现有数据，只要重新调用模型验证或实例化，model_post_init都会自动执行，确保新的子模型实例完成字段注入
• 复杂嵌套扩展：如果存在多层嵌套模型，可在对应父模型的model_post_init中逐层注入所需字段，无需编写复杂的前置/后置验证器逻辑

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