Pydantic与ABC结合子类验证：无需元编程的可行方案？

核心问题在于Pydantic默认无法自动区分抽象基类的子类，导致反序列化时丢失子类字段或尝试实例化抽象类。官方推荐的无元编程解决方案是使用鉴别器字段，利用子类的唯一标识字段（比如你的source或frequency）让Pydantic自动匹配对应子类。

具体实现步骤 #

1. 修改抽象基类，添加鉴别器
   在Datasource基类中，用Field(discriminator="字段名")标记用于区分子类的唯一字段（比如source，因为每个子类的source是固定Literal值）。同时保留ABC的抽象方法约束。

   from abc import ABC, abstractmethod
   from pydantic import BaseModel, Field
   from typing import Literal, List
   
   class Datasource(BaseModel, ABC):
       # 用source作为鉴别器，Pydantic会根据该字段值匹配子类
       source: str = Field(..., discriminator="source")
       frequency: str
   
       @abstractmethod
       def hello(self) -> str:
           pass
   

2. 定义子类，保留Literal约束
   每个子类正常定义source的Literal固定值和独有字段，无需额外修改：

   class FirstStaticDatasource(Datasource):
       source: Literal["first_static"]
       frequency: Literal["static"]
       a: int
   
       def hello(self) -> str:
           return "First Static Hello"
   
   class FirstDynamicDatasource(Datasource):
       source: Literal["first_dynamic"]
       frequency: Literal["dynamic"]
       b: str
   
       def hello(self) -> str:
           return "First Dynamic Hello"
   
   class SecondStaticDatasource(Datasource):
       source: Literal["second_static"]
       frequency: Literal["static"]
       c: bool
   
       def hello(self) -> str:
           return "Second Static Hello"
   
   class SecondDynamicDatasource(Datasource):
       source: Literal["second_dynamic"]
       frequency: Literal["dynamic"]
       d: float
   
       def hello(self) -> str:
           return "Second Dynamic Hello"
   

3. Settings类直接使用基类作为列表类型
   无需使用TypeVar或Union子类集合，直接用list[Datasource]即可：

   class Settings(BaseModel):
       datas: List[Datasource]
   

验证效果 #

反序列化时，Pydantic会自动根据source字段的值实例化对应的子类，子类的独有字段会被完整保留，同时抽象方法的约束依然生效（未实现hello的子类会在定义时报错）：

json_data = """
{
    "datas": [
        {"source": "first_static", "frequency": "static", "a": 123},
        {"source": "first_dynamic", "frequency": "dynamic", "b": "test"},
        {"source": "second_static", "frequency": "static", "c": true},
        {"source": "second_dynamic", "frequency": "dynamic", "d": 4.56}
    ]
}
"""

# 反序列化
settings = Settings.model_validate_json(json_data)

# 验证子类实例和字段
assert isinstance(settings.datas[0], FirstStaticDatasource)
assert settings.datas[0].a == 123
assert settings.datas[0].hello() == "First Static Hello"

assert isinstance(settings.datas[1], FirstDynamicDatasource)
assert settings.datas[1].b == "test"

为什么这比__subclasses__更好？ #

• 完全基于Pydantic原生特性，无需依赖元编程手段，代码更稳定、易维护
• 显式指定鉴别字段，逻辑清晰，避免子类自动注册带来的潜在问题
• 保留ABC的抽象方法约束，确保所有子类都实现了必要的方法

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