You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何定义允许固定词与指定格式的pydantic.BaseModel?

实现Pydantic中tags字段的自定义验证需求

可以通过Pydantic的field_validator来实现对tags列表中每个元素的自定义校验,以下是完整实现代码:

from pydantic import BaseModel, field_validator, ValidationError
import re

class Product(BaseModel):
    name: str
    tags: list[str]

    @field_validator('tags', each_item=True)
    def validate_tag(cls, v):
        # 检查固定合法值
        if v in {"mean", "std"}:
            return v
        # 检查extra_开头的格式
        if not v.startswith("extra_"):
            raise ValueError(f"无效标签: {v},仅支持'mean'、'std'或'extra_<一位小数浮点数>'格式")
        # 提取数字部分并验证格式
        num_part = v.split("_")[-1]
        # 用正则匹配一位小数的浮点数(包括整数形式如10转成的10.0)
        pattern = r'^\d+(\.\d)?$'
        if not re.match(pattern, num_part):
            raise ValueError(f"无效的extra标签格式: {v},小数部分仅允许保留1位")
        # 额外验证转成浮点数后确实只有一位小数(避免类似1.00这种字符串)
        try:
            num = float(num_part)
            if not (num * 10).is_integer():
                raise ValueError(f"无效的extra标签数值: {v},必须是保留1位小数的浮点数")
        except ValueError:
            raise ValueError(f"无效的extra标签数值: {v},无法解析为浮点数")
        return v

# 测试合法用例
valid_product = Product(name="test", tags=["mean", "extra_2.5", "std", "extra_10.0"])
print("合法实例验证通过:", valid_product)

# 测试非法用例
try:
    invalid_product = Product(name="test", tags=["mean", "extra_3.14", "invalid_tag"])
except ValidationError as e:
    print("非法实例验证报错:")
    print(e)

代码说明

  • 使用field_validator并设置each_item=True,会对tags列表中的每一个元素单独进行校验
  • 先判断元素是否为固定合法值mean或std,符合则直接通过
  • 对extra_开头的标签,先用正则匹配格式(确保是整数或一位小数的数字字符串),再转成浮点数验证确实只有一位小数(避免1.00这种不符合要求的字符串)
  • 不符合规则的标签会抛出明确的ValueError,最终会被Pydantic封装为ValidationError返回

内容的提问来源于stack exchange,提问作者aura

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.01 12:28:18