如何运行一段python代码_Python退出时强制运行一段代码如何实现 Python退出时强制运行一段代码实现方法...
本篇文章小編給大家分享一下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()
如果clean()函數有參數,那么你可以不用裝飾器,而是直接調用atexit.register(clean_1, 參數1, 參數2, 參數3='xxx')。
如果程序是被你沒有處理過的系統信號殺死的,那么注冊的函數無法正常執行。
如果發生了嚴重的Python內部錯誤,你注冊的函數無法正常執行。
如果你手動調用了os._exit(),你注冊的函數無法正常執行。
總結
以上是生活随笔為你收集整理的如何运行一段python代码_Python退出时强制运行一段代码如何实现 Python退出时强制运行一段代码实现方法...的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: python什么模块动态调用链接库_py
- 下一篇: .mb是什么文件_神经网络长什么样不知道