Flask——基于python完整实现客户端和服务器后端流式请求及响应

文章目录

看了很多相关博客,但是都没有本地客户端和服务器后端的完整代码示例,有的也只说了如何流式获取后端结果,基本没有讲两端如何同时实现流式输入输出,特此整理总结,给大家交流学习和使用!

本地客户端

  • requests.post得到流式回复的重要参数:
    • stream:需要设置为True;
    • response.iter_content:使用该函数接收返回的流式数据。
python 复制代码
import requests
import time
import json

def generate_stream_data():
    # 假设这是要发送的文本列表
    is_end = False
    lines = ["Hello", "world", "this", "is", "a", "stream", "of", "text"]
    for line in lines:
        print(line)
        if lines.index(line) == len(lines) - 1:
            is_end = True
        yield json.dumps({'line': line, 'is_end': is_end}) + '\n'
        time.sleep(0.5)
        # 模拟数据处理时间

def get_stream_response(response):
    # 流式接收response
    rec_data_list = []
    temp_data = ''
    for chunk in response.iter_content(chunk_size=1):
        temp_data += chunk.decode('utf-8')
        if temp_data.endswith('\n'):
            temp_json = json.loads(temp_data)
            rec_data_list.append(temp_json)
            print(temp_data)
            temp_data = ''
            if temp_json['is_end']:
                break
    print(rec_data_list)
    print("----------------------------")
    print(temp_data)
    return rec_data_list

def stream_upload(url):
    
    # 流式接收response
    response = requests.post(url, data=generate_stream_data(), stream=True)
    
    final_response = get_stream_response(response)
    
    return final_response

url = 'http://127.0.0.1:5000/stream'
response = stream_upload(url)

Flask服务器后端

  • flask.request流式获取数据::
    • 使用request.stream.read读取数据,而不是get_data()等一次性函数。
python 复制代码
from flask import Flask, Response, request
import time
import json
import requests

app = Flask(__name__)

def process_stream_data(stream_data):
    # 假设这是要发送的数据
    print("开始生成新的数据流")
    is_end = False
    print(stream_data)
    for idx, line in enumerate(stream_data):
        if idx == len(stream_data)-1:
            is_end = True
        print(line)
        yield json.dumps(line)+"\n"
        time.sleep(0.5)
        # 模拟数据处理时间

def get_stream_request(chunk_size=1):
    req_data_list = []
    temp_data = ''
    while True:
        chunk = request.stream.read(chunk_size)
        temp_data += chunk.decode('utf-8')
        if temp_data.endswith('\n'):
            temp_json = json.loads(temp_data)
            req_data_list.append(temp_json)
            print(temp_data)
            temp_data = ''
            if temp_json['is_end']:
                return req_data_list

@app.route('/stream', methods=['POST'])
def stream_text():
    
    data = get_stream_request()

    print("----------------------------")
    
    return Response(process_stream_data(data))

if __name__ == "__main__":
    app.run(host='0.0.0.0', port=5000, debug=True)

客户端/服务器端流式接收[打字机]效果

相关推荐
Shartin22 分钟前
Can201-Introduction to Networking: Application Layer应用层
服务器·开发语言·php
cooldream20091 小时前
Python 包管理新时代:深入了解 `uv` 的使用与实践
python·uv·包管理器
之歆1 小时前
Python-魔术方法-创建、初始化与销毁-hash-bool-可视化-运算符重载-容器和大小-可调用对象-上下文管理-反射-描述器-二分-学习笔记
笔记·python·学习
胖达不服输1 小时前
「日拱一码」025 机器学习——评价指标
人工智能·python·机器学习·评价指标
apihz1 小时前
VM虚拟机全版本网盘+免费本地网络穿透端口映射实时同步动态家庭IP教程
android·服务器·开发语言·网络·数据库·网络协议·tcp/ip
乌云暮年2 小时前
Linux常用命令
linux·服务器·ssh·php
LCG元2 小时前
MCP选型指南:AWS vs Azure vs GCP vs 国内云厂商深度对比
flask·azure·aws
wanhengidc2 小时前
企业选择大带宽服务器租用的原因有哪些?
运维·服务器
brave_zhao2 小时前
JavaBeanUtils javaBean转map, 实体类转map,实体集合转List<Map>
linux·windows·python
egoist20233 小时前
【Linux仓库】虚拟地址空间【进程·陆】
linux·运维·服务器·操作系统·进程·虚拟地址空间·fork