前言
在进行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。