Python装饰器应用案例

最新推荐文章于 2025-03-18 16:16:53 发布

原创最新推荐文章于 2025-03-18 16:16:53 发布 · 181 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#装饰器

Python 专栏收录该内容

2 篇文章

订阅专栏

装饰器功能：函数超时则终止

# -*- coding: utf-8 -*-  
from threading import Thread  
import time  
   
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)