装饰器(无参)
- 它是一个函数
- 函数作为它的形参
- 返回值也是一个函数
- 可以使用@functionname方式,简化调用
注:此处装饰器的定义只是就目前所学(基于函数的理解)的总结,并不准确,只是方便理解。
装饰器是高阶函数,但装饰器是对传入函数的功能的装饰(功能增强)
import datetime
import time
def logger(fn):
def wrapper(*args, **kwargs):
print('args={}, kwargs={}'.format(args, kwargs))
start = datetime.datetime.now()
ret = fn(*args, **kwargs)
duration = datetime.datetime.now() - start
print('function {} took {}s'.format(fn.__name__, duration.total_seconds()))
return ret
return wrapper
@logger # 等价于add = logger(add)
def add(x, y):
time.sleep(2)
return x + y
print(add(4, 5))
函数的调用写法与原函数add()的调用写法相同,但调用的方式不同了。原函数加上装饰器后,把原函数当作参数传给了logger(),原函数的参数由*args和**kwargs收集,函数logger()内部形成了闭包。
上面函数对原函数增加了打印功能,但并没有对原函数修改,这就是装饰器的好处