view functions can return generators as responses directly

This commit is contained in:
pgjones 2022-06-09 09:30:21 +01:00 committed by David Lord
parent 7f2a0f4806
commit 762382e436
No known key found for this signature in database
GPG key ID: 7A1C87E3F5BC42A8
5 changed files with 48 additions and 2 deletions

View file

@ -1276,6 +1276,11 @@ def test_make_response(app, req_ctx):
assert rv.data == b"W00t"
assert rv.mimetype == "text/html"
rv = flask.make_response(c for c in "Hello")
assert rv.status_code == 200
assert rv.data == b"Hello"
assert rv.mimetype == "text/html"
def test_make_response_with_response_instance(app, req_ctx):
rv = flask.make_response(flask.jsonify({"msg": "W00t"}), 400)

View file

@ -1,9 +1,11 @@
from __future__ import annotations
import typing as t
from http import HTTPStatus
from flask import Flask
from flask import jsonify
from flask import stream_template
from flask.templating import render_template
from flask.views import View
from flask.wrappers import Response
@ -26,6 +28,25 @@ def hello_json() -> Response:
return jsonify({"response": "Hello, World!"})
@app.route("/generator")
def hello_generator() -> t.Generator[str, None, None]:
def show() -> t.Generator[str, None, None]:
for x in range(100):
yield f"data:{x}\n\n"
return show()
@app.route("/generator-expression")
def hello_generator_expression() -> t.Iterator[bytes]:
return (f"data:{x}\n\n".encode() for x in range(100))
@app.route("/iterator")
def hello_iterator() -> t.Iterator[str]:
return iter([f"data:{x}\n\n" for x in range(100)])
@app.route("/status")
@app.route("/status/<int:code>")
def tuple_status(code: int = 200) -> tuple[str, int]:
@ -48,6 +69,11 @@ def return_template(name: str | None = None) -> str:
return render_template("index.html", name=name)
@app.route("/template")
def return_template_stream() -> t.Iterator[str]:
return stream_template("index.html", name="Hello")
class RenderTemplateView(View):
def __init__(self: RenderTemplateView, template_name: str) -> None:
self.template_name = template_name