在 Python 中实现请求的重试

Question

当使用requests 包发送POST 请求时，如何实现5 次重试计数，间隔10 秒。 我找到了很多 GET 请求的示例，而不是 post。

这是我目前正在使用的，有时我会收到 503 错误。如果我收到错误的响应 HTTP 代码，我只需要实现重试。

for x in final_payload:
    post_response = requests.post(url=endpoint, data=json.dumps(x), headers=headers)

#Email me the error
if str(post_response.status_code) not in ["201","200"]:
        email(str(post_response.status_code))

Answer 1

我发现重试的默认行为不适用于 POST。为此需要添加 method_whitelist，例如下面：

def retry_session(retries=5):
    session = Session()
    retries = Retry(total=retries,
                backoff_factor=0.1,
                status_forcelist=[500, 502, 503, 504],
                method_whitelist=frozenset(['GET', 'POST']))

    session.mount('https://', HTTPAdapter(max_retries=retries))
    session.mount('http://', HTTPAdapter(max_retries=retries))

    return session

Answer 2

您可以将urllib3.util.retry 模块与requests 结合使用以获得以下内容：

from urllib3.util.retry import Retry
import requests
from requests.adapters import HTTPAdapter

def retry_session(retries, session=None, backoff_factor=0.3):
    session = session or requests.Session()
    retry = Retry(
        total=retries,
        read=retries,
        connect=retries,
        backoff_factor=backoff_factor,
        method_whitelist=False,
    )
    adapter = HTTPAdapter(max_retries=retry)
    session.mount('http://', adapter)
    session.mount('https://', adapter)
    return session

用法：

session = retry_session(retries=5)
session.post(url=endpoint, data=json.dumps(x), headers=headers)

注意：您还可以从Retry 类继承并自定义重试行为和重试间隔。