詳解Python如何優(yōu)雅的重試

更新時(shí)間：2023年07月21日 14:17:50 作者：Harpsichord1207

這篇文章主要為大家介紹了Python如何優(yōu)雅的重試詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

正文

在編碼中，一些涉及網(wǎng)絡(luò)連接的代碼片段經(jīng)常需要重試，本文講解了如何一步一步實(shí)現(xiàn)一個(gè)優(yōu)雅的retry裝飾器以及tenacity庫(kù)的使用。

原始版本v0.0

假如有一個(gè)函數(shù)形式如下，函數(shù)有一些建立網(wǎng)絡(luò)連接的邏輯

def f():
    # do some connections
    return 0

為了避免偶爾的網(wǎng)絡(luò)連接失敗，需要加上重試機(jī)制，那么最簡(jiǎn)單的形式就是在對(duì)應(yīng)的代碼片段加一個(gè)循環(huán)，循環(huán)體里使用異常捕獲，連接成功時(shí)退出循環(huán)，否則就重復(fù)執(zhí)行相關(guān)邏輯，此時(shí)修改之后的函數(shù)f如下

def f():
    while 1:
        try:
            # do some connections
            break
        except ConnectionError:
            continue
    return 0

裝飾器版本v1.0

可以使用裝飾器對(duì)代碼進(jìn)行抽象。例如現(xiàn)在有兩個(gè)函數(shù)f1和f2需要加上重試機(jī)制，寫(xiě)一個(gè)名為retry的裝飾器函數(shù)，用其裝飾f1和f2即可。這樣做避免了對(duì)老代碼的修改，同時(shí)也實(shí)現(xiàn)了代碼復(fù)用。示例如下

def retry(f):
    def wrap(*args, **kwargs):
        while 1:
            try:
                return f(*args, **kwargs)
            except ConnectionError:
                continue
    return wrap

@retry
def f1():
    # do some connections
    return 0

@retry
def f2():
    # do some other connections
    return 0

帶參數(shù)的裝飾器版本v1.1

v1.0的版本retry裝飾器還有一些問(wèn)題，假如有的函數(shù)想重試3次，有的想重試5次，重試的間隔也根據(jù)不同函數(shù)不一樣，v1.0是無(wú)法實(shí)現(xiàn)的。此時(shí)可以借助帶參數(shù)的三層裝飾器，例如以下代碼實(shí)現(xiàn)的retry裝飾器，可以傳入times和interval兩個(gè)參數(shù)來(lái)設(shè)定重試次數(shù)和重試間隔

def retry(times, interval):
    def decorator(f)
        def wrap(*args, **kwargs):
            while times:
                try:
                    return f(*args, **kwargs)
                except ConnectionError:
                    times -= 1
                    time.sleep(interval)
                    continue
        return wrap
    return decorator

# 重試3次每次間隔10秒
@retry(times=3, interval=10)
def f1():
    # do some connections
    return 0

# 重試5次每次間隔15秒
@retry(times=5, interval=15)
def f2():
    # do some other connections
    return 0

裝飾器支持參數(shù)之后，可以根據(jù)需要定義更豐富的參數(shù)，比如通過(guò)參數(shù)來(lái)設(shè)定需要捕獲哪些異常等。

tenacity版本

tenacity是一個(gè)第三方開(kāi)源庫(kù)，用于函數(shù)的重試，實(shí)際上它的功能與原理是上面自己寫(xiě)的代碼類(lèi)似的！只是它可定義的參數(shù)更豐富，如果不想重復(fù)造輪子，拿來(lái)直接用就可以。代碼示例如下

from tenacity import retry, stop_after_attempt, wait_fixed

# 不帶任何參數(shù)的重試
@retry
def f():
    # do some connections
    return 0

# 重試5次每次間隔15秒
@retry(stop=stop_after_attempt(5), wait=wait_fixed(15))
def f():
    # do some connections
    return 0

以上就是詳解Python如何優(yōu)雅的重試的詳細(xì)內(nèi)容，更多關(guān)于Python優(yōu)雅重試的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: