一个用于统计函数调用消耗时间的装饰器

2019独角兽企业重金招聘Python工程师标准>>>

装饰器前面提过了，采用python的闭包特性实现：

from time import time
from time import sleep

def count_time():
    def tmp(func):
        def wrapped(*args, **kargs):
            begin_time = time()
            result = func(*args, **kargs)
            end_time = time()
            cost_time = end_time - begin_time
            print '%s called cost time : %s' %(func.__name__, cost_time)
            return result
        return wrapped
    return tmp

@count_time()
def test():
    sleep(0.5)

if __name__ == '__main__':
    test()

我们尝试以下的代码：

class Test:
    @count_time()
    def test(self):
        print 'haha'

if __name__ == '__main__':
#    test()
    a = Test()
    a.test()

代码仍然可以正常工作，因为a.test()仅仅就是给test添加了一个额外的参数a而已。

转载于:https://my.oschina.net/inevermore/blog/388645