Fastapi custom response model

Question

I have a router that fetches all data from the database. Here is my code:

@router.get('/articles/', response_model=List[articles_schema.Articles])
async def main_endpoint():
    query = articles_model.articles.select().where(articles_model.articles.c.status == 2)
    return await db.database.fetch_all(query)

The response is an array that contains JSON objects like this

[
    {
        "title": "example1",
        "content": "example_content1"
    },
    {
        "title": "example2",
        "content": "example_content2"
    },
]

But I want to make the response like this:

{
    "items": [
        {
            "title": "example1",
            "content": "example_content1"
        },
        {
            "title": "example2",
            "content": "example_content2"
        },
    ]
}

How can I achieve that? Please help. Thank you in advance

Jonathan Vargas · Accepted Answer · 2022-11-02 03:24:18Z

10

Also, You can create a custom responses using generic types as follow if you plan to reuse a response template

from typing import Any, Generic, List, Optional, TypeVar
from pydantic import BaseModel
from pydantic.generics import GenericModel

DataType = TypeVar("DataType")

class IResponseBase(GenericModel, Generic[DataType]):
    message: str = ""
    meta: dict = {}
    items: Optional[DataType] = None


@router.get('/articles/', response_model=IResponseBase[List[Articles]])
async def main_endpoint():
    query = articles_model.articles.select().where(articles_model.articles.c.status == 2)
    items=await db.database.fetch_all(query)
    return IResponseBase[List[Articles]](items=items)

You can find a FastAPI template here https://github.com/jonra1993/fastapi-alembic-sqlmodel-async/blob/main/fastapi-alembic-sqlmodel-async/app/schemas/response_schema.py

edited Nov 2, 2022 at 3:24

answered Mar 20, 2022 at 2:09

Jonathan Vargas

4401 gold badge7 silver badges15 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Paul P · Accepted Answer · 2021-10-09 17:42:16Z

8

You could simply define another model containing the items list as a field:

from pydantic import BaseModel
from typing import List

class ResponseModel(BaseModel):
    items: List[articles_schema.Articles]

and use it in the response:

@router.get('/articles/', response_model=ResponseModel)
async def main_endpoint():
    query = articles_model.articles.select().where(
        articles_model.articles.c.status == 2
    )
    return ResponseModel(
        items=await db.database.fetch_all(query),
    )

answered Oct 9, 2021 at 17:42

Paul P

4,0372 gold badges15 silver badges28 bronze badges

Collectives™ on Stack Overflow

Fastapi custom response model

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related