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

1from fastapi import FastAPI, Request 5 ctx1abcde

2from fastapi.responses import JSONResponse 5 ctx1abcde

3 

4 

5class UnicornException(Exception): 5 ctx1abcde

6 def __init__(self, name: str): 5 ctx1abcde

7 self.name = name 5 ctx1abcde

8 

9 

10app = FastAPI() 5 ctx1abcde

11 

12 

13@app.exception_handler(UnicornException) 5 ctx1abcde

14async def unicorn_exception_handler(request: Request, exc: UnicornException): 5 ctx1abcde

15 return JSONResponse( 5 ctx1abcde

16 status_code=418, 

17 content={"message": f"Oops! {exc.name} did something. There goes a rainbow..."}, 

18 ) 

19 

20 

21@app.get("/unicorns/{name}") 5 ctx1abcde

22async def read_unicorn(name: str): 5 ctx1abcde

23 if name == "yolo": 5 ctx1abcde

24 raise UnicornException(name=name) 5 ctx1abcde

25 return {"unicorn_name": name} 5 ctx1abcde