30 lines
744 B
Python
30 lines
744 B
Python
|
from fastapi import Depends, FastAPI, HTTPException
|
||
|
|
from typing_extensions import Annotated
|
||
|
|
|
||
|
|
app = FastAPI()
|
||
|
|
|
||
|
|
|
||
|
|
class InternalError(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def get_username():
|
||
|
|
try:
|
||
|
|
yield "Rick"
|
||
|
|
except InternalError:
|
||
|
|
print("We don't swallow the internal error here, we raise again 😎")
|
||
|
|
raise
|
||
|
|
|
||
|
|
|
||
|
|
@app.get("/items/{item_id}")
|
||
|
|
def get_item(item_id: str, username: Annotated[str, Depends(get_username)]):
|
||
|
|
if item_id == "portal-gun":
|
||
|
|
raise InternalError(
|
||
|
|
f"The portal gun is too dangerous to be owned by {username}"
|
||
|
|
)
|
||
|
|
if item_id != "plumbus":
|
||
|
|
raise HTTPException(
|
||
|
|
status_code=404, detail="Item not found, there's only a plumbus here"
|
||
|
|
)
|
||
|
|
return item_id
|