1. 程式人生 > 實用技巧 >Python 程式報錯崩潰後,如何倒回到崩潰的位置?

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實現的。