def foo():
print "foo"
print foo.__name__
#outputs: foo
# With a decorator, it gets messy
def bar(func):
def wrapper():
print "bar"
return func()
return wrapper
@bar
def foo():
print "foo"
print foo.__name__
#outputs: wrapper
# "functools" can help for that
import functools
def bar(func):
# We say that "wrapper", is wrapping "func"
# and the magic begins
@functools.wraps(func)
def wrapper():
print "bar"
return func()
return wrapper
@bar
def foo():
print "foo"
print foo.__name__
#outputs: foo
使用functools進行裝飾的原因見下面:
1、裝飾器是在Python 2.4之後纔有的特性,所以請檢查你的版本。
2、請記住:裝飾器將會帶來性能問題。
3、裝飾是不可逆的。雖然已經有hacks設計出了可逆的裝飾器,但是基本沒人這麼做。所以一旦一個函數被裝飾過了,就無法還原了。
4、裝飾器包裝了函數,使得調試更加困難。
Python 2.5 解決上面提到的第四個問題。Python 2.5 之後,包含了一個functools模塊,這個模塊提供一個functools.wraps方法,這個方法將被包裝的函數的name, module 和 docstring 都複製到包裝好的函數上去。顯然,functools.wraps也是一個裝飾器