使用 Pydantic 模型的 FastAPI 查询参数

Question

我有一个 Pydantic 模型如下

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

通过此设置，我希望获得如下 OpenAPI 架构：

那么，如何使用 Pydantic 模型来获取 FastAPI 中的 from 查询参数呢？

Answer 1

你可以做这样的事情，

from fastapi import FastAPI, Depends

from pydantic import BaseModel

app = FastAPI()


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


@app.get("/")
def read_root(student: Student = Depends()):
    return {"name": student.name, "age": student.age}

另请注意，查询参数通常是“可选”字段，如果您希望将它们设为可选，请使用Optional类型提示，

from fastapi import FastAPI, Depends
from typing import Optional
from pydantic import BaseModel

app = FastAPI()


class Student(BaseModel):
    name: str
    age: Optional[int]


@app.get("/")
def read_root(student: Student = Depends()):
    return {"name": student.name, "age": student.age}