Python: 使用装饰器“@”取得函数执行时间

Python中可以使用装饰器对函数进行装饰(或说包装),利用这个特性,可以很方便、简洁地解决一些问题,比如获得函数执行时间的问题。

首先,我们定义一个函数,如下:

def exe_time(func):
	def new_func(*args, **args2):
		t0 = time.time()
		print "@%s, {%s} start" % (time.strftime("%X", time.localtime()), func.__name__)
		back = func(*args, **args2)
		print "@%s, {%s} end" % (time.strftime("%X", time.localtime()), func.__name__)
		print "@%.3fs taken for {%s}" % (time.time() - t0, func.__name__)
		return back
	return new_func

当然,不要忘了用 import time 导入 time 模块。这个函数就可以用作我们计算函数执行时间的修饰器了。

接下来,我们就可以在需要计时的函数前一行引用它作为装饰,比如:

@exe_time
def foo():
	for i in xrange(10000000):
		pass

注意最上方的 @exe_time,装饰器的语法以 @ 开头,接着是装饰函数,在本例中为 exe_time。这时,执行函数 foo,装饰器就会在控制台打印出这个函数的执行时间了。

完整代码如下:

# -*- coding: utf-8 -*-
import time

# --exe_time
def exe_time(func):
	def new_func(*args, **args2):
		t0 = time.time()
		print "@%s, {%s} start" % (time.strftime("%X", time.localtime()), func.__name__)
		back = func(*args, **args2)
		print "@%s, {%s} end" % (time.strftime("%X", time.localtime()), func.__name__)
		print "@%.3fs taken for {%s}" % (time.time() - t0, func.__name__)
		return back
	return new_func
# --end of exe_time

@exe_time
def foo():
	for i in xrange(10000000):
		pass

if __name__ == "__main__":
	foo()

在我的本本上的运行结果为:

@13:12:27, {foo} start
@13:12:27, {foo} end
@0.203s taken for {foo}

当然,上面只是一个很简单的示例,事实上,本例中的装饰器 exe_time 不仅可以装饰类似上面 foo 这样不带参数的函数,也能装饰带任意参数的函数,甚至还可以装饰类的方法,用法与上面是一样的。

除了计算运行时间外,装饰器还可以有很多用途,比如记录运行日志等,更多的用途等待更多的朋友去发掘。

分类:编程标签:Python

相关文章:

评论:

暂无评论

发表评论: