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

1import time 3 ctx1abc

2from collections.abc import Callable 3 ctx1abc

3 

4from fastapi import APIRouter, FastAPI, Request, Response 3 ctx1abc

5from fastapi.routing import APIRoute 3 ctx1abc

6 

7 

8class TimedRoute(APIRoute): 3 ctx1abc

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

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

11 

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

13 before = time.time() 3 ctx1def

14 response: Response = await original_route_handler(request) 3 ctx1def

15 duration = time.time() - before 3 ctx1def

16 response.headers["X-Response-Time"] = str(duration) 3 ctx1def

17 print(f"route duration: {duration}") 3 ctx1def

18 print(f"route response: {response}") 3 ctx1def

19 print(f"route response headers: {response.headers}") 3 ctx1def

20 return response 3 ctx1def

21 

22 return custom_route_handler 3 ctx1abc

23 

24 

25app = FastAPI() 3 ctx1abc

26router = APIRouter(route_class=TimedRoute) 3 ctx1abc

27 

28 

29@app.get("/") 3 ctx1abc

30async def not_timed(): 3 ctx1abc

31 return {"message": "Not timed"} 3 ctx1ghi

32 

33 

34@router.get("/timed") 3 ctx1abc

35async def timed(): 3 ctx1abc

36 return {"message": "It's the time of my life"} 3 ctx1def

37 

38 

39app.include_router(router) 3 ctx1abc