В моем приложении 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
Вы сможете сделать это с помощью валидатора модели.