实现此目的的一种方法是使用装饰器( 装饰器的PEP)(有关装饰器的一系列教程文章的第一篇)。这是一个可以满足您需求的示例。
from functools import wraps
from time import time
def timed(f):
@wraps(f)
def wrapper(*args, **kwds):
start = time()
result = f(*args, **kwds)
elapsed = time() - start
print "%s took %d time to finish" % (f.__name__, elapsed)
return result
return wrapper
@timed
def somefunction(countto):
for i in xrange(countto):
pass
return "Done"
为了显示它是如何工作的,我从python提示符下调用了该函数:
>>> timedec.somefunction(10000000)
somefunction took 0 time to finish
'Done'
>>> timedec.somefunction(100000000)
somefunction took 2 time to finish
'Done'
>>> timedec.somefunction(1000000000)
somefunction took 22 time to finish
'Done'