Coverage for docs_src/custom_request_and_route/tutorial001.py: 100%

1import gzip 6 ctx1abcdef

2from typing import Callable, List 6 ctx1abcdef

3 

4from fastapi import Body, FastAPI, Request, Response 6 ctx1abcdef

5from fastapi.routing import APIRoute 6 ctx1abcdef

6 

7 

8class GzipRequest(Request): 6 ctx1abcdef

9 async def body(self) -> bytes: 6 ctx1abcdef

10 if not hasattr(self, "_body"): 6 ctx1ghijkl

11 body = await super().body() 6 ctx1ghijkl

12 if "gzip" in self.headers.getlist("Content-Encoding"): 6 ctx1ghijkl

13 body = gzip.decompress(body) 6 ctx1ghijkl

14 self._body = body 6 ctx1ghijkl

15 return self._body 6 ctx1ghijkl

16 

17 

18class GzipRoute(APIRoute): 6 ctx1abcdef

19 def get_route_handler(self) -> Callable: 6 ctx1abcdef

20 original_route_handler = super().get_route_handler() 6 ctx1abcdef

21 

22 async def custom_route_handler(request: Request) -> Response: 6 ctx1abcdef

23 request = GzipRequest(request.scope, request.receive) 12 ctx1gmhniojpkqlr

24 return await original_route_handler(request) 12 ctx1gmhniojpkqlr

25 

26 return custom_route_handler 6 ctx1abcdef

27 

28 

29app = FastAPI() 6 ctx1abcdef

30app.router.route_class = GzipRoute 6 ctx1abcdef

31 

32 

33@app.post("/sum") 6 ctx1abcdef

34async def sum_numbers(numbers: List[int] = Body()): 6 ctx1abcdef

35 return {"sum": sum(numbers)} 6 ctx1ghijkl