Pydantic多类型阈值模型中target约束未按类型生效问题咨询

你遇到的问题核心是Pydantic默认处理Union类型时，会按模型定义顺序尝试匹配，当target为整数时，优先匹配了ThresholdUpperLimit（因为其target接受整数类型），完全跳过了ThresholdRange的校验逻辑——本质是没通过**鉴别器（discriminator）**指定以type字段作为模型匹配的核心依据。

使用Pydantic的discriminator特性，明确根据type字段的值选择对应的校验模型，确保类型与校验规则严格绑定。

完整代码实现 #

from pydantic import BaseModel, Field, field_validator
from pydantic.types import PositiveInt
from typing import List, Union

# upper_limit类型阈值模型：type固定为"upper_limit"，target为正整数
class ThresholdUpperLimit(BaseModel):
    type: str = "upper_limit"
    target: PositiveInt  # 可根据需求调整为普通int，接受任意整数

# range类型阈值模型：type固定为"range"，target为0-10之间的浮点数
class ThresholdRange(BaseModel):
    type: str = "range"
    target: float = Field(..., gt=0, lt=10)  # 0 < target < 10的范围约束

    @field_validator('target')
    def int_to_float(cls, value):
        # 可选逻辑：允许输入整数自动转为浮点数，若需严格禁止整数可删除此验证器
        if isinstance(value, int):
            return float(value)
        return value

# 根模型：通过discriminator指定以type字段区分不同阈值模型
class RootModel(BaseModel):
    threshold: List[Union[ThresholdUpperLimit, ThresholdRange]] = Field(..., discriminator='type')

验证效果 #

测试1：range类型+超出范围的整数target #

test_data = {"threshold": [{"type": "range", "target": 12}]}
try:
    RootModel(**test_data)
except Exception as e:
    print(e)

输出（触发range的范围约束）：

1 validation error for RootModel -> threshold -> 0 -> target
Input should be less than 10 [type=less_than, input_value=12.0, input_type=float]

测试2：range类型+合法整数target（自动转浮点数） #

test_data = {"threshold": [{"type": "range", "target": 5}]}
model = RootModel(**test_data)
print(model.threshold[0].target)  # 输出：5.0

测试3：upper_limit类型+合法整数target #

test_data = {"threshold": [{"type": "upper_limit", "target": 12}]}
model = RootModel(**test_data)
print(model.threshold[0].target)  # 输出：12

关键说明 #

• discriminator='type'是核心：告诉Pydantic先检查每个阈值对象的type字段值，再匹配对应模型，彻底避免Union的顺序匹配漏洞
• 若需要严格禁止range类型的target输入整数，直接删除ThresholdRange中的int_to_float验证器即可，此时输入整数会触发类型错误
• 可根据实际需求调整target的约束规则（比如ThresholdUpperLimit的target改为普通int，或ThresholdRange的范围调整）

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