mirror of
https://github.com/fastapi/fastapi.git
synced 2026-02-26 03:36:14 -05:00
7250c194da
Co-authored-by: Sebastián Ramírez <tiangolo@gmail.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
from typing import Any
|
|
|
|
from starlette.responses import FileResponse as FileResponse # noqa
|
|
from starlette.responses import HTMLResponse as HTMLResponse # noqa
|
|
from starlette.responses import JSONResponse as JSONResponse # noqa
|
|
from starlette.responses import PlainTextResponse as PlainTextResponse # noqa
|
|
from starlette.responses import RedirectResponse as RedirectResponse # noqa
|
|
from starlette.responses import Response as Response # noqa
|
|
from starlette.responses import StreamingResponse as StreamingResponse # noqa
|
|
|
|
try:
|
|
import ujson
|
|
except ImportError: # pragma: nocover
|
|
ujson = None # type: ignore
|
|
|
|
|
|
try:
|
|
import orjson
|
|
except ImportError: # pragma: nocover
|
|
orjson = None # type: ignore
|
|
|
|
|
|
class UJSONResponse(JSONResponse):
|
|
def render(self, content: Any) -> bytes:
|
|
assert ujson is not None, "ujson must be installed to use UJSONResponse"
|
|
return ujson.dumps(content, ensure_ascii=False).encode("utf-8")
|
|
|
|
|
|
class ORJSONResponse(JSONResponse):
|
|
media_type = "application/json"
|
|
|
|
def render(self, content: Any) -> bytes:
|
|
assert orjson is not None, "orjson must be installed to use ORJSONResponse"
|
|
return orjson.dumps(
|
|
content, option=orjson.OPT_NON_STR_KEYS | orjson.OPT_SERIALIZE_NUMPY
|
|
)
|