如何创建无需传入关键字参数的Pydantic类型？

当然可行，你可以通过重写Pydantic模型的初始化逻辑或者使用模型验证器，实现直接传入字符串位置参数实例化，同时保持DRY（仅维护一处合法数据类型列表），且无需使用Enum。

方法一：重写__init__方法（简单直接） #

这种方式直接在模型的构造方法中接收位置参数，传递给模型字段并触发验证：

from typing import Literal
from pydantic import BaseModel, ValidationError

# 仅需维护这一处合法数据类型
VALID_DATA_TYPES = {"type1", "type2", "type3", "type4"}
# 生成对应的Literal类型，用于类型提示
DataTypeLiteral = Literal[tuple(VALID_DATA_TYPES)]

class DataType(BaseModel):
    value: DataTypeLiteral

    def __init__(self, value: DataTypeLiteral, **kwargs):
        # 将位置参数传递给模型的value字段
        super().__init__(value=value, **kwargs)

# 验证通过
dt_valid = DataType("type1")
print(dt_valid.value)  # 输出: type1

# 验证失败，抛出ValidationError
try:
    dt_invalid = DataType("error")
except ValidationError as e:
    print(e)

方法二：使用model_validator（更灵活） #

通过Pydantic的模型验证器，自动将单个字符串输入转换为模型所需的字段格式，适合需要兼容多种输入形式的场景：

from typing import Literal
from pydantic import BaseModel, model_validator, ValidationError

VALID_DATA_TYPES = {"type1", "type2", "type3", "type4"}
DataTypeLiteral = Literal[tuple(VALID_DATA_TYPES)]

class DataType(BaseModel):
    value: DataTypeLiteral

    @model_validator(mode='before')
    def handle_single_input(cls, values):
        # 如果输入是单个字符串，转换为模型字段字典
        if isinstance(values, str):
            return {"value": values}
        return values

    # 可选：让实例直接表现为字符串的形式
    def __str__(self):
        return self.value

    def __repr__(self):
        return f"DataType({repr(self.value)})"

# 验证通过
dt_valid = DataType("type1")
print(dt_valid)  # 输出: type1

# 验证失败
try:
    dt_invalid = DataType("error")
except ValidationError as e:
    print(e)

优势说明 #

• DRY原则：合法数据类型仅在VALID_DATA_TYPES中定义一次，Literal类型和验证逻辑都基于此，新增类型只需修改这一处。
• 直接实例化：支持DataType("type1")的简洁写法，无需传入关键字参数。
• 完整验证：保留Pydantic的ValidationError抛出机制，非法输入会被及时拦截。

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