python 写一个计算运行时间的装饰器

最新推荐文章于 2024-11-10 00:00:00 发布

转载最新推荐文章于 2024-11-10 00:00:00 发布 · 201 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/jzdwajue/p/7396757.html

文章标签：

#python #面试

本文介绍了一个使用Python装饰器实现的功能，该功能能够监控被装饰函数的执行时间，并根据执行时间的长短输出不同的提示信息。通过具体示例展示了如何定义和使用装饰器来完成这一任务。

面试题之中的一个。

写一个装饰器wraps，它接收一个參数t。假设函数运行时间超过1秒，输出“bad”，否则输出“goods”。

首先，计算函数的运行时间：

import time


start = time.clock()
for i in range(1000000):
    pass
end = time.clock()


print "cost time = %f s" % (end-start)

结果：

>>> 
cost time = 0.092749 s
>>>

然后：

import time


def warps(t):
    def deco(func):
        def _deco(*args,**kwargs):
            start = time.clock()
            func(*args,**kwargs)
            end = time.clock()
            if end - start > t:
                print 'bad'
            else:
                print 'goods'
        return _deco
    return deco


@warps(10)
def myfunc(*args,**kwargs):
    for i in range(100000):
        pass
    
myfunc()

结果：