fastapi response_model typing泛型实现统一json格式，data可以为单体也可以为list

你可以使用Union类型来实现这个功能。Union类型允许你定义多个类型中的任意一个作为属性的类型。对于你的情况，可以使用Union类型将单体和list类型的数据都包含在内：

from typing import List, Union

from fastapi import FastAPI

app = FastAPI()

class ItemBase:
    name: str
    description: str = None

class Item(ItemBase):
    id: int

class ItemCreate(ItemBase):
    pass

@app.post("/items/", response_model=Union[Item, List[Item]])
async def create_item(item: ItemCreate):
    return [Item(id=1, **item.dict()), Item(id=2, **item.dict())]

在上面的代码中，我们在response_model参数中定义了Union[Item, List[Item]]类型，这意味着返回的数据可以是一个单体Item对象或者是一个Item对象的列表。在这个示例中，我们返回了一个包含两个Item对象的列表。

相关问题

fastapi response_model typing泛型实现统一json格式，类似Java的Result规定统一返回

你可以使用FastAPI中的Response类来实现统一的JSON格式返回。Response类允许你定义返回的数据类型以及HTTP状态码，这样你就可以在整个应用程序中使用统一的返回格式。

下面是一个示例，展示如何使用Response类和自定义的Result类来实现统一的JSON格式返回：

from fastapi import FastAPI, Response
from pydantic import BaseModel
from typing import Generic, TypeVar, List

app = FastAPI()

T = TypeVar('T')

class Result(Generic[T]):
    success: bool
    data: T
    error: str = None

class Item(BaseModel):
    name: str
    description: str = None

@app.post("/items/", response_model=Result[List[Item]])
async def create_item(item: Item, response: Response):
    items = [Item(name=item.name, description=item.description), Item(name=item.name, description=item.description)]
    result = Result(success=True, data=items)
    response.status_code = 200 if result.success else 400
    return result

在上面的代码中，我们定义了一个自定义的Result类，用于统一的JSON格式返回。在create_item路由函数中，我们将response_model参数设置为Result[List[Item]]，这意味着返回的数据是一个Result对象，其中data属性是一个Item对象的列表。在函数中，我们创建了一个包含两个Item对象的列表，并将其赋值给Result对象的data属性。然后，我们将Result对象作为返回值返回，并将HTTP状态码也设置为200或400，具体取决于Result对象的success属性。

这样，你就可以在整个应用程序中使用统一的JSON格式返回了。

fastapi response_model Generic自定义返回JSON数组

如果你想要自定义 FastAPI 的 response_model 参数来返回一个 JSON 数组，你可以使用 List[Response[Dict[str, Any]]] 来实现。这里的 Dict[str, Any] 表示返回的 JSON 对象可以包含任意的键值对，具体的键和值可以根据你的需要进行定义，示例代码如下：

from typing import Generic, List, TypeVar, Dict, Any
from fastapi import FastAPI
from pydantic import BaseModel

app = FastAPI()

T = TypeVar('T')
class Response(Generic[T]):
    data: T
    code: int = 0
    msg: str = "success"

class User(BaseModel):
    name: str
    age: int

@app.get("/users/", response_model=List[Response[Dict[str, Any]]])
async def read_users():
    users = [{"name": "Alice", "age": 20}, {"name": "Bob", "age": 25}]
    response = [Response[Dict[str, Any]](data=user_dict) for user_dict in users]
    return response

在这个示例代码中，我们使用了一个类似于之前的 Response 泛型类，但是 data 属性的类型是 Dict[str, Any]，这样就可以包含任意的键值对。在 read_users 函数中，我们使用了 List[Response[Dict[str, Any]]] 作为 response_model 参数，表示返回一个 Response 泛型类的列表，每个 Response 对象的 data 属性都是一个包含任意键值对的字典。我们根据实际数据构造了一个 Response 泛型类的列表，并将其作为返回值。

使用这种方式可以方便地自定义返回的 JSON 数组，可以根据实际需要定义不同的 Response 类型，以及每个 Response 对象的 data 属性的键值对。