Pydantic 进阶：不执行函数即可验证其参数的巧妙方法

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

def form_validator_model(func: collections.abc.Callable) -> Type[pydantic.BaseModel]:
    """
    从函数类型注解动态生成 Pydantic 验证模型。

    Args:
        func: 带有类型提示的目标函数。

    Returns:
        一个动态生成的 Pydantic BaseModel 子类，可用于验证函数的参数。
    """
    ann = func.__annotations__.copy()
    # 移除返回值注解，因为 Pydantic 模型字段不关心函数的返回值类型
    ann.pop('return', None) 

    # 动态创建 Pydantic BaseModel 类
    # 类名通常以原函数名加上 '_Validator' 后缀，方便识别
    # 基类为 (pydantic.BaseModel,)，字典中包含类的属性，这里主要是 __annotations__
    return type(f'{func.__name__}_Validator', (pydantic.BaseModel,), {'__annotations__': ann})

# 示例函数
def foo(x: int, y: str, z: Optional[list] = None):
    """一个带有类型提示的示例函数"""
    print(f"Function foo called with x={x}, y={y}, z={z}") # 实际调用时会打印
    pass

# 1. 生成验证模型
# 调用 form_validator_model 函数，传入 foo 函数
FooValidator = form_validator_model(foo)

print(f"生成的验证模型类名: {FooValidator.__name__}")
print(f"模型字段 (基于函数注解): {FooValidator.model_fields.keys()}")

# 2. 尝试验证正确的参数
print("\n--- 验证正确的参数 ---")
try:
    valid_kwargs = {'x': 10, 'y': 'hello'}
    # 实例化 FooValidator，Pydantic 会自动验证传入的关键字参数
    validated_data = FooValidator(**valid_kwargs)
    print(f"验证成功！Pydantic 模型数据: {validated_data.model_dump()}")
    # 此时，foo 函数并未被调用
    # foo(**validated_data.model_dump()) # 如果需要，可以再调用函数
except pydantic.ValidationError as e:
    print(f"验证失败: {e}")

# 3. 尝试验证错误的参数
print("\n--- 验证错误的参数 ---")
try:
    invalid_kwargs = {'x': 'not_an_int', 'y': 123, 'extra_field': True}
    validated_data = FooValidator(**invalid_kwargs)
    print(f"验证成功！Pydantic 模型数据: {validated_data.model_dump()}")
except pydantic.ValidationError as e:
    print(f"验证失败: {e}")

# 4. 验证缺少必要参数的情况
print("\n--- 验证缺少必要参数的情况 ---")
try:
    missing_kwargs = {'x': 5} # 缺少 'y'
    validated_data = FooValidator(**missing_kwargs)
    print(f"验证成功！Pydantic 模型数据: {validated_data.model_dump()}")
except pydantic.ValidationError as e:
    print(f"验证失败: {e}")

# 5. 原始问题中提供的例子
print("\n--- 原始问题中的例子 ---")
def func(a: str, b: int) -> str:
    """一个简单的函数，返回字符串"""
    return a * b

ModelForFunc = form_validator_model(func)
try:
    # 'b' 期望是 int，传入 'bye' 会引发 ValidationError
    ModelForFunc(a='hi', b='bye') 
except pydantic.ValidationError as e:
    print(f"原始例子验证失败 (符合预期): {e}")

try:
    # 正确的参数
    ModelForFunc(a='hello', b=3)
    print("原始例子验证成功 (正确参数)")
except pydantic.ValidationError as e:
    print(f"原始例子验证失败 (不应失败): {e}")