python - FastAPI - 未填充 ENUM 类型模型

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

创建 Status 类通过从 继承str 和 Enum

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

引用文献

Working with Python enumerations--(FastAPI doc)

[BUG] docs don't show nested enum attribute for body--(Issue #329)