如何在Pydantic中通用化Config类，适配不同Params子类

问题描述 #

我希望在Pydantic中实现代码通用化。现有两个不同文件的配置结构：两个Config类均包含相同的Context类（含name字符串字段），仅params字段对应的Params类不同（一个是含int类型delay字段的ParamsDelay，另一个是含float类型shoot字段的ParamsShoot）。我想将Context和Config抽至单个文件，通过传入不同的Params类（来自不同文件）来复用Config。我曾尝试定义ParamsBase基类并拆分初始化，但未成功，尝试的代码如下：

from pydantic import BaseModel

class Context(BaseModel):
    name: str

class ParamsBase(BaseModel):
    pass

class Config(BaseModel):
    context: Context
    params: ParamsBase

    def __init__(self, ParamsClass, **data):
        super().__init__(**data, params.validator=ParamsClass)

调用方式设想为：

class ParamsShoot(BaseModel):
    shoot: float

Config(ParamsShoot, **{"params": {"shoot": 30.0}, "context": {"name": "test"}})

请问是否有可行的实现方案？

解决方案 #

你可以通过**泛型（Generic）**来实现这个需求，Pydantic支持泛型模型，能完美适配这种需要动态指定字段类型的场景。

实现步骤 #

1. 定义Context类和ParamsBase基类（保持原有结构即可）
2. 让Config同时继承Generic和BaseModel，并指定泛型参数来对应不同的Params子类
3. 在Config中用泛型参数定义params字段的类型

完整代码示例 #

from pydantic import BaseModel
from typing import Generic, TypeVar

# 定义泛型类型变量，约束为ParamsBase的子类
ParamsT = TypeVar('ParamsT', bound='ParamsBase')

class Context(BaseModel):
    name: str

class ParamsBase(BaseModel):
    pass

# 泛型Config类，通过ParamsT动态绑定params的具体类型
class Config(BaseModel, Generic[ParamsT]):
    context: Context
    params: ParamsT

# 不同业务的Params子类（可放在其他独立文件中）
class ParamsDelay(ParamsBase):
    delay: int

class ParamsShoot(ParamsBase):
    shoot: float

# 实际调用示例
if __name__ == "__main__":
    # 使用ParamsShoot初始化Config
    shoot_config = Config[ParamsShoot](
        context={"name": "shoot_config"},
        params={"shoot": 30.0}
    )
    print(shoot_config)
    # 输出: context=Context(name='shoot_config') params=ParamsShoot(shoot=30.0)

    # 使用ParamsDelay初始化Config
    delay_config = Config[ParamsDelay](
        context={"name": "delay_config"},
        params={"delay": 5}
    )
    print(delay_config)
    # 输出: context=Context(name='delay_config') params=ParamsDelay(delay=5)

方案优势 #

• 完全保留Pydantic的类型校验能力，传入不符合Params类结构的数据会直接触发验证错误
• 泛型语法清晰，IDE能提供准确的代码补全和类型提示
• 无需修改Config核心逻辑，只需通过泛型参数切换Params类型即可实现复用

替代方案：动态创建模型 #

如果不想使用泛型，也可以通过create_model动态生成对应Params类型的Config类：

from pydantic import BaseModel, create_model

class Context(BaseModel):
    name: str

class ParamsBase(BaseModel):
    pass

class ParamsDelay(ParamsBase):
    delay: int

class ParamsShoot(ParamsBase):
    shoot: float

def create_config_class(params_cls: type[ParamsBase]) -> type[BaseModel]:
    return create_model(
        f"ConfigFor{params_cls.__name__}",
        context=(Context, ...),
        params=(params_cls, ...),
        __base__=BaseModel
    )

# 使用示例
ShootConfig = create_config_class(ParamsShoot)
shoot_config = ShootConfig(context={"name": "test"}, params={"shoot": 30.0})
print(shoot_config)

不过这种方式的类型提示友好度不如泛型方案，适合对类型要求不高的场景。

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