forked from orbit-oss/flask
view functions can return generators as responses directly
This commit is contained in:
parent
7f2a0f4806
commit
762382e436
5 changed files with 48 additions and 2 deletions
|
|
@ -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)
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue