为函数动态地添加功能,却又不想改变原函数的定义:
那么就可以用一个函数B接受函数A为参数,然后B再返回一个函数。A是被修饰的函数,返回的函数是修饰过后的函数,我们通过调用修饰后的函数就可以实现动态添加功能。
def log(func):
def wrapper(*args, **kw):
print 'call %s():' % func.__name__
return func(*args, **kw)
return wrapper
def now():
print '2013-12-25'
log(now)()
log(now)是修饰后的函数,加上()就是对wrapper的调用,输出结果为
call now():
2013-12-25
上面是一种基本的装饰器实现方式,但是仔细想想,我希望每次调用now都给它动态添加功能,难道我每次调用now都得像log(now)()这么写?其实python含有对装饰器的语法级支持:
def log(func):
def wrapper(*args, **kw):
print 'call %s():' % func.__name__
return func(*args, **kw)
return wrapper
@log
def now():
print '2013-12-25'
now()
输出和上面是一样的。
在now函数的定义前加上@log标记,now变量就是指向了修饰后的函数,即相当于执行了now=log(now),now变量不再指向原函数对象
上面的是不接受参数的装饰器,下面来看接受参数的装饰器:
def log(text):
def decorator(func):
def wrapper(*args, **kw):
print '%s %s():' % (text, func.__name__)
return func(*args, **kw)
return wrapper
return decorator
@log('execute')
def now():
print '2013-12-25'
now()
输出为:
execute now():
2013-12-25
没有语法级支持的话,本来应该是这么写得:
def now():
print '2013-12-25'
def log(text):
def decorator(func):
def wrapper(*args, **kw):
print '%s %s():' % (text, func.__name__)
return func(*args, **kw)
return wrapper
return decorator
log('execute')(now)()
上面的两种装饰器的定义都没错,但是还差一步。上面那样经过修饰的函数,它们的name属性已经从’now’变成了’wrapper’.可能会影响一些依赖函数签名的代码执行。
无参的decorator:
import functools
def log(func):
@functools.wraps(func)
def wrapper(*args, **kw):
print 'call %s():' % func.__name__
return func(*args, **kw)
return wrapper
或者有参的decorator:
import functoolsdef 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
本文是初学廖雪峰python教程,所写笔记,代码示例是廖大的。