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

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

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

用装饰器实现函数计时器

# coding:utf-8from functools import wrapsimport timedef 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_timerdef 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-8from functools import wrapsimport timeclass 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]    '''

总结

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

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