脫離苦海,從避免濫用try...except...開始
有不少人在寫 Python 代碼時,喜歡用 try...except Exception,更有甚者一層套一層,不管有沒有用,先套了再說:
ss="dp-xml">
- ss="alt">def func():
- ss=""> try:
- ss="alt"> "函數(shù)內(nèi)部代碼"
- ss=""> except Exception as e:
- ss="alt"> print('函數(shù)錯誤:', e)
- ss="">
- ss="alt">
- ss="">try:
- ss="alt"> func()
- ss="">except Exception as e:
- ss="alt"> print('函數(shù)錯誤:', e)
根本不管是否有必要,總之套上了try...except...就有了安全感。
俄羅斯套娃套多了以后,噩夢開始了。我們來看看下面這段報錯:
你倒是給我說說,是哪個函數(shù)出了問題?
如果你飽受濫用try...except...之苦,下面三個方法可以讓你脫離苦海。
把問題暴露出來
在程序開發(fā)的初期,不要用try...except...。讓 Python 把問題暴露出來。通過 Python 的報錯,你可以直接看到是哪一行代碼有問題,具體是什么問題。
甚至有時候,不僅不需要捕獲異常,你還應(yīng)該主動拋出異常。在項目完成以后,如果你做的是一個第三方庫,是用來給別人調(diào)用的,那么,你應(yīng)該多拋出異常,而不是擅自返回一個普通的錯誤信息。
例如,你要實現(xiàn)一個函數(shù):query_name,傳入?yún)?shù)是數(shù)字 id,輸出用戶名。你可能會這樣寫:
ss="dp-xml">
- ss="alt">def query_name(user_id):
- ss=""> if not isinstance(user_id, int):
- ss="alt"> return {'success': False, 'msg': '用戶 id 必須是整型'}
- ss=""> ...
但實際上,更好的做法是,直接拋出一個異常:
ss="dp-xml">
- ss="alt">def query_name(user_id):
- ss=""> if not isinstance(user_id, int):
- ss="alt"> raise Exception('用戶 id 必須是整型'}
- ss=""> ...
甚至在某些情況下,你可以使用 Python 的斷言:
ss="dp-xml">
- ss="alt">def query_name(user_id):
- ss=""> assert isinstance(user_id, int), '用戶 id 必須是整型'
- ss="alt"> ...
如下圖所示:
只要 user_id不是整型,就拋出AssertionError。
我們直接執(zhí)行python3 xxx.py時,這些斷言語句會正常工作。但我們可以通過python3 -o xxx.py來讓所有assert xxx語句失效。
盡量早地讓異常暴露出來,才能更早地解決問題。
捕獲具體異常而不是所有異常
只捕獲你明確知道的異常。這些異常你知道它為什么會出現(xiàn),并且你知道應(yīng)該怎么解決它。
例如,我們使用requests請求網(wǎng)站,由于網(wǎng)絡(luò)問題,有時候可能會請求超時。一旦超時 requests 就會拋出超時異常,如下圖所示:
這種情況下,你知道這個地方可能會出現(xiàn)Timeout異常,并且你知道出現(xiàn)的時候,重試就可以了。于是,你可以捕獲這個異常:
大家注意,在這個地方,requests 執(zhí)行了.json()方法。如果URL 返回的內(nèi)容可能不是 JSON 格式的字符串,這里就會報JSONDecodeError,如下圖所示:
如果你不做區(qū)分,一股腦直接用 except Exception,那么你怎么知道,到底是你能夠正常處理的超時問題,還是你不能正常處理的網(wǎng)站內(nèi)容返回異常?
所以,只捕獲你知道它為什么會發(fā)生并且你知道如何處理的異常。對于你無法預(yù)料的或者無法處理的異常,直接拋出。不要擅自捕獲。
強行打印報錯信息
如果實在是萬不得已,你必須用try...except Exception,如何把具體報錯的位置打印出來呢?其實也是有方法的。那就是使用 Python 自帶的traceback模塊。
它的用法非常簡單:
ss="dp-xml">
- ss="alt">import traceback
- ss="">
- ss="alt">try:
- ss=""> 1 + 'a'
- ss="alt">except Exception:
- ss=""> print(traceback.format_exc())
運行效果如下圖所示:
成功把異常所在的行數(shù)和具體的錯誤類型打印了出來。顯然,這樣寫你需要平白無故多寫很多代碼。
總結(jié)try...except...會讓你的代碼看起來沒有問題,但也有可能會掩蓋問題,讓你無法發(fā)現(xiàn)哪里有問題。所以,從看了這篇文章開始,刪除不必要的try...except...。
擁抱異常,讓你無法處理的異常拋出來。程序出現(xiàn)了問題應(yīng)該停止運行,而不是帶著問題繼續(xù)運行,這樣可能會演變成更大的問題。