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

1import gzip 7 ctx1abcdefg

2from collections.abc import Callable 7 ctx1abcdefg

3 

4from fastapi import Body, FastAPI, Request, Response 7 ctx1abcdefg

5from fastapi.routing import APIRoute 7 ctx1abcdefg

6 

7 

8class GzipRequest(Request): 7 ctx1abcdefg

9 async def body(self) -> bytes: 7 ctx1abcdefg

10 if not hasattr(self, "_body"): 7 ctx1lhimjnk

11 body = await super().body() 7 ctx1lhimjnk

12 if "gzip" in self.headers.getlist("Content-Encoding"): 7 ctx1lhimjnk

13 body = gzip.decompress(body) 4 ctx1hijk

14 self._body = body 7 ctx1lhimjnk

15 return self._body 7 ctx1lhimjnk

16 

17 

18class GzipRoute(APIRoute): 7 ctx1abcdefg

19 def get_route_handler(self) -> Callable: 7 ctx1abcdefg

20 original_route_handler = super().get_route_handler() 11 ctx1abocdefpqgr

21 

22 async def custom_route_handler(request: Request) -> Response: 11 ctx1abocdefpqgr

23 request = GzipRequest(request.scope, request.receive) 11 ctx1lhsitmjunkv

24 return await original_route_handler(request) 11 ctx1lhsitmjunkv

25 

26 return custom_route_handler 11 ctx1abocdefpqgr

27 

28 

29app = FastAPI() 7 ctx1abcdefg

30app.router.route_class = GzipRoute 7 ctx1abcdefg

31 

32 

33@app.post("/sum") 7 ctx1abcdefg

34async def sum_numbers(numbers: list[int] = Body()): 7 ctx1abcdefg

35 return {"sum": sum(numbers)} 7 ctx1lhimjnk