在我的 fastapi 应用程序中,我创建了一个包含两个字段(以及其他字段)的 pydantic BaseModel: “relation_type”和“document_list”(两者都是可选的)。我想验证一下,如果“relation_type”有一个值,那么“document_list”必须至少有一个元素。否则会显示验证错误。我该怎么办?
class TipoRelacionEnum(str, Enum):
nota_credito = "01"
nota_debito = "02"
devolucion_mercancias = "03"
sustitucion = "04"
traslado = "05"
facturacion_generada = "06"
anticipo = "07"
class Cfdi(BaseModel):
relation_type: Optional[Annotated[TipoRelacionEnum, Field(title="Tipo de Relación",
description="""Se debe registrar la clave de la relación que existe entre este
comprobante que se está generando y el o los CFDI previos.""",
examples=[
"04",
"01",
],
max_length=2)]] = None
document_list: list[str] | None = None
是的,您可以使用model_validator来实现
from pydantic import BaseModel, model_validator
from typing import Optional
class Cfdi(BaseModel):
relation_type: Optional[...] = None
document_list: Optional[list[str]] = None
@model_validator(mode="after")
@classmethod
def check_two_fields_together(cls, data: dict) -> dict:
if relation_type and len(document_list) < 1:
raise ValueError("Error text")
return data