如何在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()，你註冊的函式無法正常執行。

原文地址：