1. 程式人生 > 程式設計 >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函式裡面的內容總會執行。

如下圖所示:

Python退出時強制執行一段程式碼的實現方法

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退出強制執行程式碼內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!