33 lines
701 B
Python
33 lines
701 B
Python
|
from typing import Union
|
||
|
|
|
||
|
|
from fastapi import FastAPI
|
||
|
|
from fastapi.responses import FileResponse
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
|
||
|
|
class Item(BaseModel):
|
||
|
|
id: str
|
||
|
|
value: str
|
||
|
|
|
||
|
|
|
||
|
|
responses = {
|
||
|
|
404: {"description": "Item not found"},
|
||
|
|
302: {"description": "The item was moved"},
|
||
|
|
403: {"description": "Not enough privileges"},
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
app = FastAPI()
|
||
|
|
|
||
|
|
|
||
|
|
@app.get(
|
||
|
|
"/items/{item_id}",
|
||
|
|
response_model=Item,
|
||
|
|
responses={**responses, 200: {"content": {"image/png": {}}}},
|
||
|
|
)
|
||
|
|
async def read_item(item_id: str, img: Union[bool, None] = None):
|
||
|
|
if img:
|
||
|
|
return FileResponse("image.png", media_type="image/png")
|
||
|
|
else:
|
||
|
|
return {"id": "foo", "value": "there goes my hero"}
|