如何为Pydantic基类实现自动生成虚拟测试数据的子类实例化方法

我明白你现在的痛点——手动映射各种类型组合确实太繁琐了，根本没法扩展。其实我们可以利用Python的类型解析工具（比如typing模块里的get_args和get_origin），结合Pydantic字段的元信息，来优雅地实现自动生成虚拟数据的逻辑。

下面是完整的可运行实现：

from __future__ import annotations
from pydantic import BaseModel
from typing import Union, get_args, get_origin


def generate_dummy_value(annotation) -> any:
    # 处理可选类型（Union[X, None]）
    origin = get_origin(annotation)
    if origin is Union:
        args = get_args(annotation)
        # 如果包含None，优先返回None（和你的示例逻辑一致）
        if type(None) in args:
            return None
        # 否则取第一个类型生成对应值
        else:
            return generate_dummy_value(args[0])
    
    # 处理容器类型
    if origin is list:
        return []
    if origin is dict:
        return {}
    
    # 处理基础类型
    if annotation is int:
        return 0
    elif annotation is str:
        return ""
    elif annotation is float:
        return 0.0
    elif annotation is bool:
        return False
    
    # 对于未覆盖的类型，可以返回None或者根据需求扩展
    return None


class BaseModelWrapper(BaseModel):
    @classmethod
    def make_dummy(cls) -> BaseModelWrapper:
        dummy_kwargs = {}
        for name, field in cls.model_fields.items():
            # 跳过有默认值的字段（Pydantic会自动填充默认值）
            if field.default is not None or not field.is_required():
                continue
            
            # 为必填字段生成虚拟值
            dummy_value = generate_dummy_value(field.annotation)
            dummy_kwargs[name] = dummy_value
        
        # 传入生成的参数创建实例
        return cls(**dummy_kwargs)


class XXX(BaseModelWrapper):
    a: int | None
    b: str
    c: int
    d: int | None = None
    e: list[str]


# 测试验证，和你期望的结果完全一致
dummy_instance = XXX.make_dummy()
print(dummy_instance)
# 输出：XXX(a=None, b='', c=0, d=None, e=[])

关键逻辑说明 #

• 类型解析：用get_origin识别复合类型（比如Union、list），get_args获取类型的具体参数，轻松处理int | None这类联合类型。
• 字段筛选：只处理必填且无默认值的字段，有默认值的字段交给Pydantic自动处理，减少冗余逻辑。
• 可扩展性：如果需要支持更多类型（比如嵌套Pydantic模型、datetime），只需在generate_dummy_value里添加分支即可。比如支持嵌套模型的扩展：

  class YYY(BaseModelWrapper):
      f: str
      g: int
  
  class ZZZ(BaseModelWrapper):
      h: YYY
      i: list[YYY]
  
  # 生成嵌套虚拟数据
  zzz_dummy = ZZZ.make_dummy()
  print(zzz_dummy)
  # 输出：ZZZ(h=YYY(f='', g=0), i=[])
  

备注：内容来源于stack exchange，提问作者Matt