快捷導(dǎo)航

Python實(shí)現(xiàn)自定義異常堆棧信息的示例代碼

更新時(shí)間：2022年07月19日 10:06:01 作者：古明地覺

當(dāng)我們的程序報(bào)錯(cuò)時(shí)，解釋器會(huì)將整個(gè)異常的堆棧信息全部輸出出來。解釋器會(huì)將異常產(chǎn)生的整個(gè)調(diào)用鏈都給打印出來，那么問題來了，我們能不能自定義這些報(bào)錯(cuò)信息呢？本文就來為大家詳細(xì)講講

當(dāng)我們的程序報(bào)錯(cuò)時(shí)，解釋器會(huì)將整個(gè)異常的堆棧信息全部輸出出來，舉個(gè)例子：

def?foo():
????raise?RuntimeError("拋一個(gè)異常")

def?bar():
????foo()

def?main():
????bar()

main()

如果執(zhí)行這段代碼，會(huì)得到以下報(bào)錯(cuò)信息：

解釋器會(huì)將異常產(chǎn)生的整個(gè)調(diào)用鏈都給打印出來，那么問題來了，我們能不能自定義這些報(bào)錯(cuò)信息呢？

答案是可以的，我們只要拿到這些報(bào)錯(cuò)信息，然后再進(jìn)行修改即可。那么如何才能拿到呢？顯然需要借助于 traceback 對(duì)象。

import?sys

def?foo():
????raise?RuntimeError("拋一個(gè)異常")

def?bar():
????foo()

def?main():
????bar()

try:
????main()
except?Exception:
????#?sys.exc_info()?返回一個(gè)元組，里面有三個(gè)元素
????#?分別是：異常類型、異常值、異常的堆棧信息
????exc_type,?exc_val,?exc_tb?=?sys.exc_info()

????print(exc_type)
????print(exc_val,?type(exc_val))
????print(exc_tb)
????"""
????<class?'RuntimeError'>
????拋一個(gè)異常?<class?'RuntimeError'>
????<traceback?object?at?0x7fe35811de40>
????"""

每一個(gè)函數(shù)在運(yùn)行時(shí)都會(huì)創(chuàng)建一個(gè)棧幀，棧幀在 CPython 里面由 PyFrameObject 結(jié)構(gòu)體表示。同時(shí)每個(gè)棧幀都會(huì)對(duì)應(yīng)一個(gè) PyTracebackObject，也就是異常堆棧，即代碼中的 exc_tb。

上面的代碼在執(zhí)行函數(shù) foo 的時(shí)候出現(xiàn)了異常，那么解釋器會(huì)創(chuàng)建對(duì)應(yīng)的 traceback 對(duì)象，由于 foo 內(nèi)部沒有異常捕獲，因此會(huì)回退到上一級(jí)棧幀，繼續(xù)尋找異常捕獲邏輯。而 foo 的上一級(jí)棧幀是 bar 的棧幀，因此解釋器同樣會(huì)基于 bar 的棧幀創(chuàng)建 traceback 對(duì)象，并且 bar 的 traceback 對(duì)象的 tb_next 指向 foo 的 traceback 對(duì)象。

就這樣棧幀一層層的回退，整個(gè)過程我們稱之為棧幀展開。在棧幀展開的過程中，解釋器不斷地創(chuàng)建與各個(gè)棧幀對(duì)應(yīng)的 traceback，并將其鏈接成鏈表。

如果回退到最外層（模塊）也沒有找到異常捕獲邏輯，那么解釋器就要輸出異常信息了，從模塊對(duì)應(yīng)的 traceback 開始不斷遍歷，將每一層信息都輸出出來，就是我們看到的樣子。

而我們也可以在拿到 traceback 之后，手動(dòng)輸出出來。

import?sys
import?traceback

def?foo():
????raise?RuntimeError("拋一個(gè)異常")

def?bar():
????foo()

def?main():
????bar()

try:
????main()
except?Exception:
????exc_type,?exc_val,?exc_tb?=?sys.exc_info()

