一日一技：方法不对，代码翻倍。Requests如何正确重试？

发表于 更新于 阅读次数： 本文字数： 548 阅读时长 ≈ 2 分钟

程序员是一个需要持续学习的群体，如果你发现你现在写的代码跟你5年前的代码没什么区别，说明你掉队了。

我们在做Python开发时，经常使用一些第三方库，这些库很多年来持续添加了新功能。但我发现很多同学在使用这些第三方库时，根本不会使用新的功能。他们的代码跟几年前没有任何区别。

举个例子，使用Request发起HTTP请求，请求失败时，不管什么原因，原地重试最多3次。很多人主要有下面3种写法来重试。

常见的老方法

使用第三方库

这类同学会使用一些专业做重试的第三方库，例如tenacity。详见我的这篇文章：Tenacity——Exception Retry 从此无比简单

手动写装饰器

这类同学会使用装饰器，所以一般会手写装饰器从而复用，例如：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
def retry(func):
    def wrap(*args, **kwargs):
        for _ in range(3):
            try:
                result = func(*args, **kwargs)
                return result
            except Exception as e:
                print('报错了，重试')
        return {}
    return wrap


@retry
def make_request(url):
    print('以下是发起请求的相关代码')

反复for循环

还有一些同学，写代码走的是野路子：

1
2
3
4
5
6
7
8
def login():
    for i in range(10):  # 重试10次
        try:
            resp = requests.get('某某URL')
            return resp.json()
        except Exception as e:
            print(f'请求报错了，重试第{i}次')
            continue

这类同学基本不会复用代码。代码里面要向N个url发起请求，他们就会在N个地方像上面这样写代码。

新的方法

这里我虽然说是新方法，但是这个方法应该至少在9年前就能用了。只是网上用的人比较少。我们可以使用requests自带的HTTPAdapter来实现自动重试。当我们不关心具体报错是什么，只需要机械重试时，就可以使用这个方法：

1
2
3
4
5
6
7
8
9
10
11
12
import requests
from requests.adapters import HTTPAdapter, Retry

session = requests.Session()
retries = Retry(total=3, backoff_factor=1)
session.mount('http://', HTTPAdapter(max_retries=retries))
session.mount('https://', HTTPAdapter(max_retries=retries))

# 接下来使用session发起的所有请求，默认最多会重试3次
session.get('http://httpbin.org/delay/5', timeout=2)
session.get('https://www.kingname.info')
...