在 Requests.request 中设置最大重试次数，增强代码鲁棒性

2024-03-01 04:03:24

在 Requests.request 中设置最大重试次数

问题

Requests 库，一个广泛用于 Python 的 HTTP 库，经常抛出 requests.exception.ConnectionError 异常，提示“最大重试次数超过”的错误信息。虽然这意味着 Requests 会在访问数据时尝试多次，但库文档中却没有提到这一功能。

解决方案

尽管 Requests 库没有直接设置最大重试次数的方法，但有两种解决方案可以实现这一目标：

1. 第三方库

我们可以使用 retrying 等第三方库来控制重试行为。以下代码片段演示了如何使用 retrying 库设置最大重试次数为 3：

import requests
from retrying import retry

@retry(stop_max_attempt_number=3)
def get_data(url):
    return requests.get(url)

# 调用 get_data() 尝试最多 3 次
get_data('https://example.com')

2. 会话适配器

会话适配器允许我们自定义 Requests 库的行为。我们可以创建自己的适配器来处理重试：

import requests

class RetryAdapter(requests.adapters.BaseAdapter):
    def __init__(self, max_retries=3):
        super().__init__()
        self.max_retries = max_retries

    def send(self, request, **kwargs):
        retries = 0
        while retries < self.max_retries:
            try:
                return super().send(request, **kwargs)
            except requests.exceptions.ConnectionError:
                retries += 1

# 创建适配器并注册到会话
session = requests.Session()
session.mount('https://', RetryAdapter(max_retries=3))

# 使用会话进行请求，将重试最多 3 次
session.get('https://example.com')