32 lines
785 B
Python
32 lines
785 B
Python
|
from fastapi import Depends, FastAPI, HTTPException
|
||
|
|
from typing_extensions import Annotated
|
||
|
|
|
||
|
|
app = FastAPI()
|
||
|
|
|
||
|
|
|
||
|
|
data = {
|
||
|
|
"plumbus": {"description": "Freshly pickled plumbus", "owner": "Morty"},
|
||
|
|
"portal-gun": {"description": "Gun to create portals", "owner": "Rick"},
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
class OwnerError(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def get_username():
|
||
|
|
try:
|
||
|
|
yield "Rick"
|
||
|
|
except OwnerError as e:
|
||
|
|
raise HTTPException(status_code=400, detail=f"Owner error: {e}")
|
||
|
|
|
||
|
|
|
||
|
|
@app.get("/items/{item_id}")
|
||
|
|
def get_item(item_id: str, username: Annotated[str, Depends(get_username)]):
|
||
|
|
if item_id not in data:
|
||
|
|
raise HTTPException(status_code=404, detail="Item not found")
|
||
|
|
item = data[item_id]
|
||
|
|
if item["owner"] != username:
|
||
|
|
raise OwnerError(username)
|
||
|
|
return item
|