FastAPI

Question

下面是我的fastAPI代码

from typing import Optional, Set

from fastapi import FastAPI
from pydantic import BaseModel, HttpUrl, Field
from enum import Enum

app = FastAPI()


class Status(Enum):
    RECEIVED = 'RECEIVED'
    CREATED = 'CREATED'
    CREATE_ERROR = 'CREATE_ERROR'


class Item(BaseModel):
    name: str
    description: Optional[str] = None
    price: float
    tax: Optional[float] = None
    tags: Set[str] = []
    status: Status = None


@app.put("/items/{item_id}")
async def update_item(item_id: int, item: Item):
    results = {"item_id": item_id, "item": item}
    return results

下面是生成的 swagger 文档。未显示状态。我是 pydantic 的新手，我不确定如何在文档中显示状态

Answer 1

通过继承 str 和 [=20] 创建 Status class =]Enum

class Status(<b>str, Enum</b>):
    RECEIVED = 'RECEIVED'
    CREATED = 'CREATED'
    CREATE_ERROR = 'CREATE_ERROR'

参考资料

FastAPI - 未填充 ENUM 类型模型

FastAPI - ENUM type models not populated

python

swagger

openapi

pydantic