歡迎您光臨本站 註冊首頁

Python退出時強制運行一段代碼的實現方法

←手機掃碼閱讀     zhang3221994 @ 2020-04-29 , reply:0

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

這段邏輯本身非常簡單:

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函數裡面的內容總會執行。

如下圖所示:

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(),你註冊的函數無法正常執行。



[zhang3221994 ] Python退出時強制運行一段代碼的實現方法已經有286次圍觀

http://coctec.com/docs/python/shhow-post-232126.html