fastapi

1
from typing import Union
2

3
from fastapi import FastAPI
4
from fastapi.responses import FileResponse
5
from pydantic import BaseModel
6

7

8
class Item(BaseModel):
9
    id: str
10
    value: str
11

12

13
responses = {
14
    404: {"description": "Item not found"},
15
    302: {"description": "The item was moved"},
16
    403: {"description": "Not enough privileges"},
17
}
18

19

20
app = FastAPI()
21

22

23
@app.get(
24
    "/items/{item_id}",
25
    response_model=Item,
26
    responses={**responses, 200: {"content": {"image/png": {}}}},
27
)
28
async def read_item(item_id: str, img: Union[bool, None] = None):
29
    if img:
30
        return FileResponse("image.png", media_type="image/png")
31
    else:
32
        return {"id": "foo", "value": "there goes my hero"}
33