如何在Pydantic model_validate中选择性排除SQLAlchemy关联字段

针对你遇到的Pydantic v2中选择性排除ORM关联字段的需求，有几种实用的实现方式：

1. 直接使用model_validate的exclude参数 #

Pydantic v2的model_validate方法原生支持exclude参数，能在验证ORM对象时直接指定要排除的字段——既不会触发ORM的懒加载，也不会让该字段出现在最终的Pydantic模型中。

修改你的get_many方法：

@staticmethod
async def get_many(
    db: AsyncSession, skip: int = 0, limit: int = 20
) -> tuple[list[Role], str]:
    count = await db.scalar(select(func.count(RoleOrm.uid)))
    if not count:
        return [], "0"
    statement = (
        select(RoleOrm)
        .offset(skip)
        .limit(limit)
    )
    results = await db.execute(statement)
    # 调用时指定exclude排除role_permissions
    roles = [Role.model_validate(orm, exclude={"role_permissions"}) for orm in results.unique().scalars().all()]
    return roles, str(count)

get_by_uid方法保持原有逻辑即可，无需额外配置，自然会加载关联数据。

2. 创建差异化的Pydantic模型 #

如果不同场景下的字段差异较大，或者需要排除的字段不止一个，建议创建多个Pydantic模型适配不同需求：

# 基础模型，仅包含核心字段
class RoleBase(BaseModel):
    # 核心字段定义...

# 包含关联字段的模型，用于单个查询场景
class Role(RoleBase):
    role_permissions: List[RolePermission] | None

# 排除关联字段的模型，用于批量查询场景
class RoleWithoutPermissions(RoleBase):
    pass

然后对应修改仓库方法：

# get_by_uid返回包含关联字段的Role模型
@staticmethod
async def get_by_uid(db: AsyncSession, uid: str) -> Role:
    # ... 原有逻辑 ...
    return Role.model_validate(role_orm)

# get_many返回无关联字段的RoleWithoutPermissions模型
@staticmethod
async def get_many(
    db: AsyncSession, skip: int = 0, limit: int = 20
) -> tuple[list[RoleWithoutPermissions], str]:
    # ... 原有逻辑 ...
    roles = [RoleWithoutPermissions.model_validate(orm) for orm in results.unique().scalars().all()]
    return roles, str(count)

这种方式语义清晰，代码可读性高，适合长期维护。

3. 利用上下文(Context)动态控制字段加载 #

如果需要更灵活的动态切换，可以通过context参数结合模型验证器实现。你之前尝试context未达预期，可能是验证器写法问题，正确实现如下：

from pydantic import model_validator, ValidationInfo

class Role(BaseModel):
    # 字段定义...
    role_permissions: List[RolePermission] | None = None

    @model_validator(mode='before')
    def handle_role_permissions(cls, obj, info: ValidationInfo):
        # 从context中获取是否排除字段的标记
        exclude_perms = info.context.get('exclude_role_permissions', False)
        if exclude_perms and hasattr(obj, 'role_permissions'):
            # 临时移除ORM对象的该属性，避免触发懒加载
            delattr(obj, 'role_permissions')
        return obj

调用时传入context参数：

# get_many中传入标记，排除字段
roles = [Role.model_validate(orm, context={'exclude_role_permissions': True}) for orm in results.unique().scalars().all()]

# get_by_uid中不传入标记，默认保留字段
role = Role.model_validate(role_orm)

这种方式适合需要动态调整字段加载的场景，但需注意操作ORM对象属性的副作用。

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