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

1import gzip 3 ctx1abc

2from collections.abc import Callable 3 ctx1abc

3from typing import Annotated 3 ctx1abc

4 

5from fastapi import Body, FastAPI, Request, Response 3 ctx1abc

6from fastapi.routing import APIRoute 3 ctx1abc

7 

8 

9class GzipRequest(Request): 3 ctx1abc

10 async def body(self) -> bytes: 3 ctx1abc

11 if not hasattr(self, "_body"): 3 ctx1def

12 body = await super().body() 3 ctx1def

13 if "gzip" in self.headers.getlist("Content-Encoding"): 3 ctx1def

14 body = gzip.decompress(body) 3 ctx1def

15 self._body = body 3 ctx1def

16 return self._body 3 ctx1def

17 

18 

19class GzipRoute(APIRoute): 3 ctx1abc

20 def get_route_handler(self) -> Callable: 3 ctx1abc

21 original_route_handler = super().get_route_handler() 3 ctx1abc

22 

23 async def custom_route_handler(request: Request) -> Response: 3 ctx1abc

24 request = GzipRequest(request.scope, request.receive) 6 ctx1dgehfi

25 return await original_route_handler(request) 6 ctx1dgehfi

26 

27 return custom_route_handler 3 ctx1abc

28 

29 

30app = FastAPI() 3 ctx1abc

31app.router.route_class = GzipRoute 3 ctx1abc

32 

33 

34@app.post("/sum") 3 ctx1abc

35async def sum_numbers(numbers: Annotated[list[int], Body()]): 3 ctx1abc

36 return {"sum": sum(numbers)} 3 ctx1def