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

1import time 6 ctx1abcdef

2from typing import Callable 6 ctx1abcdef

3 

4from fastapi import APIRouter, FastAPI, Request, Response 6 ctx1abcdef

5from fastapi.routing import APIRoute 6 ctx1abcdef

6 

7 

8class TimedRoute(APIRoute): 6 ctx1abcdef

9 def get_route_handler(self) -> Callable: 6 ctx1abcdef

10 original_route_handler = super().get_route_handler() 6 ctx1abcdef

11 

12 async def custom_route_handler(request: Request) -> Response: 6 ctx1abcdef

13 before = time.time() 6 ctx1ghijkl

14 response: Response = await original_route_handler(request) 6 ctx1ghijkl

15 duration = time.time() - before 6 ctx1ghijkl

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

17 print(f"route duration: {duration}") 6 ctx1ghijkl

18 print(f"route response: {response}") 6 ctx1ghijkl

19 print(f"route response headers: {response.headers}") 6 ctx1ghijkl

20 return response 6 ctx1ghijkl

21 

22 return custom_route_handler 6 ctx1abcdef

23 

24 

25app = FastAPI() 6 ctx1abcdef

26router = APIRouter(route_class=TimedRoute) 6 ctx1abcdef

27 

28 

29@app.get("/") 6 ctx1abcdef

30async def not_timed(): 6 ctx1abcdef

31 return {"message": "Not timed"} 6 ctx1mnopqr

32 

33 

34@router.get("/timed") 6 ctx1abcdef

35async def timed(): 6 ctx1abcdef

36 return {"message": "It's the time of my life"} 6 ctx1ghijkl

37 

38 

39app.include_router(router) 6 ctx1abcdef