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
app = FastAPI()
14

15

16
@app.get(
17
    "/items/{item_id}",
18
    response_model=Item,
19
    responses={
20
        200: {
21
            "content": {"image/png": {}},
22
            "description": "Return the JSON item or an image.",
23
        }
24
    },
25
)
26
async def read_item(item_id: str, img: Union[bool, None] = None):
27
    if img:
28
        return FileResponse("image.png", media_type="image/png")
29
    else:
30
        return {"id": "foo", "value": "there goes my hero"}
31