Flask：将流作为响应发送

在本文中，我们将介绍如何使用Flask框架将流作为响应发送给客户端。发送流作为响应的场景很多，比如实时音视频流的播放、文件下载或者实时日志查看等。Flask提供了灵活的方式来处理这些需求，并且能够高效地进行流式数据的传输。接下来的内容将详细介绍如何在Flask应用中发送流作为响应。

阅读更多：Flask 教程

创建Flask应用

首先，我们需要创建一个基本的Flask应用。以下是一个简单的示例代码：

from flask import Flask
app = Flask(__name__)

@app.route('/')
def hello_world():
    return 'Hello, World!'

if __name__ == '__main__':
    app.run()

在这个示例中，我们创建了一个名为app的Flask应用，并定义了一个路由/，当访问这个路由时，返回字符串”Hello, World!”。

将流作为响应发送

接下来，我们将介绍如何将流作为响应发送给客户端。Flask提供了stream_with_context()函数来生成一个特殊的响应对象，该对象可以逐行传输数据。我们可以利用这个功能来处理大文件或者实时生成的数据。

以下是一个将文件作为响应发送的示例代码：

from flask import Flask, Response, stream_with_context

app = Flask(__name__)

@app.route('/')
def stream_file():
    def generate():
        with open('example.txt', 'r') as f:
            for line in f:
                yield line

    return Response(stream_with_context(generate()))

if __name__ == '__main__':
    app.run()

在这个示例中，我们定义了一个名为stream_file的路由，该路由对应的处理函数中打开了一个文件并逐行生成文件内容。yield关键字用于将每行数据逐一传输给客户端。

在Flask的响应中，我们使用Response对象来装饰生成器函数，这样生成器函数中的数据将被逐个传输给客户端。

中途终止流响应

有时候，我们希望提供一种机制，让客户端能够中途终止对流响应的请求。Flask提供了should_send_flask函数来判断是否继续发送响应。以下是一个示例代码：

from flask import Flask, Response, stream_with_context, request

app = Flask(__name__)

@app.route('/stream')
def stream_data():
    def generate():
        for i in range(10):
            if request.args.get('stop') == 'true':
                break
            yield f'Data: {i}\n'

    return Response(stream_with_context(generate()))

if __name__ == '__main__':
    app.run()