Python系列：如何提高python程序代码的健壮性-51CTO.COM

前言

在编程的时候,我们难免会遇到一些不可靠的情况,比如网络请求失败,数据库连接超时等等。这些不确定性会让我们的程序容易出现各种错误和异常。那么如何来增加程序的容错性和健壮性呢?

可能大多数人会想到使用try except来进行异常捕捉进行失败重试(Retry)。虽然try-escept一个非常常见和有效的方式来增强程序稳定性，但是可能一不小心就会造成栈溢出。

所以接下来我就来介绍一个另外的一个专门用于失败重试的库：retrying。

定义

在Python生态中,retrying库提供了非常便捷的装饰器和函数来帮助我们轻松添加失败重试机制。它可以自定义重试策略、停止条件、等待间隔等,对各种异常进行捕捉处理。使用retrying可以大大减少我们重复编写失败重试轮询的代码量。

1.下载retrying

pip install retrying

2.无参数重试

我们可以直接在函数上使用装饰器@retry来进行失败重试

import retrying
@retry
def func():
    for item in range(0,100):
        result=item / 0
        print(result)
        return result

func()

但是这种方式并不建议使用，就像上面的代码，我们都知道0作为除数就会报错，在上面的func函数中，因为加了@retry装饰器进行失败重试，这样就就会进入一个死循环一直失败一直重试。

所以我们在进行失败重试的时候最好是需要加上一些参数来限制失败重试。

3.有参数重试

(1) stop_max_attempt_number

在retry中传入stop_max_attempt_number参数后可以指定失败重试的次数

@retry(stop_max_attempt_number=2)
def func():
    print(f"记录失败重试")
    for item in range(0,100):
        result=item / 0
        print(result)
        return result

func()

因为这里我们指定了失败后进行两次重试，如果重试执行两次后还是报错则结束重试，将错误信息抛出来。

(2) wait_fixed传入wati_fixed后，可以指定重试的时间

from retrying import retry
import time

# 设置三秒重试一次
@retry(wait_fixed=3000)  
def func():
    print(f"记录失败重试:",time.strftime("%Y-%m-%d %H:%M:%S"))

    result=1 / 0
    print(result)
    return result

func()

配置重试间隔时间后，成语遇到执行失败或者报错后，就会根据设置的重试时间去进行重试执行

(3) wait_random_min和wait_random_max

通常wait_random_min和wait_random_max是一起搭配使用的，可以设置一个重试等待的时间，然后会在设置的时间区间内随机取一个等待时间进行重试

from retrying import retry
import time


@retry(wait_random_min=1000,wait_random_max=9000)
def func():
    print(f"记录失败重试:",time.strftime("%Y-%m-%d %H:%M:%S"))

    result=1 / 0
    print(result)
    return result

func()

(4) wait_exponential_multiplier和wait_exponential_max

官方解释为：以指数的形式产生两次retrying之间的停留时间, 产生的值为2^previous_attempt_number * wait_exponential_multiplier, previous_attempt_number是前面已经retry的次数, 如果产生的这个值超过了wait_exponential_max的大小, 那么之后两个retrying之间的停留值都为wait_exponential_max

通俗来点讲就是每次重试的时间以wait_exponential_multiplier设置的值2，如果重试后还是失败则继续2，直到最后的值等于或则超过wait_exponential_max设置的值后，后面的每一次重试等待时间都是wait_exponential_max设置的值

from retrying import retry
import time

@retry(wait_exponential_multiplier=1000,wait_exponential_max=10000)
def func():
    print(f"记录失败重试:",time.strftime("%Y-%m-%d %H:%M:%S"))

    result=1 / 0
    print(result)
    return result

func()

(5)wait_func

在前面介绍的参数都是如何配置失败冲重试的等待时间或者重试次数之类的，但是我们不能时时刻刻盯着程序，在程序代码发生错误时我们应该要进行发送短信或者邮件之类的提醒才行

在这里就可以使用到wait_func参数，它接收一个可执行函数，返回一个具体的间隔时间数值，单位ms。接收的函数须接收两个参数：attempt_number当前运行次数，delay_since_first_attempt_ms当前重试机制运行时间(单位ms)

from retrying import retry
import time


def func_demo(attempt_number,delay_since_first_attempt_ms):
    print("函数运行失败后运行该函数")

    if attempt_number == 5:
        print("已经重试失败五次了，开始准备发送提醒")

    if attempt_number == 10:
        print("已经重试失败超10次了，发送邮件给相关人员紧急处理")

    if attempt_number >10:
        print("重试时间过长，做一些其他临时方案进行补救")

    # return一个重试的时间
    return 2000


@retry(wait_func=func_demo)
def func():
    print(f"记录失败重试:",time.strftime("%Y-%m-%d %H:%M:%S"))

    result=1 / 0

    return result

func()

使用wait_func通过调用其他可执行的函数，我们可以借助它来做一些临时的补救措施，避免程序一直无法运行而产生的影响。

(6) 其他参数

在retry中还存在有很多参数，有兴趣的小伙伴可以去详细了解下

stop_max_attempt_number：在停止之前尝试的最大次数，最后一次如果还是有异常则会抛出异常，停止运行，默认为5次
stop_max_delay：最大延迟时间，大概意思就是：如果调用的函数出现异常，那么就会重复调用这个函数，最大调用时间，默认为100毫秒
wait_fixed：两次调用方法期间停留时长, 如果出现异常则会一直重复调用，默认 1000毫秒
wait_random_min：在两次调用方法停留时长，停留最短时间，默认为0
wait_random_max：在两次调用方法停留时长，停留最长时间，默认为1000毫秒
wait_incrementing_increment：每调用一次则会增加的时长，默认 100毫秒
wait_exponential_multiplier和wait_exponential_max：以指数的形式产生两次「retrying」之间的停留时间，产生的值为2^previous_attempt_number * wait_exponential_multiplier，previous_attempt_number是前面已经「retry」的次数，如果产生的这个值超过了wait_exponential_max的大小，那么之后两个「retrying」之间的停留值都为wait_exponential_max
retry_on_exception: 指定一个函数，如果此函数返回指定异常，则会重试，如果不是指定的异常则会退出
retry_on_result：指定一个函数，如果指定的函数返回True，则重试，否则抛出异常退出
wrap_exception：参数设置为True/False，如果指定的异常类型，包裹在RetryError中，会看到RetryError和程序抛的Exception error
stop_func: 每次抛出异常时都会执行的函数，如果和stop_max_delay、stop_max_attempt_number配合使用，则后两者会失效 (指定的stop_func会有两个参数：attempts, delay)
wait_func：和stop_func用法差不多。