Coverage for fastapi/responses.py: 100%

1from typing import Any 7 ctx1abcdefg

2 

3from starlette.responses import FileResponse as FileResponse # noqa 7 ctx1abcdefg

4from starlette.responses import HTMLResponse as HTMLResponse # noqa 7 ctx1abcdefg

5from starlette.responses import JSONResponse as JSONResponse # noqa 7 ctx1abcdefg

6from starlette.responses import PlainTextResponse as PlainTextResponse # noqa 7 ctx1abcdefg

7from starlette.responses import RedirectResponse as RedirectResponse # noqa 7 ctx1abcdefg

8from starlette.responses import Response as Response # noqa 7 ctx1abcdefg

9from starlette.responses import StreamingResponse as StreamingResponse # noqa 7 ctx1abcdefg

10 

11try: 7 ctx1abcdefg

12 import ujson 7 ctx1abcdefg

13except ImportError: # pragma: nocover 

14 ujson = None # type: ignore 

15 

16 

17try: 7 ctx1abcdefg

18 import orjson 7 ctx1abcdefg

19except ImportError: # pragma: nocover 

20 orjson = None # type: ignore 

21 

22 

23class UJSONResponse(JSONResponse): 7 ctx1abcdefg

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: 7 ctx1abcdefg

32 assert ujson is not None, "ujson must be installed to use UJSONResponse" 7 ctx1hijklmn

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

34 

35 

36class ORJSONResponse(JSONResponse): 7 ctx1abcdefg

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: 7 ctx1abcdefg

45 assert orjson is not None, "orjson must be installed to use ORJSONResponse" 14 ctx1opqrstuvwxyzAB

46 return orjson.dumps( 14 ctx1opqrstuvwxyzAB

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

48 )