python计时器代码_$如何用Python装饰器实现一个代码计时器？

有时候我们很希望看到程序中某个函数或某个代码段的耗时情况，那么该如何办呢？本文用两种方式实现了代码计时器的功能，第一种方式是采用装饰器来实现，第二种方式采用上下文管理器实现。

其实计算代码的运行时间，最朴素的想法就是先记录下来某段代码刚开始运行时的时间，等到运行完之后，再看一下结束时的时间，最后和开始运行时的时间求个差值，就是这段代码所花费的时间。

下面两种计时器的实现方式就是用到这样一种非常简单的方法。

用装饰器实现函数计时器

# coding:utf-8

from functools import wraps

import time

def func_timer(function):

'''

用装饰器实现函数计时

:param function: 需要计时的函数

:return: None

'''

@wraps(function)

def function_timer(*args, **kwargs):

print '[Function: {name} start...]'.format(name = function.__name__)

t0 = time.time()

result = function(*args, **kwargs)

t1 = time.time()

print '[Function: {name} finished, spent time: {time:.2f}s]'.format(name = function.__name__,time = t1 - t0)

return result

return function_timer

@func_timer

def test(x,y):

s = x + y

time.sleep(1.5)

print 'the sum is: {0}'.format(s)

if __name__ == '__main__':

test(1,2)

# 输出结果

'''

[Function: test start...]

the sum is: 3

[Function: test finished, spent time: 1.50s]

'''

用上下文管理器实现代码段计时器

上下文管理器其实是一个实现了__enter__和__exit__两个特殊方法的对象，可以用with语法调用。可以参照操作文件的with oepn操作，比如：

with open('data.txt','r') as fin:

data = fin.read()

使用with上下文管理器操作文件的好处就是，不用担心文件使用完之后忘记关闭，上下文管理器会自动帮你关闭。

那么下面就用上下文管理器来实现一个代码段计时器：

# coding:utf-8

from functools import wraps

import time

class MyTimer(object):

'''

用上下文管理器计时

'''

def __enter__(self):

self.t0 = time.time()

def __exit__(self, exc_type, exc_val, exc_tb):

print '[finished, spent time: {time:.2f}s]'.format(time = time.time() - self.t0)

def test(x,y):

s = x + y

time.sleep(1.5)

print 'the sum is: {0}'.format(s)

if __name__ == '__main__':

with MyTimer() as t:

test(1,2)

time.sleep(1)

print 'do other things'

# 输出：

'''

the sum is: 3

do other things

[finished, spent time: 2.53s]

'''

总结

可以看出，上述两种计时器的实现方式各有优缺点，用装饰器实现的计时器优点是使用起来非常方便，给要计时的函数加一个装饰器即可，但不足之处是无法对一个代码片段进行计时。而用上下文管理器实现的计时器就可以对任意一个代码段进行计时，弥补了装饰器计时器的缺陷。

具体用哪种计时器，还是要根据实际情况来选择。