在 pydantic BaseModel 上验证多个字段

Question

在我的 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

Answer 1

是的，您可以使用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