Проверка нескольких полей в pydantic BaseModel

В моем приложении 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

Вы сможете сделать это с помощью валидатора модели.

M.O. 30.03.2024 14:01

Пожалуйста, взгляните на этот ответ

Chris 31.03.2024 17:02
Почему в Python есть оператор "pass"?
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
0
2
232
1
Перейти к ответу Данный вопрос помечен как решенный

Ответы 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

Другие вопросы по теме