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

1import time 7 ctx1abcdefg

2from typing import Callable 7 ctx1abcdefg

3 

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

5from fastapi.routing import APIRoute 7 ctx1abcdefg

6 

7 

8class TimedRoute(APIRoute): 7 ctx1abcdefg

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

10 original_route_handler = super().get_route_handler() 7 ctx1abcdefg

11 

12 async def custom_route_handler(request: Request) -> Response: 7 ctx1abcdefg

13 before = time.time() 7 ctx1hijklmn

14 response: Response = await original_route_handler(request) 7 ctx1hijklmn

15 duration = time.time() - before 7 ctx1hijklmn

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

17 print(f"route duration: {duration}") 7 ctx1hijklmn

18 print(f"route response: {response}") 7 ctx1hijklmn

19 print(f"route response headers: {response.headers}") 7 ctx1hijklmn

20 return response 7 ctx1hijklmn

21 

22 return custom_route_handler 7 ctx1abcdefg

23 

24 

25app = FastAPI() 7 ctx1abcdefg

26router = APIRouter(route_class=TimedRoute) 7 ctx1abcdefg

27 

28 

29@app.get("/") 7 ctx1abcdefg

30async def not_timed(): 7 ctx1abcdefg

31 return {"message": "Not timed"} 7 ctx1opqrstu

32 

33 

34@router.get("/timed") 7 ctx1abcdefg

35async def timed(): 7 ctx1abcdefg

36 return {"message": "It's the time of my life"} 7 ctx1hijklmn

37 

38 

39app.include_router(router) 7 ctx1abcdefg