Python 程式報錯崩潰後,如何倒回到崩潰的位置?
假設我們有一段程式,從 Redis 中讀取資料,解析以後提取出裡面的 name 欄位:
import json
import redis client = redis.Redis()
def read():
while True:
data = client.lpop('info')
if data:
yield json.loads(data)
else:
break def parse():
for data in self.read():
print(data['name']) if __name__ == '__main__':parse()
程式碼的邏輯本身很簡單,從 Redis 中一條一條讀取資料,讀到的資料是 JSON 字串,所以先使用json.loads解析成字典。然後讀取字典中的name對應的值。一直讀到Redis 列表為空。
我們執行一下看看:
報錯了,說明Redis 中的某一條資料有問題。你想看看這條有問題的資料,但是現在程式已經崩潰了,程式結束了,這條有問題的資料也就永久丟失了。你再也不可能知道它長什麼樣了。
玩過《火焰紋章-風花雪月》的朋友都知道,主角有一個技能叫做天刻之脈動,如果隊友死了,他可以逆轉時間,回到隊友被殺之前,從而改變隊友的命運。
那麼,在Python裡面我們有沒有什麼辦法讓程式起死回生,看到當初導致程式報錯的那一行程式碼呢?如果你是使用python3 xxx.py執行的程式,那麼確實,除非你能重新匯入剛才的資料,否則無法知道。
但是,如果你是使用如下命令:python3 -i xxx.py啟動的程式,那麼世界就不一樣了,你的程式獲得了起死回生的能力。你可以重新回到事故現場。
我們恢復一下 Redis 的資料(當然,在生產環境裡面你可能就沒有辦法恢復了。但現在寫文章的示例資料,我還是可以回覆的^_^)
然後使用python3 -i read_name.py重新執行這個程式:
可以看到,現在雖然程式崩潰了,但是卻出現了 Python 的互動環境。程式並沒有完全退出。這樣一來,我們就可以輸入魔法指令,讓程式倒退回到報錯的那個地方。輸入命令:
import pdbpdb.pm()
執行效果如下圖所示:
現在,我們已經回到了報錯的那一行了。報錯報的是data這個字典沒有name這個 key,那麼我們就來看看這個字典裡面有什麼,直接輸入變數名data:
原來,這一條有問題的資料,是把name寫成了name1。
總結
pdb是Python自帶的除錯工具。我們使用的PyCharm的除錯功能,也是基於pdb實現的。