如何将 iter() 应用于分页 API？

Question

我观看了 Raymond Hettinger 的 Idiomatic Python 谈话，并了解了 sentinel argument 至 iter()。 我想尝试将其应用于我正在迭代使用分页的 API 的一段代码（它是 Twilio，但与我的问题无关）。

我有一个返回的 API：数据列表和下一页 URL。当分页用尽时，下一页 URL 作为空字符串返回。我将获取函数编写为生成器，大致如下：

def fetch(url):
    while url:
        data = requests.get(url).json()
        url = data['next_page_uri']
        for row in data[resource]:
            yield row

此代码工作正常，但我想尝试删除 while 循环并将其替换为使用 next_page_uri 值作为标记参数的对 iter() 的调用。 或者，这可以写成yield from吗？

Answer 1

我认为这可能是您的意思……但正如 cmets 中所述，它没有多大帮助：

def fetch_paged(url):
    while url:
        res = requests.get(url)
        res.raise_for_status()
        data = res.json()
        yield data
        url = data['next_page_uri']

def fetch(url):
    for data in fetch_paged(url):
        yield from data[resource]

（我借此机会拨打了raise_for_status() 的电话，如果不成功，我将拨打raise，即res.status_code < 400，回复）

不确定它是否“更好”，但如果您要大量重用 fetch_paged 功能，则可能

注意：许多其他 API 将此 next_page_uri 放入 standard ways 中的响应标头中，requests 库知道如何处理并通过 res.links 属性公开