如何在 Python 中重试循环

在编写 Python 程序时，重试循环是一种常见的编程技巧。重试循环可以帮助我们优化程序的稳定性和可靠性，以应对网络延迟或其他不可预测的因素。在本文中，我们将介绍如何在 Python 中实现重试循环，并且提供了一些编程技巧和注意事项。

1. 使用异常处理语句

在 Python 中实现重试循环的最简单方法是使用异常处理语句来捕获特定异常，并在异常发生时重新执行循环。示例如下：

import requests
import time

url = "https://example.com/api"

for i in range(3):
    try:
        response = requests.get(url)
        break # 如果没有异常，跳出循环
    except requests.exceptions.RequestException as e:
        print("Error: ", e)
        time.sleep(3) # 等待 3 秒
else:
    print("请求失败")

在上面的代码中，我们使用了 requests 库来发送 GET 请求，并设置了最多 3 次重试的机会。如果请求成功，我们将会跳出循环。但是如果请求失败，我们将会捕获 requests.exceptions.RequestException 异常，并等待 3 秒钟再次执行循环。如果循环执行 3 次后仍然失败，我们将会得到 “请求失败” 的输出。

2. 设置超时

有时候，代码执行时间过长会导致代码中断的情况发生。因此，在实现重试循环时，我们需要设置一个超时时间，以便在代码执行时间过长时自动退出循环，并防止代码永久运行。示例如下：

import requests
import time

url = "https://example.com/api"
timeout = 5 # 超时时间为 5 秒

for i in range(3):
    try:
        response = requests.get(url, timeout=timeout)
        response.raise_for_status() # 抛出 HTTPError 异常
        break # 如果没有异常，跳出循环
    except (requests.exceptions.RequestException, requests.exceptions.HTTPError) as e:
        print("Error: ", e)
        time.sleep(3) # 等待 3 秒
else:
    print("请求失败")

在上面的示例中，我们设置了一个超时时间为 5 秒。如果请求在 5 秒内没有完成，程序将会抛出一个异常，并调用对应的异常处理语句。我们还使用了 response.raise_for_status() 方法检查 HTTP 响应是否存在异常。如果存在异常，我们将会捕获 requests.exceptions.HTTPError 异常，然后重新执行循环。

3. 为每个重试设置不同的延迟时间

在实现重试循环时，我们需要为每个重试设置不同的延迟时间。这样可以使重试的时间间隔更加均匀，从而减少网络拥塞的影响。示例如下：

import requests
import time
import random

url = "https://example.com/api"

for i in range(3):
    try:
        response = requests.get(url)
        response.raise_for_status() # 抛出 HTTPError 异常
        break # 如果没有异常，跳出循环
    except (requests.exceptions.RequestException, requests.exceptions.HTTPError) as e:
        print("Error: ", e)
        delay = 2 ** i + random.random() # 1, 2, 4 秒的随机时间间隔
        time.sleep(delay)
else:
    print("请求失败")

上面的示例中，我们将每个重试的延迟时间设置为一系列的 2 的幂次方加上一个随机的浮点数。这种方式可以使每个重试之间的时间间隔不同，并且避免了同时访问重试的网络流量拥塞问题。

注意事项：

1. 尽量减少循环次数

在实现重试循环时，我们应该尽量减少循环的次数。如果循环次数过多，可能会导致网络传输和系统资源消耗过高，可能会影响程序的稳定性和性能。因此，我们应该根据实际情况合理设置重试次数。

2. 记录日志记录

在实现重试循环时，我们应该将执行过程中的各种异常情况记录到日志中，便于进行排查问题和程序优化。我们可以使用 Python 的日志库（logging）来记录日志。

3. 避免在循环内进行写操作

在实现重试循环时，我们应该避免在循环内进行写操作。如果每次执行循环都产生一个新的写操作，可能会导致写入数据不稳定和内存泄漏等问题。因此，我们应该在循环外部定义写操作，并在循环内使用这些操作。

总结

在 Python 中实现重试循环的方法有很多，我们可以使用异常处理语句、设置超时和为每个重试设置不同的延迟时间等。在编写程序时，我们应该根据实际情况合理设置重试次数，并记录日志记录以方便进行排查问题和程序优化。我们还应该避免在循环内进行写操作以避免数据写入异常。

火焰兔

海报 链接