mirror of https://github.com/tiangolo/fastapi.git
32 lines
686 B
Python
32 lines
686 B
Python
from fastapi import APIRouter, FastAPI
|
|
from fastapi.responses import JSONResponse
|
|
|
|
|
|
class UnicornException(Exception):
|
|
def __init__(self, name: str):
|
|
self.name = name
|
|
|
|
|
|
router = APIRouter(
|
|
prefix="/unicorns",
|
|
exception_handlers={
|
|
UnicornException: lambda request, exc: JSONResponse(
|
|
status_code=418,
|
|
content={
|
|
"message": f"Oops! {exc.name} did something. There goes a rainbow..."
|
|
},
|
|
)
|
|
},
|
|
)
|
|
|
|
|
|
@router.get("/{name}")
|
|
async def read_unicorn(name: str):
|
|
if name == "yolo":
|
|
raise UnicornException(name=name)
|
|
return {"unicorn_name": name}
|
|
|
|
|
|
app = FastAPI()
|
|
app.include_router(router)
|