Coverage for fastapi/responses.py: 100%

1from typing import Any 6 ctx1abcdef

2 

3from starlette.responses import FileResponse as FileResponse # noqa 6 ctx1abcdef

4from starlette.responses import HTMLResponse as HTMLResponse # noqa 6 ctx1abcdef

5from starlette.responses import JSONResponse as JSONResponse # noqa 6 ctx1abcdef

6from starlette.responses import PlainTextResponse as PlainTextResponse # noqa 6 ctx1abcdef

7from starlette.responses import RedirectResponse as RedirectResponse # noqa 6 ctx1abcdef

8from starlette.responses import Response as Response # noqa 6 ctx1abcdef

9from starlette.responses import StreamingResponse as StreamingResponse # noqa 6 ctx1abcdef

10 

11try: 6 ctx1abcdef

12 import ujson 6 ctx1abcdef

13except ImportError: # pragma: nocover 

14 ujson = None # type: ignore 

15 

16 

17try: 6 ctx1abcdef

18 import orjson 6 ctx1abcdef

19except ImportError: # pragma: nocover 

20 orjson = None # type: ignore 

21 

22 

23class UJSONResponse(JSONResponse): 6 ctx1abcdef

24 """ 

25 JSON response using the high-performance ujson library to serialize data to JSON. 

26 

27 Read more about it in the 

28 [FastAPI docs for Custom Response - HTML, Stream, File, others](https://fastapi.tiangolo.com/advanced/custom-response/). 

29 """ 

30 

31 def render(self, content: Any) -> bytes: 6 ctx1abcdef

32 assert ujson is not None, "ujson must be installed to use UJSONResponse" 6 ctx1ghijkl

33 return ujson.dumps(content, ensure_ascii=False).encode("utf-8") 6 ctx1ghijkl

34 

35 

36class ORJSONResponse(JSONResponse): 6 ctx1abcdef

37 """ 

38 JSON response using the high-performance orjson library to serialize data to JSON. 

39 

40 Read more about it in the 

41 [FastAPI docs for Custom Response - HTML, Stream, File, others](https://fastapi.tiangolo.com/advanced/custom-response/). 

42 """ 

43 

44 def render(self, content: Any) -> bytes: 6 ctx1abcdef

45 assert orjson is not None, "orjson must be installed to use ORJSONResponse" 12 ctx1mnopqrstuvwx

46 return orjson.dumps( 12 ctx1mnopqrstuvwx

47 content, option=orjson.OPT_NON_STR_KEYS | orjson.OPT_SERIALIZE_NUMPY 

48 )