Python中如何访问方法的类型提示以实现自动参数类型验证

当然有办法！你完全可以利用Python的标准库工具自动提取函数的类型提示，不用每次手动重复参数和类型信息，完美符合Python的简洁风格。下面是具体的实现思路和代码示例：

核心思路：提取函数的类型提示 #

Python提供了inspect模块来获取函数的签名信息，结合typing.get_type_hints()可以可靠地提取函数的类型注解——后者比直接访问函数的__annotations__属性更智能，能自动解析类型别名、ForwardRef（比如字符串形式的类型提示）等情况。

封装成装饰器自动验证 #

最优雅的方式是把验证逻辑封装成一个装饰器，这样你只需要给目标函数加个装饰器，就能自动完成参数类型验证，完全不用手动调用验证函数。

完整实现代码

import inspect
from typing import get_type_hints, Union

def validate_args(func):
    # 提前获取函数的类型提示和参数签名，避免每次调用重复计算
    func_type_hints = get_type_hints(func)
    func_signature = inspect.signature(func)

    def wrapper(*args, **kwargs):
        # 将传入的参数与函数签名绑定，自动处理位置参数、关键字参数和默认值
        bound_arguments = func_signature.bind(*args, **kwargs)
        bound_arguments.apply_defaults()

        # 遍历每个绑定的参数，逐一验证类型
        for param_name, param_value in bound_arguments.arguments.items():
            expected_type = func_type_hints.get(param_name)
            if not expected_type:
                continue  # 没有类型提示的参数跳过验证

            # 处理Union类型（比如Union[int, str]）的情况
            if isinstance(expected_type, Union):
                allowed_types = expected_type.__args__
                if not isinstance(param_value, allowed_types):
                    raise TypeError(
                        f"参数 '{param_name}' 预期类型为 {expected_type}，实际传入 {type(param_value)}"
                    )
            else:
                # 普通类型的验证
                if not isinstance(param_value, expected_type):
                    raise TypeError(
                        f"参数 '{param_name}' 预期类型为 {expected_type}，实际传入 {type(param_value)}"
                    )

        # 类型验证通过，执行原函数
        return func(*args, **kwargs)

    return wrapper

用法示例

现在你的find_integer_sum函数只需要加上装饰器即可，完全不用手动写验证调用：

@validate_args
def find_integer_sum(a: int, b: int):
    return a + b

# 测试正常调用
print(find_integer_sum(3, 5))  # 输出: 8

# 测试类型错误的情况，会自动抛出异常
find_integer_sum(3, "5")  # 抛出 TypeError: 参数 'b' 预期类型为 <class 'int'>，实际传入 <class 'str'>

扩展：处理复杂类型验证 #

上面的实现能覆盖基础类型和Union类型，但对于泛型类型（比如List[int]、Dict[str, float]），由于Python的类型擦除机制，isinstance无法直接验证元素类型。如果你的项目需要这类复杂验证，推荐使用第三方库：

• pydantic：它的@validate_arguments装饰器可以自动处理几乎所有Python类型提示，包括泛型、嵌套模型等，功能非常强大。
• typing-inspect：可以帮助你解析更复杂的类型结构，自己实现泛型验证逻辑。

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