概念:在代码运行期间动态增加功能的方式,称之为“装饰器”(Decorator).本质上,decorator就是一个返回函数的高阶函数
我们要定义一个能打印日志的decorator,可以定义如下
def log(func):
def wrapper(*args, **kw):
print('call %s():' % func.__name__)
return func(*args, **kw)
return wrapper
把@log放到now()函数的定义处,相当于执行了语句 now = log(now)
如果decorator本身需要传入参数,那就需要编写一个返回decorator的高阶函数
def log(text):
def decorator(func):
def wrapper(*args, **kw):
print('%s %s():' % (text, func.__name__))
return func(*args, **kw)
return wrapper
return decorator
这个3层嵌套的decorator用法如下
@log('execute')
def now():
print('2015-3-25')
把@log放到now()函数的定义处,相当于执行了语句 now = log('execute')(now)
以上两种decorator的定义都没有问题,但还差最后一步。因为我们讲了函数也是对象,它有name等属性,但你去看经过decorator装饰之后的函数,它们的name已经从原来的’now’变成了’wrapper’。Python内置的functools.wraps装饰器可以解决这个问题
import functools
def log(func):
@functools.wraps(func)
def wrapper(*args, **kw):
print('call %s():' % func.__name__)
return func(*args, **kw)
return wrapper
def log(text):
def decorator(func):
@functools.wraps(func)
def wrapper(*args, **kw):
print('%s %s():' % (text, func.__name__))
return func(*args, **kw)
return wrapper
return decorator