902133edd3
- 8 DB models (services, incidents, monitors, subscribers, etc.) - Full CRUD API for services, incidents, monitors - Public status page with live data - Incident detail page with timeline - API key authentication - Uptime monitoring scheduler - 13 tests passing - TECHNICAL_DESIGN.md with full spec
98 lines
4.1 KiB
Python
98 lines
4.1 KiB
Python
import importlib
|
|
from typing import Any, Protocol, cast
|
|
|
|
from fastapi.exceptions import FastAPIDeprecationWarning
|
|
from fastapi.sse import EventSourceResponse as EventSourceResponse # noqa
|
|
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
|
|
from typing_extensions import deprecated
|
|
|
|
|
|
class _UjsonModule(Protocol):
|
|
def dumps(self, __obj: Any, *, ensure_ascii: bool = ...) -> str: ...
|
|
|
|
|
|
class _OrjsonModule(Protocol):
|
|
OPT_NON_STR_KEYS: int
|
|
OPT_SERIALIZE_NUMPY: int
|
|
|
|
def dumps(self, __obj: Any, *, option: int = ...) -> bytes: ...
|
|
|
|
|
|
try:
|
|
ujson = cast(_UjsonModule, importlib.import_module("ujson"))
|
|
except ModuleNotFoundError: # pragma: nocover
|
|
ujson = None # type: ignore # ty: ignore[unused-ignore-comment]
|
|
|
|
|
|
try:
|
|
orjson = cast(_OrjsonModule, importlib.import_module("orjson"))
|
|
except ModuleNotFoundError: # pragma: nocover
|
|
orjson = None # type: ignore # ty: ignore[unused-ignore-comment]
|
|
|
|
|
|
@deprecated(
|
|
"UJSONResponse is deprecated, FastAPI now serializes data directly to JSON "
|
|
"bytes via Pydantic when a return type or response model is set, which is "
|
|
"faster and doesn't need a custom response class. Read more in the FastAPI "
|
|
"docs: https://fastapi.tiangolo.com/advanced/custom-response/#orjson-or-response-model "
|
|
"and https://fastapi.tiangolo.com/tutorial/response-model/",
|
|
category=FastAPIDeprecationWarning,
|
|
stacklevel=2,
|
|
)
|
|
class UJSONResponse(JSONResponse):
|
|
"""JSON response using the ujson library to serialize data to JSON.
|
|
|
|
**Deprecated**: `UJSONResponse` is deprecated. FastAPI now serializes data
|
|
directly to JSON bytes via Pydantic when a return type or response model is
|
|
set, which is faster and doesn't need a custom response class.
|
|
|
|
Read more in the
|
|
[FastAPI docs for Custom Response](https://fastapi.tiangolo.com/advanced/custom-response/#orjson-or-response-model)
|
|
and the
|
|
[FastAPI docs for Response Model](https://fastapi.tiangolo.com/tutorial/response-model/).
|
|
|
|
**Note**: `ujson` is not included with FastAPI and must be installed
|
|
separately, e.g. `pip install ujson`.
|
|
"""
|
|
|
|
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")
|
|
|
|
|
|
@deprecated(
|
|
"ORJSONResponse is deprecated, FastAPI now serializes data directly to JSON "
|
|
"bytes via Pydantic when a return type or response model is set, which is "
|
|
"faster and doesn't need a custom response class. Read more in the FastAPI "
|
|
"docs: https://fastapi.tiangolo.com/advanced/custom-response/#orjson-or-response-model "
|
|
"and https://fastapi.tiangolo.com/tutorial/response-model/",
|
|
category=FastAPIDeprecationWarning,
|
|
stacklevel=2,
|
|
)
|
|
class ORJSONResponse(JSONResponse):
|
|
"""JSON response using the orjson library to serialize data to JSON.
|
|
|
|
**Deprecated**: `ORJSONResponse` is deprecated. FastAPI now serializes data
|
|
directly to JSON bytes via Pydantic when a return type or response model is
|
|
set, which is faster and doesn't need a custom response class.
|
|
|
|
Read more in the
|
|
[FastAPI docs for Custom Response](https://fastapi.tiangolo.com/advanced/custom-response/#orjson-or-response-model)
|
|
and the
|
|
[FastAPI docs for Response Model](https://fastapi.tiangolo.com/tutorial/response-model/).
|
|
|
|
**Note**: `orjson` is not included with FastAPI and must be installed
|
|
separately, e.g. `pip install orjson`.
|
|
"""
|
|
|
|
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
|
|
)
|