原文地址:http://blog.csdn.net/TangHuanan/article/details/45094497
被装饰过的函数的函数名
import time
def decorator(fun):
def wrapper():
start = time.time()
fun()
runtime = time.time()-start
print runtime
return wrapper
@decorator
def do_something():
print "play game"
print do_something.__name__
结果如下:
wrapper
可以看出, do_something的函数名变成了wrapper,这不是我们想要的。原因估计各位也都清楚了。那要怎么去解决呢?
import time
def decorator(fun):
def wrapper():
start = time.time()
fun()
runtime = time.time()-start
print runtime
wrapper.__name__ = fun.__name__
return wrapper
@decorator
def do_something():
print "play game"
print do_something.__name__
结果如下:
do_something
但是这个看起来是不是很不专业, python的unctools.wraps提供了解决方法
import time
import functools
def decorator(fun):
@functools.wraps(fun)
def wrapper():
start = time.time()
fun()
runtime = time.time()-start
print runtime
return wrapper
@decorator
def do_something():
print "play game"
print do_something.__name__
结果如下:
do_something