如何方便的给你的python函数进行参数类型检查

python有时候很麻烦的一点，为了运算速度，不给你做类型检查。这个特性被各路大佬当成重载的秘宝，但是有时候对工程性的项目来说并不安全。

这里介绍一个notice工具，会对函数形参标注类型和实参类型不一样时进行输出提醒（不报错，不影响运行）

from typing import get_type_hints
from functools import wraps
from inspect import getfullargspec
 
# 定义函数参数类型的检查函数
def parameter_check(obj, **kwargs):
    hints = get_type_hints(obj)
    for label_name, label_type in hints.items():
        # print(label_name)
        # print(label_type)
        # 返回类型不检查 跳过 只检查实际传入参数的类型是否正确
        if label_name == "return":
            continue
        # 判断实际传入的参数是否与函数标签中的参数一致
        if not isinstance(kwargs[label_name], label_type):
            print(f"参数：{label_name} 类型错误 应该为：{label_type}")

# 使用装饰器进行函数包裹
def wrapped_func(decorator):
    @wraps(decorator)
    def wrapped_decorator(*args, **kwargs):
        func_args = getfullargspec(decorator)[0]
        kwargs.update(dict(zip(func_args, args)))
        parameter_check(decorator, **kwargs)
        return decorator(**kwargs)
    return wrapped_decorator

@wrapped_func
def add0(a: int, b: int) -> int:
    return a + b


@wrapped_func
def add1(a: int, b: float = 520.1314) -> float:
    return a + b

# print(add0(1, 1))
print(add0("hello", "world"))

运行结果如下图：

 

 可以看出，进行函数包裹后，仍然能运行函数，只是运行前进行了类型检查和提醒。