關於Python錯誤重試方法總結
阿新 • • 發佈:2021-01-04
前言
Tenacity是一個 Apache 2.0授權的通用重試庫,用 Python 編寫,用於簡化向幾乎所有內容新增重試行為的任務。它起源於一個重新嘗試的分支,可惜這個分支已經不復存在了。
使用Tenacity可以用來進行測試用例的重跑,爬蟲指令碼的重跑,以及搶票的失敗重搶等等。。。可以使用的場景也是比較多。
使用
首先安裝Tenacity
pip install Tenacity
無限重試
第一個重試案例,因為一直是丟擲異常錯誤,所以無限進行重試執行
from tenacity import retry @retry() def test_retry(): print('失敗重試中') raise Exception test_retry()
成功則停止
我們來優化成功一次後程式則終止,否則繼續重試。
from tenacity import retry import random @retry() def test_retry(): if random.randint(0,10) > 1: print('失敗重試中') raise Exception else: print('成功') test_retry()
重試次數
畢竟一直重試需要消耗很多資源,所以我們可以設定一些重試的次數,比如在失敗多少次後停止重試,不管有沒有成功。
from tenacity import retry,stop_after_attempt import random @retry(stop=stop_after_attempt(7)) def test_retry(): # if random.randint(0,10) > 1: print('失敗重試中') raise Exception # else: # print('成功') test_retry()
重試時間
也可以設定執行的時間
from tenacity import retry,stop_after_attempt,stop_after_delay import random from time import sleep @retry(stop=stop_after_delay(3)) def test_retry(): # if random.randint(0,10) > 1: sleep(1) print('失敗重試中') raise Exception # else: # print('成功') test_retry()
條件組合
甚至可以使用多個組合條件進行停止,哪個條件先觸發則執行哪個
from tenacity import retry,stop_after_delay import random from time import sleep @retry(stop=stop_after_delay(3) | stop_after_attempt(2)) def test_retry(): # if random.randint(0,10) > 1: sleep(1) print('失敗重試中') raise Exception # else: # print('成功') test_retry()
重試間隔
重試之間的間隔時間太短,所以讓我們在重試之間等待2秒鐘
from tenacity import retry,stop_after_delay,wait_fixed import random import time @retry(wait=wait_fixed(2)) def test_retry(): # if random.randint(0,10) > 1: print('失敗重試中') print(time.ctime()) raise Exception # else: # print('成功') test_retry()
重試隨機間隔
我們還可以設定一些等待時間範圍
from tenacity import retry,wait_fixed,wait_random import random import time @retry(wait=wait_random(min=1,max=2)) def test_retry(): # if random.randint(0,10) > 1: print('失敗重試中') print(time.ctime()) raise Exception # else: # print('成功') test_retry()
重試前日誌
在執行之前列印日誌
from tenacity import retry,before_log import logging import sys logging.basicConfig(stream=sys.stderr,level=logging.DEBUG) logger = logging.getLogger(__name__) @retry(stop=stop_after_attempt(3),before=before_log(logger,logging.DEBUG)) def test_retry(): print('失敗重試中') raise Exception('Fail') test_retry()
重試後日志
那麼相同的,可以在執行失敗後列印日誌
from tenacity import retry,after_log import logging import sys logging.basicConfig(stream=sys.stderr,after=after_log(logger,logging.DEBUG)) def test_retry(): print('失敗重試中') raise Exception('Fail') test_retry()
基本常用的功能就這些了,如果有需要深入瞭解的可以訪問github地址:https://github.com/jd/tenacity
到此這篇關於關於Python錯誤重試方法總結的文章就介紹到這了,更多相關Python錯誤重試方法內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!