Accept AsyncIterables being passed to Response

Fixes pallets/flask#5322

Author: mjsir911

src/quart/typing.py

@@ -9,6 +9,7 @@
     AnyStr,
     AsyncContextManager,
     AsyncGenerator,
+    AsyncIterator,
     Awaitable,
     Callable,
     Dict,

src/quart/wrappers/response.py

@@ -102,19 +102,28 @@ async def __anext__(self) -> bytes:
 
 
 class IterableBody(ResponseBody):
-    def __init__(self, iterable: AsyncGenerator[bytes, None] | Iterable) -> None:
+    def __init__(self, iterable: AsyncIterable[bytes] | Iterable) -> None:
         self.iter: AsyncGenerator[bytes, None]
         if isasyncgen(iterable):
             self.iter = iterable
         elif isgenerator(iterable):
             self.iter = run_sync_iterable(iterable)
-        else:
+        elif isinstance(iterable, AsyncIterable):
 
             async def _aiter() -> AsyncGenerator[bytes, None]:
-                for data in iterable:  # type: ignore
+                async for data in iterable:
                     yield data
 
             self.iter = _aiter()
+        elif isinstance(iterable, Iterable):
+
+            async def _aiter() -> AsyncGenerator[bytes, None]:
+                for data in iterable:
+                    yield data
+
+            self.iter = _aiter()
+        else:
+            raise ValueError("unreachable?")
 
     async def __aenter__(self) -> IterableBody:
         return self
@@ -262,7 +271,7 @@ class Response(SansIOResponse):
 
     def __init__(
         self,
-        response: ResponseBody | AnyStr | Iterable | None = None,
+        response: ResponseBody | AnyStr | Iterable | AsyncIterable | None = None,
         status: int | None = None,
         headers: dict | Headers | None = None,
         mimetype: str | None = None,

tests/test_templating.py

@@ -9,6 +9,7 @@
     g,
     Quart,
     render_template_string,
+    Response,
     ResponseReturnValue,
     session,
     stream_template_string,
@@ -148,3 +149,11 @@ async def index() -> ResponseReturnValue:
     test_client = app.test_client()
     response = await test_client.get("/")
     assert (await response.data) == b"42"
+
+    @app.get("/2")
+    async def index2() -> ResponseReturnValue:
+        return Response(await stream_template_string("{{ config }}", config=43))
+
+    test_client = app.test_client()
+    response = await test_client.get("/2")
+    assert (await response.data) == b"43"