Python之装饰器的实例

1.1装饰器的应用：参数类型检查

函数参数的检查，一定是在函数外
函数应该作为参数，传入到检查函数中
检查函数拿到函数传入的实际参数，与形参声明对比
__annotations__属性是一个字典，其中包括返回值类型的声明。假设要做位置参数的判断，无法和字典中的声明对应。使用inspect模块
inspect模块：提供获取对象信息的函数，可以检查函数和类、类型检查

from functools import wraps
import inspect
def check(fn):
    @wraps(fn)
    def _check(*args,**kwargs):
        sig = inspect.signature(fn)
        params = sig.parameters     #传入的参数是有序的字典，标识的是函数的签名
        keys = list(params.keys())  #把形参的参数从字典中提取出来，利用list函数迭代出来，在利用索引取值
        values = list(params.values())#把形参的参数注解的类型拿到，用list函数迭代，在利用索引取值
        for k,v in enumerate(args):   #args实参迭代，k可以利用在索引取值上
            print('k={},v={}'.format(k,v))
            if isinstance(v,values[k].annotation):  #拿到参数注解的类型 eg：int or str
                print("==")
        for key,val in kwargs.items():
            if isinstance(val,params[key].annotation):
                print("==")
        result = fn(*args,**kwargs)
        return result
    return _check
@check
def add(x,y:str)->str:
    return x+y
add(x='a',y='b')   #print == 'ab'


def say_hi(func):
    def wrapper(*args,**kwargs):
        print('hi')
        ret = func(*args,**kwargs)
        print('bye')
        return ret
    return wrapper
def say_no(func):
    def wrapper(*args,**kwargs):
        print('yo')
        return func(*args,**kwargs)  #43
        # print('rock & roll')
    return wrapper

@say_hi  # second 其次进行装饰    func = say_hi(func1)  最后执行func
@say_no  # first 先进行装饰func  func1 = say_no(func)   say_no函数已经执行，但是return func #43这个位置没有被调用所以内层函数不执行
def func():
    print('rock & roll')

func()  # 相当于调用 @say_hi的内层嵌套函数

'''
def say_hi(func):
    def wrapper(*args,**kwargs):
        print('hi')
        # ret = func(*args,**kwargs)
        print('yo')
        # return func(*args,**kwargs)
        print('rock & roll')
        print('bye')
        return ret
    return wrapper
def say_no(func):
    def wrapper(*args,**kwargs):
        print('yo')
        # return func(*args,**kwargs)
        print('rock & roll')
    return wrapper
'''

来源：oschina

链接：https://my.oschina.net/u/4403012/blog/4001060

标签

def

wrapper