Python中的函数是对象。也因此,函数可以被当做变量使用。
以下代码片段来自于: http://www.sharejs.com/codes/python/8361
# -*- 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)解析@timelimited(2)过程:
执行timelimited(2)
def timelimited(timeout): def decorator(function): def decorator2(*args,**kwargs): ....... 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),可以看到最后返回了decorator函数,其内部参数timeout即为2.此时@timelimited(2)可以看成是@decorator
@decorator
@decorator def somefunction(secs):python解析器遇到@,且后面跟着函数时,会把函数somefunction当做参数传递给decorator函数并执行,即decorator(somefunction),本例中执行 decorator(fn_1)
def decorator(function): def decorator2(*args,**kwargs): ....... 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此例中,执行 decorator(fn_1)后返回的是decorator2,decorator2中function参数为fn_1对象,
最后用返回的decorator2函数替换somefunction,本例中是用decorator2替换了原来的fn_1因此,后面直接调用fn_1(4)时,就是调用了decorator2(4),再在decorator2执行过程中,把参数传给function函数变量执行,最后返回想要的结果。
吐槽一下:感觉示例代码中的decorator2命名为wrapper会更合适一点
昨晚看<<Python in Practice>>看的兴奋了,睡不着,觉得今天得记录下,所以写了这篇文章,不足或错误之处,请大家指正,谢谢!
参考文章
How can I make a chain of function decorators in Python? <<Python in Practice>>http://www.sharejs.com/codes/python/8361