怎么實現在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函數里面的內容總會執行。

如下圖所示：

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

感謝各位的閱讀，以上就是“怎么實現在Python退出時強制運行一段代碼”的內容了，經過本文的學習后，相信大家對怎么實現在Python退出時強制運行一段代碼這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是億速云，小編將為大家推送更多相關知識點的文章，歡迎關注！