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

站在用戶的角度思考問題,與客戶深入溝通,找到蒼溪網(wǎng)站設(shè)計與蒼溪網(wǎng)站推廣的解決方案,憑借多年的經(jīng)驗(yàn),讓設(shè)計與互聯(lián)網(wǎng)技術(shù)結(jié)合,創(chuàng)造個性化、用戶體驗(yàn)好的作品,建站類型包括:成都網(wǎng)站設(shè)計、成都網(wǎng)站建設(shè)、企業(yè)官網(wǎng)、英文網(wǎng)站、手機(jī)端網(wǎng)站、網(wǎng)站推廣、域名申請、網(wǎng)站空間、企業(yè)郵箱。業(yè)務(wù)覆蓋蒼溪地區(qū)。
這段邏輯本身非常簡單:
setup()
test()
clean()
但由于測試的代碼比較復(fù)雜,你總是在調(diào)試的時候程序異常,導(dǎo)致每次clean()函數(shù)還沒有來得及運(yùn)行,程序就崩潰了。
你可能想到,如果這樣寫會怎么樣呢:
setup()
try:
text()
except Exception as e:
print('運(yùn)行異常:', e)
clean()
似乎看起來,程序一定會運(yùn)行到clean()函數(shù),但是,如果你代碼寫的多,你就應(yīng)該知道,濫用try…except…會讓你非常痛苦。例如它突然給你打印一個運(yùn)行異常: 1。你根本不知道是哪里出了問題,也不知道具體出了什么問題。為了找到問題,你必須讓程序把錯誤爆出來。但這樣一來,clean()又不能正常運(yùn)行了。
有什么辦法,既能讓程序報錯,又能在報錯已經(jīng)還能運(yùn)行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使用中有下面幾個注意事項(xiàng):
你可以注冊多個退出函數(shù),他們會按照注冊時間從晚到早以此執(zhí)行。例如:
import atexit
@atexit.register
def clean_1():
...
@atexit.register
def clean_2():
...
會先運(yùn)行clean_2()后運(yùn)行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í)行。
本文標(biāo)題:Python退出時強(qiáng)制運(yùn)行一段代碼具體方法
網(wǎng)頁鏈接:http://www.5511xx.com/article/dpocsdd.html


咨詢
建站咨詢
