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

来源:互联网 发布:数据库事务一致性 编辑:程序博客网 时间:2024/06/11 04:28

面试题之一。

写一个装饰器wraps,它接收一个参数t,如果函数执行时间超过1秒,输出“bad”,否则输出“goods”。


首先,计算函数的执行时间:

import timestart = time.clock()for i in range(1000000):    passend = time.clock()print "cost time = %f s" % (end-start)
结果:

>>> cost time = 0.092749 s>>> 


然后:

import timedef 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()  

结果:

>>> goods>>> 







参考文档:九步学习装饰器



0 0
原创粉丝点击