实例讲解Python的函数闭包使用中应注意的问题
from functools import wraps from time import sleep def retry(attempts=3, wait=2): if attempts 5: retry_times = 3 else: retry_times = attempts if wait 5: retry_wait = 2 else: retry_wait = after def retry_decorator(func): @wraps(func) def wrapped_function(*args, **kwargs): while retry_times > 0: try: return func(*args, **kwargs) except : sleep(retry_wait) retry_times -= 1 return wrapped_function return retry_decorator
简易版的retry装饰器,需要的变量被闭包完美捕捉,逻辑也挺简单明了。问的人说逻辑看着挺正常的,但就是一直报变量retry_times找不到(unresolved reference)的错误提示。
没错仔细捋一下,这是一道送分题呢:闭包捕获的变量(retry_times,retry_wait)相当时引用的retry函数的局部变量,当在wrapped_function的局部作用于里面操作不可变类型的数据时,会生成新的局部变量,但是新生成的局部变量retry_times在使用时还没来得及初始化,因此会提示找不到变量;retry_wait相反能被好好的使用到。
python是duck-typing的编程语言,就算有warning照样跑,写个简单到极限的的函数,用一下装饰器,在wrapped_function逻辑里打个断点看一下各个变量的值也是很快能找到问题的(直接跑也能看到错误:UnboundLocalError: local variable 'retry_attempts' referenced before assignment, 至少比warning msg有用):
@retry(7, 8) def test(): print 23333 raise Exception('Call me exception 2333.') if __name__ == '__main__': test() output: UnboundLocalError: local variable 'retry_times' referenced before assignment
要解决这种问题也好办,用一个可变的容器把要用的不可变类型的数据包装一下就行了(说个好久没写C#代码记不太清楚完全不负责任的题外话,就像在C#.net里面,碰到闭包的时候,会自动生成一个混淆过名字的类然后把要被捕捉的值当作类的属性存着,这样在使用的时候就能轻松get,著名的老赵好像有一篇文章讲Lazy Evaluation的好像涉及到这个话题):
def retry(attempts=3, wait=2): temp_dict = { 'retry_times': 3 if attempts 5 else attempts, 'retry_wait': 2 if wait 5 else wait } def retry_decorate(fn): @wraps(fn) def wrapped_function(*args, **kwargs): print id(temp_dict), temp_dict while temp_dict.get('retry_times') > 0: try: return fn(*args, **kwargs) except : sleep(temp_dict.get('retry_wait')) temp_dict['retry_times'] = temp_dict.get('retry_times') - 1 print id(temp_dict), temp_dict print id(temp_dict), temp_dict return wrapped_function return retry_decorate @retry(7, 8) def test(): print 23333 raise Exception('Call me exception 2333.') if __name__ == '__main__': test()
输出:
4405472064 {'retry_wait': 2, 'retry_times': 3} 4405472064 {'retry_wait': 2, 'retry_times': 3} 23333 4405472064 {'retry_wait': 2, 'retry_times': 2} 23333 4405472064 {'retry_wait': 2, 'retry_times': 1} 23333 4405472064 {'retry_wait': 2, 'retry_times': 0}
从output中可以看到,用dict包装后,程序能够正常的工作,和预期的一致,其实我们也可以从函数的闭包的值再次确认:
>>> test.func_closure[1].cell_contents {'retry_wait': 2, 'retry_times': 2}
我是结尾,PEACE!
声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn核实处理。
相关文章
相关视频
专题推荐
-
独孤九贱-php全栈开发教程
全栈 170W+
主讲:Peter-Zhu 轻松幽默、简短易学,非常适合PHP学习入门
-
玉女心经-web前端开发教程
入门 80W+
主讲:灭绝师太 由浅入深、明快简洁,非常适合前端学习入门
-
天龙八部-实战开发教程
实战 120W+
主讲:西门大官人 思路清晰、严谨规范,适合有一定web编程基础学习
上一篇: EXP/IMP迁移数据
网友评论
文明上网理性发言,请遵守 新闻评论服务协议
我要评论