昨天正當(dāng)我用十成一陽指功力戳鍵盤、昏天暗地coding的時(shí)候，正好被人問了一個(gè)問題，差點(diǎn)沒收好功，洪荒之力側(cè)漏震傷桌邊的人，廢話不多說，先上栗子（精簡(jiǎn)版，只為說明問題）：
from functools import wrapsfrom time import sleepdef 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
簡(jiǎn)易版的retry裝飾器，需要的變量被閉包完美捕捉，邏輯也挺簡(jiǎn)單明了。問的人說邏輯看著挺正常的，但就是一直報(bào)變量retry_times找不到（unresolved reference)的錯(cuò)誤提示。
沒錯(cuò)仔細(xì)捋一下，這是一道送分題呢：閉包捕獲的變量（retry_times，retry_wait）相當(dāng)時(shí)引用的retry函數(shù)的局部變量，當(dāng)在wrapped_function的局部作用于里面操作不可變類型的數(shù)據(jù)時(shí)，會(huì)生成新的局部變量，但是新生成的局部變量retry_times在使用時(shí)還沒來得及初始化，因此會(huì)提示找不到變量；retry_wait相反能被好好的使用到。
python是duck-typing的編程語言，就算有warning照樣跑，寫個(gè)簡(jiǎn)單到極限的的函數(shù)，用一下裝飾器，在wrapped_function邏輯里打個(gè)斷點(diǎn)看一下各個(gè)變量的值也是很快能找到問題的（直接跑也能看到錯(cuò)誤: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要解決這種問題也好辦，用一個(gè)可變的容器把要用的不可變類型的數(shù)據(jù)包裝一下就行了(說個(gè)好久沒寫C#代碼記不太清楚完全不負(fù)責(zé)任的題外話，就像在C#.net里面，碰到閉包的時(shí)候，會(huì)自動(dòng)生成一個(gè)混淆過名字的類然后把要被捕捉的值當(dāng)作類的屬性存著，這樣在使用的時(shí)候就能輕松get，著名的老趙好像有一篇文章講Lazy Evaluation的好像涉及到這個(gè)話題）：
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}233334405472064 {'retry_wait': 2, 'retry_times': 2}233334405472064 {'retry_wait': 2, 'retry_times': 1}233334405472064 {'retry_wait': 2, 'retry_times': 0}從output中可以看到，用dict包裝后，程序能夠正常的工作，和預(yù)期的一致，其實(shí)我們也可以從函數(shù)的閉包的值再次確認(rèn)：
>>> test.func_closure[1].cell_contents{'retry_wait': 2, 'retry_times': 2}我是結(jié)尾，PEACE!
新聞熱點(diǎn)
疑難解答
圖片精選
