Skip to content

Response

responses⚓︎

If you want to generate Schemas, pass the responses.

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
class BookBodyWithID(BaseModel):
    bid: int = Field(..., description='book id')
    age: int | None = Field(None, ge=2, le=4, description='Age')
    author: str = Field(None, min_length=2, max_length=4, description='Author')


class BookResponse(BaseModel):
    code: int = Field(0, description="status code")
    message: str = Field("ok", description="exception information")
    data: BookBodyWithID


@app.get('/book/<int:bid>', 
         tags=[book_tag], 
         responses={
             200: BookResponse, 
             # Version 2.4.0 starts supporting response for dictionary types
             201: {"content": {"text/csv": {"schema": {"type": "string"}}}}
         })
def get_book(path: BookPath, query: BookBody):
    """get a book
    get book by id, age or author
    """
    return {"code": 0, "message": "ok", "data": {}}

Now you can use string, int, and HTTPStatus as response's key.

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
from http import HTTPStatus


class BookResponse(BaseModel):
    message: str = Field(..., description="The message")


@api.get("/hello/<string:name>",
        responses={
            HTTPStatus.OK: BookResponse, 
            "201": {"content": {"text/csv": {"schema": {"type": "string"}}}},
            204: None
        })
def hello(path: HelloPath):
    message = {"message": f"""Hello {path.name}!"""}

    response = make_response(json.dumps(message), HTTPStatus.OK)
    response.mimetype = "application/json"
    return response

image-20210526104627124

Validate responses⚓︎

By default, responses are not validated. If you need to validate responses, set validate_responses to True. Here are several ways to achieve this:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
# 1. APP level
app = OpenAPI(__name__, validate_response=True)

# 2. APIBlueprint level
api = APIBlueprint(__name__, validate_response=True)

# 3. APIView level
@api_view.route("/test")
class TestAPI:
    @api_view.doc(responses={201: Response}, validate_response=True)
    def post(self):
        ...

# 4. api level
@app.post("/test", responses={201: Response}, validate_response=True)
def endpoint_test(body: BaseRequest):
    ...

You can also customize the default behavior of response validation by using a custom validate_response_callback.

1
2
3
4
5
6
7
def validate_response_callback(response: Any, responses: Optional[ResponseDict] = None) -> Any:

    # do something

    return response

app = OpenAPI(__name__, validate_response=True, validate_response_callback=validate_response_callback)

More information about OpenAPI responses⚓︎