如何優(yōu)雅地實現(xiàn)在 Python 退出時強制運行一段代碼

“

閱讀本文大概需要 2 分鐘。

設(shè)想這樣一個場景，你要給一個項目開發(fā)測試程序，程序開始運行的時候，會創(chuàng)建初始環(huán)境，測試完成以后，會清理環(huán)境。

這段邏輯本身非常簡單：

setup()
test()
clean()

但由于測試的代碼比較復(fù)雜，你總是在調(diào)試的時候程序異常，導(dǎo)致每次clean()函數(shù)還沒有來得及運行，程序就崩潰了。

你可能想到，如果這樣寫會怎么樣呢：

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

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

有什么辦法，既能讓程序報錯，又能在報錯已經(jīng)還能運行clean()呢？

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

import?atexit

@atexit.register
def?clean():
????print('清理環(huán)境相關(guān)的代碼')

setup()
test()

這樣一來，我們不需要顯式調(diào)用clean函數(shù)了。無論程序正常結(jié)束，還是程序異常報錯，clean函數(shù)里面的內(nèi)容總會執(zhí)行。

如下圖所示：

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

• 你可以注冊多個退出函數(shù)，他們會按照注冊時間從晚到早以此執(zhí)行。例如：

import?atexit

@atexit.register
def?clean_1():
????...

@atexit.register
def?clean_2():
????...

會先運行clean_2()后運行clean_1()

• 如果clean()函數(shù)有參數(shù)，那么你可以不用裝飾器，而是直接調(diào)用atexit.register(clean_1, 參數(shù)1, 參數(shù)2, 參數(shù)3='xxx')。
• 如果程序是被你沒有處理過的系統(tǒng)信號殺死的，那么注冊的函數(shù)無法正常執(zhí)行。
• 如果發(fā)生了嚴(yán)重的Python內(nèi)部錯誤，你注冊的函數(shù)無法正常執(zhí)行。
• 如果你手動調(diào)用了os._exit()，你注冊的函數(shù)無法正常執(zhí)行。