![](/img/trans.png)
[英]How to return a response with a list of different Pydantic models using FastAPI?
[英]Using different Pydantic models depending on the value of fields
我有 2 个 Pydantic 模型( var1
和var2
)。 PostExample
方法的输入可以接收第一个模型或第二个模型的数据。 Union
的使用有助于解决这个问题,但在验证过程中,它会为第一个和第二个模型抛出错误。
如何做到这一点,以便在填写字段时出错,仅针对某个模型返回验证器错误,而不是同时返回两者? (如果有帮助,可以通过字段 A 的长度来区分模型)。
主文件
@app.post("/PostExample")
def postExample(request: Union[schemas.var1, schemas.var2]):
result = post_registration_request.requsest_response()
return result
架构.py
class var1(BaseModel):
A: str
B: int
C: str
D: str
class var2(BaseModel):
A: str
E: int
F: str
您可以使用有区别的工会(@larsks 在评论中提到这一点)。 设置一个有区别的联合, “验证更快,因为它只针对一个模型进行尝试” ,以及“在失败的情况下只引发一个显式错误” 。 下面的工作示例:
应用程序.py
import schemas
from fastapi import FastAPI, Body
from typing import Union
app = FastAPI()
@app.post("/")
def submit(item: Union[schemas.Model1, schemas.Model2] = Body(..., discriminator='model_type')):
return item
架构.py
from typing import Literal
from pydantic import BaseModel
class Model1(BaseModel):
model_type: Literal['m1']
A: str
B: int
C: str
D: str
class Model2(BaseModel):
model_type: Literal['m2']
A: str
E: int
F: str
测试输入 - 输出
#1 Successful Response #2 Validation error #3 Validation error
# Request body # Request body # Request body
{ { {
"model_type": "m1", "model_type": "m1", "model_type": "m2",
"A": "string", "A": "string", "A": "string",
"B": 0, "C": "string", "C": "string",
"C": "string", "D": "string" "D": "string"
"D": "string" } }
}
# Server response # Server response # Server response
200 { {
"detail": [ "detail": [
{ {
"loc": [ "loc": [
"body", "body",
"Model1", "Model2",
"B" "E"
], ],
"msg": "field required", "msg": "field required",
"type": "value_error.missing" "type": "value_error.missing"
} },
] {
} "loc": [
"body",
"Model2",
"F"
],
"msg": "field required",
"type": "value_error.missing"
}
]
}
另一种方法是尝试解析模型(基于您作为查询/路径参数传递的鉴别器),如此处所述(更新 1) 。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.