from functools import wraps from time import sleep def retry(attempts=3, wait=2): if attempts < 0 or attempts > 5: retry_times = 3 else: retry_times = attempts if wait < 0 or 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 < 0 or attempts > 5 else attempts, 'retry_wait': 2 if wait < 0 or 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!
聲明:本網頁內容旨在傳播知識,若有侵權等問題請及時與本網聯系,我們將在第一時間刪除處理。TEL:177 7030 7066 E-MAIL:11247931@qq.com