Pydantic 进阶：不调用函数验证其参数的类型提示

import collections.abc
import pydantic
from typing import Optional, Type

def form_validator_model(func: collections.abc.Callable) -> Type[pydantic.BaseModel]:
    """
    根据函数的类型提示动态生成一个 Pydantic 验证模型。

    Args:
        func: 需要为其生成验证模型的函数。

    Returns:
        一个 Pydantic BaseModel 类型，可用于验证 func 的参数。
    """
    # 复制函数的 __annotations__，因为我们可能需要修改它
    annotations = func.__annotations__.copy()

    # 移除函数的返回值注解，因为它不是参数
    # Pydantic 模型只关注输入字段
    annotations.pop('return', None)

    # 动态创建 Pydantic BaseModel 类
    # type() 函数的第三个参数是类的属性字典
    # 我们将 __annotations__ 直接传递给它，Pydantic 会自动识别并创建字段
    model_name = f'{func.__name__}_Validator'
    DynamicValidatorModel = type(model_name, (pydantic.BaseModel,), {'__annotations__': annotations})
    return DynamicValidatorModel

# 定义一个示例函数
def foo(x: int, y: str, z: Optional[list] = None) -> str:
    """一个带有类型提示的示例函数。"""
    return f"{x} - {y} - {z}"

# 使用辅助函数生成验证模型
FooValidator = form_validator_model(foo)

# 示例 1: 有效的参数
print("--- 验证有效参数 ---")
try:
    valid_kwargs = {'x': 10, 'y': 'hello', 'z': [1, 2, 3]}
    validated_data = FooValidator(**valid_kwargs)
    print(f"验证成功: {validated_data.model_dump()}")
    # 此时，validated_data 是一个 FooValidator 实例，其属性已通过验证
    # 我们可以访问它们，例如 validated_data.x, validated_data.y
except pydantic.ValidationError as e:
    print(f"验证失败: {e}")

print("\n--- 验证部分参数 (可选参数缺失) ---")
try:
    valid_kwargs_partial = {'x': 20, 'y': 'world'} # z 缺失，但它是 Optional
    validated_data_partial = FooValidator(**valid_kwargs_partial)
    print(f"验证成功 (部分参数): {validated_data_partial.model_dump()}")
except pydantic.ValidationError as e:
    print(f"验证失败: {e}")

# 示例 2: 无效的参数类型
print("\n--- 验证无效参数类型 ---")
try:
    invalid_kwargs_type = {'x': 'not_an_int', 'y': 'hi'}
    FooValidator(**invalid_kwargs_type) # 这将引发 ValidationError
except pydantic.ValidationError as e:
    print(f"验证失败: {e}")
    # Pydantic 会详细指出哪个字段验证失败以及原因

# 示例 3: 缺少必需参数
print("\n--- 验证缺少必需参数 ---")
try:
    missing_kwargs = {'y': 'only_y'}
    FooValidator(**missing_kwargs) # 这将引发 ValidationError
except pydantic.ValidationError as e:
    print(f"验证失败: {e}")