????error_msg?=?"".join(
????????traceback.format_exception(
????????????exc_type,?exc_val,?exc_tb)
????)
????print(error_msg)
????"""
????Traceback?(most?recent?call?last):
??????File?"/Users/satori/Desktop/project/main.py",?line?14,?in?<module>
????????main()
??????File?"/Users/satori/Desktop/project/main.py",?line?11,?in?main
????????bar()
??????File?"/Users/satori/Desktop/project/main.py",?line?8,?in?bar
????????foo()
??????File?"/Users/satori/Desktop/project/main.py",?line?5,?in?foo
????????raise?RuntimeError("拋一個(gè)異常")
????RuntimeError:?拋一個(gè)異常
????"""

Python 有一個(gè)標(biāo)準(zhǔn)模塊也叫 traceback，使用它的 format_exception 函數(shù)，我們可以拿到格式化后的異常堆棧信息，而該函數(shù)接收的參數(shù)正是 exc_type, exc_val, exc_tb。

雖然我們拿到了異常的堆棧信息，但是還不夠，因?yàn)檫@里是通過手動(dòng)捕獲異常的方式。而在生產(chǎn)上，很多時(shí)候我們并不知道哪里會(huì)拋出異常，所以我們需要在不使用異常捕獲邏輯的前提下，自動(dòng)捕獲異常。是不是有點(diǎn)繞了呢？我們舉個(gè)例子。

import?sys

def?catch(exc_type,?exc_val,?exc_tb):
????"""
????光看名字，就應(yīng)該知道這三個(gè)參數(shù)的含義了
????"""
????print("報(bào)錯(cuò)啦")
????print(exc_val)

#?之后當(dāng)出現(xiàn)異常的時(shí)候，如果沒有異常捕獲邏輯
#?就會(huì)執(zhí)行?catch?函數(shù)
sys.excepthook?=?catch

raise?ZeroDivisionError("除零錯(cuò)誤")
"""
報(bào)錯(cuò)啦
除零錯(cuò)誤
"""

代碼中我們 raise 了一個(gè)異常，默認(rèn)情況下解釋器應(yīng)該將異常輸出到 stderr 當(dāng)中，然后中止運(yùn)行。但是我們替換了 sys.excepthook，那么在出現(xiàn)異常的時(shí)候，解釋器會(huì)去執(zhí)行我們這里替換之后的 catch 函數(shù)，并自動(dòng)將 exc_type, exc_val, exc_tb 作為參數(shù)傳進(jìn)去。

而一旦 cache 函數(shù)執(zhí)行完畢，程序就結(jié)束了。

那么接下來我們就可以對(duì)異常輸出進(jìn)行改造了，至于怎么改，完全由你來決定，我們這里給個(gè)示例。

import?sys
from?io?import?StringIO
import?re
import?traceback
from?rich?import?print

def?catch(exc_type,?exc_val,?exc_tb):
????buf?=?StringIO()
????#?得到一個(gè)列表
????errors?=?traceback.format_exception(
????????exc_type,?exc_val,?exc_tb)
????buf.write(f"[blue]程序出現(xiàn)異常啦,?客官請(qǐng)看下面:\n\n")
????#?遍歷
????i?=?1
????while?i?<?len(errors):
????????match?=?re.search(r'File?"(.+?)",?line?(\d+),?in?(.+)\s*(.*)',
??????????????????????????errors[i].strip())
????????if?match?is?not?None:
????????????file_path,?lineno,?where,?reason?=?match.groups()
????????????buf.write(
????????????????f"在文件?[red?bold]{file_path}[/red?bold]?"
????????????????f"[green?bold]{where}[/green?bold]?的"
????????????????f"第?[yellow?bold]{lineno}[/yellow?bold]?行\(zhòng)n"
????????????)
????????????buf.write(f"執(zhí)行了?{reason.strip()}\n\n")
????????i?+=?1
????buf.write(f"[cyan?bold]{errors[-1].strip()}")
????print(buf.getvalue())

sys.excepthook?=?catch

def?foo():
????#?raise?a?exception
????{}?+?()

def?bar():
????foo()

def?main():
????bar()

main()

我們執(zhí)行這段代碼，看看它的錯(cuò)誤輸出是什么樣子。