怎么實(shí)現(xiàn)在Python退出時(shí)強(qiáng)制運(yùn)行一段代碼

這篇文章主要講解了“怎么實(shí)現(xiàn)在Python退出時(shí)強(qiáng)制運(yùn)行一段代碼”，文中的講解內(nèi)容簡(jiǎn)單清晰，易于學(xué)習(xí)與理解，下面請(qǐng)大家跟著小編的思路慢慢深入，一起來研究和學(xué)習(xí)“怎么實(shí)現(xiàn)在Python退出時(shí)強(qiáng)制運(yùn)行一段代碼”吧！

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

這段邏輯本身非常簡(jiǎn)單：

setup() test() clean()

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

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

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

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

有什么辦法，既能讓程序報(bào)錯(cuò)，又能在報(bào)錯(cuò)已經(jīng)還能運(yùn)行clean()呢?

這個(gè)時(shí)候，我們就可以使用Python自帶的atexit這個(gè)模塊了。它的使用方法非常簡(jiǎn)單：

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

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

如下圖所示：

atexit使用中有下面幾個(gè)注意事項(xiàng)：

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

import atexit  @atexit.register def clean_1():     ...  @atexit.register def clean_2():     ...

會(huì)先運(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)信號(hào)殺死的，那么注冊(cè)的函數(shù)無法正常執(zhí)行。

• 如果發(fā)生了嚴(yán)重的Python內(nèi)部錯(cuò)誤，你注冊(cè)的函數(shù)無法正常執(zhí)行。

• 如果你手動(dòng)調(diào)用了os._exit()，你注冊(cè)的函數(shù)無法正常執(zhí)行。

感謝各位的閱讀，以上就是“怎么實(shí)現(xiàn)在Python退出時(shí)強(qiáng)制運(yùn)行一段代碼”的內(nèi)容了，經(jīng)過本文的學(xué)習(xí)后，相信大家對(duì)怎么實(shí)現(xiàn)在Python退出時(shí)強(qiáng)制運(yùn)行一段代碼這一問題有了更深刻的體會(huì)，具體使用情況還需要大家實(shí)踐驗(yàn)證。這里是億速云，小編將為大家推送更多相關(guān)知識(shí)點(diǎn)的文章，歡迎關(guān)注！