mirror of https://github.com/mitsuhiko/flask.git
Armin Ronacher
14 years ago
2 changed files with 71 additions and 0 deletions
@ -0,0 +1,70 @@ |
|||||||
|
Streaming Contents |
||||||
|
================== |
||||||
|
|
||||||
|
Sometimes you want to send an enormous amount of data to the client, much |
||||||
|
more than you want to keep in memory. When you are generating the data on |
||||||
|
the fly though, how do you send that back to the client without the |
||||||
|
roundtrip to the filesystem? |
||||||
|
|
||||||
|
The answer is by using generators and direct responses. |
||||||
|
|
||||||
|
Basic Usage |
||||||
|
----------- |
||||||
|
|
||||||
|
This is a basic view function that generates a lot of CSV data on the fly. |
||||||
|
The trick is to have an inner function that uses a generator to generate |
||||||
|
data and to then invoke that function and pass it to a response object |
||||||
|
that has the ``direct_passthrough`` flag set. This flag is used to inform |
||||||
|
the system that data is generated on the fly and should be passed through |
||||||
|
without buffering: |
||||||
|
|
||||||
|
.. sourcecode:: python |
||||||
|
|
||||||
|
from flask import Response |
||||||
|
|
||||||
|
@app.route('/large.csv') |
||||||
|
def generate_large_csv(): |
||||||
|
def generate(): |
||||||
|
for row in iter_all_rows(): |
||||||
|
yield ','.join(row) + '\n' |
||||||
|
return Response(generate(), direct_passthrough=True, |
||||||
|
mimetype='text/csv') |
||||||
|
|
||||||
|
Each ``yield`` expression is directly sent to the browser. Now though |
||||||
|
that some WSGI middlewares might break streaming, so be careful there in |
||||||
|
debug environments with profilers and other things you might have enabled. |
||||||
|
|
||||||
|
Streaming from Templates |
||||||
|
------------------------ |
||||||
|
|
||||||
|
The Jinja2 template engine also supports rendering templates piece by |
||||||
|
piece. This functionality is not directly exposed by Flask because it is |
||||||
|
quite uncommon, but you can easily do it yourself: |
||||||
|
|
||||||
|
.. sourcecode:: python |
||||||
|
|
||||||
|
from flask import Response |
||||||
|
|
||||||
|
def stream_template(template_name, **context): |
||||||
|
app.update_template_context(context) |
||||||
|
t = app.jinja_env.get_template(template_name) |
||||||
|
rv = t.stream(context) |
||||||
|
rv.enable_buffering(5) |
||||||
|
return rv |
||||||
|
|
||||||
|
@app.route('/my-large-page.html') |
||||||
|
def render_large_template(): |
||||||
|
rows = iter_all_rows() |
||||||
|
return Response(stream_template('the_template.html', rows=rows), |
||||||
|
direct_passthrough=True) |
||||||
|
|
||||||
|
The trick here is to get the template object from the Jinja2 environment |
||||||
|
on the application and to call :meth:`~jinja2.Template.stream` instead of |
||||||
|
:meth:`~jinja2.Template.render` which returns a stream object instead of a |
||||||
|
string. Since we're bypassing the Flask template render functions and |
||||||
|
using the template object itself we have to make sure to update the render |
||||||
|
context ourselves by calling :meth:`~flask.Flask.update_template_context`. |
||||||
|
The template is then evaluated as the stream is iterated over. Since each |
||||||
|
time you do a yield the server will flush the content to the client you |
||||||
|
might want to buffer up a few items in the template which you can do with |
||||||
|
``rv.enable_buffering(size)``. ``5`` is a sane default. |
Loading…
Reference in new issue