python-定义带参数的装饰器-27

先建立装饰器框架：

def typeassert(*ty_args,**ty_kargs):
    def decorator(func):
        # func -> a,b 传入两个参数
        # d = {‘a’:int,'b':str}建立一个字典
        def wrapper(*args,**kargs):
            # 判断参数和类型是否在字典中对应
            return func(*args,**kargs)
        return wrapper
    return decorator

会返回一个参数的字典，我们可以通过parameters获取一系列的属性

kind:参数属于哪种类型，位置参数或者是关键字参数

default：参数的默认值

from inspect import signature
# 可以获取函数的响应

def f(a,b,c=2): pass
sig = signature(f)
c = sig.parameters['c']
print(c.default) #输出2

# 建立类型字典,返回一个BoundArguments的对象，不过类型和参数的个数必须对应，否则会报错
bargs =  sig.bind(str,int,int)

print(bargs.arguments['c']) #输出<class 'int'>

# 我们也可以使用bind_partial只对某些参数进行映射
sig.bind_partial(str) 

完成装饰器：

from inspect import signature

def typeassert(*ty_args,**ty_kargs):
    def decorator(func):
        # func -> a,b 传入两个参数
        # d = {‘a’:int,'b':str}建立一个字典
        sig = signature(func)
        btypes = sig.bind_partial(*ty_args, **ty_kargs).arguments
        def wrapper(*args,**kargs):
            # 判断参数和类型是否在字典中对应
            for name,obj in sig.bind(*args,**kargs).arguments.items():
                if name in btypes:
                    if not isinstance(obj,btypes[name]):
                        raise TypeError('"%s" must be "%s"' % (name,btypes[name]))
            return func(*args,**kargs)
        return wrapper
    return decorator
测试

@typeassert(int,str,list)
def f(a,b,c):
    print(a,b,c)

f('1','abc',[1,23,3])