flask_pydantic_query_and_body_validation_with_decorator.py

1from flask import Flask, request
2from flask_pydantic import validate
3from pydantic import BaseModel
4
5app = Flask(__name__)
6
7class QueryModel(BaseModel):
8    name: str
9
10class BodyModel(BaseModel):
11    age: int
12
13class ResponseModel(BaseModel):
14    id: int
15    name: str
16    age: int
17
18@app.route("/", methods=["POST"])
19@validate()
20def post(query: QueryModel, body: BodyModel):
21    # query and body are already validated Pydantic objects
22    name = query.name
23    age = body.age
24    return ResponseModel(id=1, name=name, age=age)
25
26if __name__ == "__main__":
27    app.run()