Python退出時強制運行一段代碼的實現(xiàn)方法
設(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ā)生了嚴重的Python內(nèi)部錯誤，你注冊的函數(shù)無法正常執(zhí)行。
- 如果你手動調(diào)用了os._exit()，你注冊的函數(shù)無法正常執(zhí)行。
到此這篇關(guān)于Python退出時強制運行一段代碼的實現(xiàn)方法的文章就介紹到這了,更多相關(guān)python退出強制運行代碼內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Python實現(xiàn)一個轉(zhuǎn)存純真IP數(shù)據(jù)庫的腳本分享
工作中我們常需要使用純真IP數(shù)據(jù)庫內(nèi)的數(shù)據(jù)做分析，下面這篇文章主要給大家介紹了利用Python如何實現(xiàn)一個轉(zhuǎn)存純真IP數(shù)據(jù)庫的相關(guān)資料，對大家具有一定的參考學(xué)習(xí)價值，需要的朋友們下面來一起看看吧。2017-05-05
這篇文章主要介紹了Pytest之測試命名規(guī)則的使用，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-04-04
python實現(xiàn)獲取單向鏈表倒數(shù)第k個結(jié)點的值示例
這篇文章主要介紹了python實現(xiàn)獲取單向鏈表倒數(shù)第k個結(jié)點的值,結(jié)合實例形式分析了Python針對單向鏈表的定義、遍歷、傳值、判斷等相關(guān)操作技巧,需要的朋友可以參考下2019-10-10
python格式化字符串的實戰(zhàn)教程(使用占位符、format方法)
我們經(jīng)常會用到%-formatting和str.format()來格式化,下面這篇文章主要給大家介紹了關(guān)于python格式化字符串的相關(guān)資料,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下2022-08-08
裝飾器是 Python 的一個重要部分。簡單地說：他們是修改其他函數(shù)的功能的函數(shù)。他們有助于讓我們的代碼更簡短，也更Pythonic。大多數(shù)初學(xué)者不知道在哪兒使用它們，所以我將要分享下，哪些區(qū)域里裝飾器可以讓你的代碼更簡潔。 首先，讓我們討論下如何寫你自己的裝飾器2021-11-11
這篇文章主要介紹了python之dlib庫的詳細安裝方法,文中有非常詳細的代碼示例,對正在學(xué)習(xí)python的小伙伴們有很好的幫助,需要的朋友可以參考下2021-04-04
這篇文章主要介紹了Python中的路徑問題，幫助大家更好的理解和學(xué)習(xí)python，感興趣的朋友可以了解下2020-09-09
Python語言實現(xiàn)機器學(xué)習(xí)的K-近鄰算法
今天學(xué)習(xí)的算法是KNN近鄰算法。KNN算法是一個監(jiān)督學(xué)習(xí)分類器類別的算法。下面我們來詳細的探討下2015-06-06
Python 使用PyQt5 完成選擇文件或目錄的對話框方法
今天小編就為大家分享一篇Python 使用PyQt5 完成選擇文件或目錄的對話框方法。具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-06-06
這篇文章主要介紹了python實現(xiàn)馬丁策略的實例詳解,本文給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-01-01