如何在Python退出時強制運行一段代碼

設想這樣一個場景,你要給一個項目開發測試程序,程序開始運行的時候,會創建初始環境,測試完成以後,會清理環境。

設想這樣一個場景,你要給一個項目開發測試程序,程序開始運行的時候,會創建初始環境,測試完成以後,會清理環境。

如何在Python退出時強制運行一段代碼如何在Python退出時強制運行一段代碼

這段邏輯本身非常簡單:

setup() 
test() 
clean() 

但由於測試的代碼比較複雜,你總是在調試的時候程序異常,導致每次clean()函數還沒有來得及運行,程序就崩潰了。

你可能想到,如果這樣寫會怎麼樣呢:

setup() 
try: 
    text() 
except Exception as e: 
    print('運行異常:', e) 
clean() 

似乎看起來,程序一定會運行到clean()函數,但是,如果你代碼寫的多,你就應該知道,濫用try...except...會讓你非常痛苦。例如它突然給你打印一個運行異常: 1。你根本不知道是哪裏出了問題,也不知道具體出了什麼問題。爲了找到問題,你必須讓程序把錯誤爆出來。但這樣一來,clean()又不能正常運行了。

有什麼辦法,既能讓程序報錯,又能在報錯已經還能運行clean()呢?

這個時候,我們就可以使用Python自帶的atexit這個模塊了。它的使用方法非常簡單:

import atexit 
@atexit.register 
def clean(): 
    print('清理環境相關的代碼')  
setup() 
test() 

這樣一來,我們不需要顯式調用clean函數了。無論程序正常結束,還是程序異常報錯,clean函數裏面的內容總會執行。

如下圖所示:

如何在Python退出時強制運行一段代碼如何在Python退出時強制運行一段代碼

atexit使用中有下面幾個注意事項:

你可以註冊多個退出函數,他們會按照註冊時間從晚到早以此執行。例如:

import atexit 
@atexit.register 
def clean_1(): 
    ...  
@atexit.register 
def clean_2(): 
    ... 
會先運行clean_2()後運行clean_1()

如果clean()函數有參數,那麼你可以不用裝飾器,而是直接調用atexit.register(clean_1, 參數1, 參數2, 參數3='xxx')。
如果程序是被你沒有處理過的系統信號殺死的,那麼註冊的函數無法正常執行。
如果發生了嚴重的Python內部錯誤,你註冊的函數無法正常執行。
如果你手動調用了os._exit(),你註冊的函數無法正常執行。

本文地址:https://www.linuxprobe.com/python-piece-of-code.html

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章