如何为Pydantic字段设置运行时动态生成的有效输入列表？

当需要为Pydantic字段设置运行时才能确定的有效值列表（比如已登录用户、动态日期范围），且允许选择1个、多个或全部有效值时，直接用Literal或List[VALID]无法满足需求，原因如下：

• Literal要求参数是编译时确定的字面量/枚举值，运行时生成的VALID列表不符合要求
• List[VALID]是错误的类型提示逻辑，它实际会被解析为“元素是VALID这个列表类型”，而非“元素属于VALID的值集合”，且无法限制元素必须在有效值内

解决方案1：使用字段验证器（Pydantic v2） #

通过@field_validator自定义验证逻辑，检查列表中每个元素是否属于运行时生成的有效值集合，还可添加非空校验：

from pydantic import BaseModel, field_validator, ValidationError

# 模拟运行时获取有效值（比如从数据库/缓存读取）
def get_valid_inputs():
    return ["user_alice", "user_bob", "user_charlie"]

VALID_OPTIONS = get_valid_inputs()

class UserSelection(BaseModel):
    selected_users: list[str]

    @field_validator("selected_users")
    def validate_all_in_valid_options(cls, value):
        # 筛选出不在有效值内的元素
        invalid_values = [item for item in value if item not in VALID_OPTIONS]
        if invalid_values:
            raise ValueError(f"无效值：{invalid_values}，允许的选项为：{VALID_OPTIONS}")
        # 可选：要求至少选择一个元素
        if not value:
            raise ValueError("必须选择至少一个用户")
        return value

# 测试用例
try:
    # 合法：选择2个有效值
    valid_case = UserSelection(selected_users=["user_alice", "user_bob"])
    print(valid_case)
    
    # 非法：包含无效值
    invalid_case = UserSelection(selected_users=["user_alice", "user_dave"])
    print(invalid_case)
except ValidationError as e:
    print(e)

解决方案2：支持单个值/列表混合输入 #

如果需要同时支持传入单个值（如"user_alice"）或列表值，可在验证器中统一格式并验证：

from pydantic import BaseModel, field_validator, ValidationError

VALID_OPTIONS = get_valid_inputs()

class UserSelection(BaseModel):
    selected_users: str | list[str]

    @field_validator("selected_users")
    def normalize_and_validate(cls, value):
        # 将单个值转为列表，统一处理
        if isinstance(value, str):
            value = [value]
        # 执行有效值校验
        invalid_values = [item for item in value if item not in VALID_OPTIONS]
        if invalid_values:
            raise ValueError(f"无效值：{invalid_values}，允许的选项为：{VALID_OPTIONS}")
        if not value:
            raise ValueError("必须选择至少一个用户")
        return value

# 测试单个值输入
single_input_case = UserSelection(selected_users="user_charlie")
print(single_input_case)  # 输出：selected_users=['user_charlie']

解决方案3：使用Annotated + 前置验证器（更简洁） #

通过Annotated结合BeforeValidator，将验证逻辑与字段类型绑定，复用性更强：

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

def validate_allowed_values(input_value):
    # 统一转为列表格式
    if isinstance(input_value, str):
        input_value = [input_value]
    # 校验有效值
    invalid_values = [item for item in input_value if item not in VALID_OPTIONS]
    if invalid_values:
        raise ValueError(f"无效值：{invalid_values}，允许的选项为：{VALID_OPTIONS}")
    if not input_value:
        raise ValueError("必须选择至少一个用户")
    return input_value

VALID_OPTIONS = get_valid_inputs()

class UserSelection(BaseModel):
    selected_users: Annotated[List[str], BeforeValidator(validate_allowed_values)]

针对Pydantic v1的兼容方案 #

如果使用Pydantic v1，将@field_validator替换为@validator即可：

from pydantic import BaseModel, validator, ValidationError

VALID_OPTIONS = get_valid_inputs()

class UserSelection(BaseModel):
    selected_users: list[str]

    @validator("selected_users")
    def validate_all_in_valid_options(cls, value):
        invalid_values = [item for item in value if item not in VALID_OPTIONS]
        if invalid_values:
            raise ValueError(f"无效值：{invalid_values}，允许的选项为：{VALID_OPTIONS}")
        if not value:
            raise ValueError("必须选择至少一个用户")
        return value

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