由于python动态类型的特点，参数类型只是作为一个提示的作用，不会实际的报错。以下参数类型装饰器主要是使用inspect模块获取修饰方法传入的参数以及参数的类型，然后通过isinstance判断类型是否相同，不同类型会raise一个报错信息
import functools
import inspect
from inspect import Parameter
"""
类型检查装饰器小demo
"""
def func_check(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
signature = inspect.signature(func)
# 参数类型
params = signature.parameters
# (input_value, )
# 判断传进来的值是否符合类型注解
for in_value, parameter in zip(args, params.values()):
# param.empty == _empty
# 获取参数的类型
param_type = parameter.annotation
# 如果不为空和类型不是注解的类型，抛出错误
if param_type is not Parameter.empty and not isinstance(
in_value, param_type):
raise TypeError(
f'parameter "{parameter.name}" should be {param_type.__name__}, but got {type(in_value).__name__}'
)
# 以赋值形式传参
for k, v in kwargs.items():
# 判断是否是自定义的参数
if k in list(params.keys()):
# 获取进来的值
in_value = kwargs[k]
# 获取自定义参数的注解类型
param_type = params[k].annotation
# 如果不为空和类型不是注解的类型，抛出错误
if param_type is not Parameter.empty and not isinstance(
in_value, param_type):
raise TypeError(
f'parameter "{params[k].name}" should be {param_type.__name__}, but got {type(in_value).__name__}'
)
# 结果类型检验
ret = func(*args, **kwargs)
ret_type = signature.return_annotation
if not isinstance(ret, (Parameter.empty, ret_type)):
raise TypeError(
f'the result of function "{func.__name__}" should be {ret_type.__name__}, but got {type(ret).__name__}'
)
return ret
return wrapper
@func_check
def test(a: int, b: str, c: str = None) -> str:
return f'{a}{b}{c}'
print(test(b='1', a=1, c='3'))
传入参数和类型不同时，错误信息提示：
TypeError: parameter "c" should be str, but got int
