如何优雅实现嵌套SQLAlchemy对象到Pydantic模型的映射？

针对SQLAlchemy关联对象无法自动映射到Pydantic模型列表的问题，有几种优雅的处理方式，以下是最常用的两种：

1. 全局模型验证器（Model Validator） #

在LifeSchema中添加前置验证器，统一处理phone_numbers字段的转换，适合需要批量处理多个字段的场景：

示例代码 #

首先定义SQLAlchemy模型：

from sqlalchemy import Column, Integer, String, ForeignKey
from sqlalchemy.orm import relationship, declarative_base

Base = declarative_base()

class PhoneNumber(Base):
    __tablename__ = "phone_numbers"
    id = Column(Integer, primary_key=True)
    number = Column(String(20))
    life_id = Column(Integer, ForeignKey("lives.id"))
    life = relationship("Life", back_populates="phone_numbers")

class Life(Base):
    __tablename__ = "lives"
    id = Column(Integer, primary_key=True)
    name = Column(String(50))
    phone_numbers = relationship("PhoneNumber", back_populates="life")

然后定义Pydantic模型及验证器：

from pydantic import BaseModel, model_validator

class PhoneNumberResource(BaseModel):
    id: int
    number: str

    class Config:
        from_attributes = True  # 支持从ORM对象直接验证

class LifeSchema(BaseModel):
    id: int
    name: str
    phone_numbers: list[PhoneNumberResource]

    @model_validator(mode="before")
    def convert_associated_objects(cls, values):
        # 检查并转换phone_numbers字段
        if "phone_numbers" in values and values["phone_numbers"]:
            values["phone_numbers"] = [
                PhoneNumberResource.model_validate(phone) 
                for phone in values["phone_numbers"]
            ]
        return values

    class Config:
        from_attributes = True

说明 #

• mode="before"表示该验证器会在Pydantic执行字段类型检查前运行，提前把SQLAlchemy的PhoneNumber实例列表转换为PhoneNumberResource列表。
• 保持了SQLAlchemy与Pydantic模型的解耦，所有转换逻辑集中在Pydantic层。

2. 字段级前置验证器（Field-level Before Validator） #

如果仅需处理phone_numbers单个字段，可使用Annotated结合字段级验证器，逻辑更精准：

示例代码 #

from pydantic import BaseModel, BeforeValidator
from typing import Annotated, List

# 定义单独的转换函数
def convert_phone_numbers(phone_numbers):
    return [PhoneNumberResource.model_validate(p) for p in phone_numbers]

class LifeSchema(BaseModel):
    id: int
    name: str
    # 给phone_numbers字段添加前置转换逻辑
    phone_numbers: Annotated[List[PhoneNumberResource], BeforeValidator(convert_phone_numbers)]

    class Config:
        from_attributes = True

class PhoneNumberResource(BaseModel):
    id: int
    number: str

    class Config:
        from_attributes = True

说明 #

• 转换逻辑完全绑定到目标字段，代码更清晰，适合单一字段的转换需求。
• 同样遵循关注点分离原则，无需修改SQLAlchemy模型代码。

不推荐的耦合方案 #

虽然可以在SQLAlchemy模型中添加to_pydantic()类的方法直接生成Pydantic实例，但这种方式会让ORM模型依赖Pydantic定义，增加耦合度，除非有特殊需求，否则不建议使用。

内容的提问来源于stack exchange，提问作者p.magalhaes