Python錯誤重試方法
阿新 • • 發佈：2021-01-03
## 前言
Tenacity是一個 Apache 2.0授權的通用重試庫，用 Python 編寫，用於簡化向幾乎所有內容新增重試行為的任務。它起源於一個重新嘗試的分支，可惜這個分支已經不復存在了。
使用Tenacity可以用來進行測試用例的重跑，爬蟲指令碼的重跑，以及搶票的失敗重搶等等。。。可以使用的場景也是比較多。
## 使用
首先安裝Tenacity
```shell
pip install Tenacity
```
### 無限重試
第一個重試案例,因為一直是丟擲異常錯誤，所以無限進行重試執行
```python
from tenacity import retry
@retry()
def test_retry():
print('失敗重試中')
raise Exception
test_retry()
```
### 成功則停止
我們來優化成功一次後程式則終止，否則繼續重試。
```python
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()
```
### 重試時間
也可以設定執行的時間
```python
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()
```
### 條件組合
甚至可以使用多個組合條件進行停止,哪個條件先觸發則執行哪個
```python
from tenacity import retry,stop_after_attempt,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秒鐘
```python
from tenacity import retry,stop_after_attempt,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()
```
### 重試隨機間隔
我們還可以設定一些等待時間範圍
```python
from tenacity import retry,stop_after_attempt,stop_after_delay,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()
```
### 重試前日誌
在執行之前列印日誌
```python
from tenacity import retry,stop_after_attempt,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()
```
### 重試後日志
那麼相同的，可以在執行失敗後列印日誌
```python
from tenacity import retry,stop_after_attempt,after_log
import logging
import sys
logging.basicConfig(stream=sys.stderr,level=logging.DEBUG)
logger = logging.getLogger(__name__)
@retry(stop=stop_after_attempt(3),after=after_log(logger,logging.DEBUG))
def test_retry():
print('失敗重試中')
raise Exception('Fail')
test_retry()
```
基本常用的功能就這些了，如果有需要深入瞭解的可以訪問github地址：https://github.com/jd/