python函数类型检查

最新推荐文章于 2024-02-28 18:30:44 发布

weixin_45340899

最新推荐文章于 2024-02-28 18:30:44 发布

阅读量450

点赞数

分类专栏： python 文章标签： python 开发语言后端

本文链接：https://blog.csdn.net/weixin_45340899/article/details/122114443

版权

python 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

from functools import wraps
from inspect import getfullargspec
from typing import get_type_hints


def validate_input(obj, **kwargs):
    hints = get_type_hints(obj)
    print(hints)
    for parm_name, parm_type in hints.items():
        if parm_name == 'return':
            continue
        if not isinstance(kwargs[parm_name], parm_type):
            raise TypeError("参数：{}类型错误，应该是{}".format(parm_name, parm_type))
    
def type_check(decorator):
    @wraps(decorator)
    def wrapper(*args, **kwargs):
        # 获取func的虚参的名称 ['a', 'b']
        func_args = getfullargspec(decorator)[0]
        print(func_args, args, kwargs)
        # ["a": 1, "b": 2]
        kwargs.update(dict(zip(func_args, args)))
        print(kwargs)
        validate_input(decorator, **kwargs)
        return decorator(**kwargs)
    return wrapper

    
    
@type_check
def func(a: int, b: int) -> int:
    return a + b


print(func(1, b=5))