如何在SQLAlchemy+Pydantic中获取关联模型的owner_name？

假设你的User SQLAlchemy模型包含name字段（示例如下）：

class User(Base):
    id = Column(Integer, primary_key=True, index=True)
    name = Column(String, index=True)  # 用户名字段
    items = relationship("Item", back_populates="owner")

以下是几种实现方式：

方法一：直接在Pydantic模型中声明owner_name字段 #

直接修改ItemInDBBase模型，新增owner_name字段：

class ItemInDBBase(BaseModel):
    id: int
    title: str
    description: str
    owner_id: int
    owner_name: str  # 新增的所有者名字段

    class Config:
        from_attributes = True

当从SQLAlchemy的Item实例转换为该Pydantic模型时，会自动从item.owner.name取值。注意：必须确保owner关联对象已被加载，否则会触发额外的SQL查询（N+1问题）。

方法二：使用字段验证器处理空值场景 #

如果需要处理owner可能为None的情况（比如外键允许为空），可以用Pydantic的field_validator：

from pydantic import field_validator

class ItemInDBBase(BaseModel):
    id: int
    title: str
    description: str
    owner_id: int | None = None
    owner_name: str | None = None

    @field_validator('owner_name', mode='before')
    def extract_owner_name(cls, _, values):
        # 从ORM实例的owner属性中获取名字
        owner = values.data.get('owner')
        return owner.name if owner else None

    class Config:
        from_attributes = True

方法三：通过@property动态生成字段 #

如果不想在模型中显式声明owner_name，可以用属性方法动态获取：

# 先定义User的Pydantic基础模型
class UserInDBBase(BaseModel):
    id: int
    name: str

    class Config:
        from_attributes = True

class ItemInDBBase(BaseModel):
    id: int
    title: str
    description: str
    owner_id: int
    owner: UserInDBBase  # 关联用户的Pydantic模型

    @property
    def owner_name(self) -> str:
        return self.owner.name

    class Config:
        from_attributes = True

# 重建模型以识别关联的UserInDBBase
ItemInDBBase.model_rebuild()

优化查询避免N+1问题 #

为了避免每个Item都触发一次查询owner的SQL，查询时使用joinedload预加载关联对象：

from sqlalchemy.orm import joinedload

# 查询时预加载所有Item对应的Owner
items = db.query(Item).options(joinedload(Item.owner)).all()
# 转换为Pydantic模型
item_dtos = ItemInDBBase.model_validate(items, many=True)

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