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

1import time 5 ctx1abcde

2from typing import Callable 5 ctx1abcde

3 

4from fastapi import APIRouter, FastAPI, Request, Response 5 ctx1abcde

5from fastapi.routing import APIRoute 5 ctx1abcde

6 

7 

8class TimedRoute(APIRoute): 5 ctx1abcde

9 def get_route_handler(self) -> Callable: 5 ctx1abcde

10 original_route_handler = super().get_route_handler() 5 ctx1abcde

11 

12 async def custom_route_handler(request: Request) -> Response: 5 ctx1abcde

13 before = time.time() 5 ctx1abcde

14 response: Response = await original_route_handler(request) 5 ctx1abcde

15 duration = time.time() - before 5 ctx1abcde

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

17 print(f"route duration: {duration}") 5 ctx1abcde

18 print(f"route response: {response}") 5 ctx1abcde

19 print(f"route response headers: {response.headers}") 5 ctx1abcde

20 return response 5 ctx1abcde

21 

22 return custom_route_handler 5 ctx1abcde

23 

24 

25app = FastAPI() 5 ctx1abcde

26router = APIRouter(route_class=TimedRoute) 5 ctx1abcde

27 

28 

29@app.get("/") 5 ctx1abcde

30async def not_timed(): 5 ctx1abcde

31 return {"message": "Not timed"} 5 ctx1abcde

32 

33 

34@router.get("/timed") 5 ctx1abcde

35async def timed(): 5 ctx1abcde

36 return {"message": "It's the time of my life"} 5 ctx1abcde

37 

38 

39app.include_router(router) 5 ctx1abcde