您好,登錄后才能下訂單哦!
小編這次要給大家分享的是如何實現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()
看完這篇關于如何實現Python退出時強制運行一段代碼的文章,如果覺得文章內容寫得不錯的話,可以把它分享出去給更多人看到。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。