Python实现API接口内容实时输出(逐块显示、打字机效果)

前言

在进行API接口访问时,经常会遇到响应数据过大导致传输阻塞和超时等问题。为了解决这些问题,可以考虑对API响应数据进行分块显示、流式输出和打字机效果处理,从而避免一次性读取和输出过多数据,提高用户体验和程序性能。

代码

在Python中,我们可以利用requests库的iter_content方法实现API接口数据分块读取和逐步输出。

import time
import requests

response = requests.post("http://www.bhshare.cn/", stream=True)
# 遍历所有的数据块并打印出来
if (response.status_code == 200):
    pre = b''
    for chunk in response.iter_content(chunk_size=2):
        if chunk:
            pre += chunk
            try:
                print(pre.decode('utf-8').replace("\n", "\\n"), end='', flush=True)
                pre = b''
                time.sleep(0.1)  # 仅仅是为了更加直观的展示“打字机”效果,实际使用时应删除
            except UnicodeDecodeError:
                # 当前chunk数据不完整无法被正确解码时会报UnicodeDecodeError异常,临时存储起来与下一个chunk合并解码
                pass

在这里插入图片描述
通过设置stream=True,我们可以获得response对象的流式响应数据。然后使用for循环和iter_content方法对响应数据进行分块读取和输出,可以通过chunk_size参数控制每次读取的数据块大小。


PS:对于接口响应数据较大,时间较长的情况,按块输出文本则自带 “打字机” 效果,比如 ChatGPT。

猜你喜欢

转载自blog.csdn.net/qq_40738764/article/details/130667585
今日推荐