58 行
2.0 KiB
Python
58 行
2.0 KiB
Python
# This file is pretty much entirely based on a snippet from asgi.py in
|
|
# the Quart repository (MIT, see README.md). That means it takes on the
|
|
# MIT licence I guess(???) If not then it's the same as every other file
|
|
# by me: 2022 n9k <https://gitler.moe/ninya9k>, AGPL 3.0 or any later
|
|
# version.
|
|
|
|
import asyncio
|
|
|
|
from werkzeug.wrappers import Response as WerkzeugResponse
|
|
from quart.app import Quart as Quart_
|
|
from quart.asgi import ASGIHTTPConnection as ASGIHTTPConnection_
|
|
from quart.utils import encode_headers
|
|
|
|
|
|
RESPONSE_ITERATOR_TIMEOUT = 10.0
|
|
|
|
|
|
class ASGIHTTPConnection(ASGIHTTPConnection_):
|
|
async def _send_response(self, send, response):
|
|
await send({
|
|
"type": "http.response.start",
|
|
"status": response.status_code,
|
|
"headers": encode_headers(response.headers),
|
|
})
|
|
|
|
if isinstance(response, WerkzeugResponse):
|
|
for data in response.response:
|
|
body = data.encode(response.charset) if isinstance(data, str) else data
|
|
await asyncio.wait_for(
|
|
send({
|
|
"type": "http.response.body",
|
|
"body": body,
|
|
"more_body": True,
|
|
}),
|
|
timeout=RESPONSE_ITERATOR_TIMEOUT,
|
|
)
|
|
else:
|
|
async with response.response as response_body:
|
|
async for data in response_body:
|
|
body = data.encode(response.charset) if isinstance(data, str) else data
|
|
await asyncio.wait_for(
|
|
send({
|
|
"type": "http.response.body",
|
|
"body": body,
|
|
"more_body": True,
|
|
}),
|
|
timeout=RESPONSE_ITERATOR_TIMEOUT,
|
|
)
|
|
await send({
|
|
"type": "http.response.body",
|
|
"body": b"",
|
|
"more_body": False,
|
|
})
|
|
|
|
|
|
class Quart(Quart_):
|
|
asgi_http_class = ASGIHTTPConnection
|