装饰器-decorator

def memo
    cache = {}
    @wraps(func)
    def wrapper(*args):
        if args not in cache:
            cache[args] = func(*args)
        return cache[args]
    return wrapper

1.我们定义如下函数,将输入参数乘2并返回

def f(x):
    '''
    double input
    '''
    return x * 2
>>> f(3), f('a')
>>> 9, 'aa'

2.如果我们要求输入参数x必须为int型,否则返回None,可以通过定义函数fn_int来实现

def fn_int(func):
    def wrapper(*args):
        for arg in args:
            if not isintance(arg, int):
            return 
        return func(*args)
    return wrapper
>>> g = fn_int(f)
>>> f(3), f('a')
6, 'aa'
>>> g(3), g('a')
6, None

3.可以通过添加装饰器符号@来将上面两步骤合并

@fn_int
def f(x):
    '''
    double input
    '''
    return x * 2
>>> f(3), f('a')
6, None

但这时会有如下的问题

>>> f.__name__, f.__doc__
wrapper, None

而我们预期的结果是

>>> f.__name__, f.__doc__
f
    double input

4.上面的问题可以由functools.wraps来解决

from functools import wraps
def fn_int(func):
    @wraps(func)  ### important line
    def wrapper(*args):
        for arg in args:
            if not isintance(arg, int):
                return 
        return func(*args)
    return wrapper
@fn_int
def f(x):
    '''
    double input
    '''
    return x * 2
>>> f(3), f('a')
6, None
>>> f.__name__, f.__doc__
f
    double input
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值