python中判断一个运行时间过长的函数

最新推荐文章于 2022-10-17 09:15:00 发布

最新推荐文章于 2022-10-17 09:15:00 发布 · 473 阅读

文章标签：

#Python #thread

python 专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种使用Python多线程实现函数执行超时的方法。通过定义一个TimeLimited类继承自Thread，可以为任意函数设置执行时限。如果函数未能在规定时间内完成，则会抛出TimeoutException异常。这种方式适用于需要限制函数执行时间的场景。


class TimeoutException(Exception):
    pass

ThreadStop = Thread._Thread__stop＃获取私有函数

def timelimited(timeout):
    def decorator(function):
        def decorator2(*args,**kwargs):
            class TimeLimited(Thread):
                def __init__(self,_error= None,):
                    Thread.__init__(self)
                    self._error =  _error

                def run(self):
                    try:
                        self.result = function(*args,**kwargs)
                    except Exception,e:
                        self._error =e

                def _stop(self):
                    if self.isAlive():
                        ThreadStop(self)

            t = TimeLimited()
            t.start()
            t.join(timeout)

            if isinstance(t._error,TimeoutException):
                t._stop()
                raise TimeoutException('timeout for %s' % (repr(function)))

            if t.isAlive():
                t._stop()
                raise TimeoutException('timeout for %s' % (repr(function)))

            if t._error is None:
                return t.result

        return decorator2
    return decorator

@timelimited(2)
def fn_1(secs):
    time.sleep(secs)
    return 'Finished'

if __name__ == "__main__":
    print fn_1(4)