python记录运行时间_python记录代码块运行时间--上下文管理器

python记录时间很简单：

start = time.time()

'''code block'''

end = time.time()

print("elapsed time:%fs" % (end-start))

像这种在执行这个语句块之前，先执行一些准备动作；在语句块执行完成后，再执行一些收尾动作的功能，可以用上下文管理器（Context Manager）实现，使用的关键词就是with（没错，就是打开文件with open(file) as f: 的那个with）。

先定义一个简单的上下文管理器：

import time

class MyTimer(object):

def __enter__(self):

self.start = time.time()

return self

def __exit__(self, *unused):

self.end = time.time()

print("elapsed time:%fs" % (self.end-self.start))

在这个类中，__enter__定义的就是进入代码块之前要运行的函数，__exit__定义的就是退出代码块之后要运行的函数。

使用上面的管理器：

def fib(n):

if n in [1, 2]:

return 1

else:

return fib(n-1) + fib(n-2)

with MyTimer():

print fib(30)

上下文管理器装饰代码块，弥补了装饰器只能装饰函数的缺点。

参考：https://blog.csdn.net/yugongpeng_blog/article/details/49427651​blog.csdn.net$如何用Python装饰器实现一个代码计时器？ - AzureSky - 博客园​www.cnblogs.com