主页 / 问题 / 1593411
Accepted
ykoavlil
Asked:2024-09-10 15:38:50 +0800 CST2024-09-10 15:38:50 +0800 CST

Pydantic 检查字符串仅包含有效值（StrEnum）

  • 772

高度简化的 json

{
    "tariffOptions": "qwertyTest,qwerty,test"
}

qwertyTest、qwerty、test - 这些都是可能的值，它们可以按任何顺序定位

可能有全部，也可能有一个/两个，或者该字段可能会丢失

您需要检查字符串是否仅包含有效值

我试图这样解决它，但它不起作用

class TariffOptions(StrEnum):
    qwerty_test = 'qwertyTest'
    qwerty = 'qwerty'
    test = 'test'

class Data(Base):
    tariff_options: Optional[TariffOptions] = None
python

1 个回答

  1. Best Answer

    在这里你本质上tariff_options应该得到一个对象列表TariffOptions，而不是Optional.您需要准备源数据，用逗号将其拆分为列表，然后将其解析为枚举列表。

    您可以使用装饰器在方法内分割字符串@model_validator(mode='before')（请参阅模型验证器） - 它以字典的形式接收 json，您可以更正此字典并从此方法返回它：

    from enum import StrEnum
from typing import Any
from pydantic import BaseModel, model_validator


class TariffOptions(StrEnum):
    qwerty_test = 'qwertyTest'
    qwerty = 'qwerty'
    test = 'test'


class Data(BaseModel):
    tariff_options: list[TariffOptions]
    
    @model_validator(mode="before")
    @classmethod
    def from_literal(cls, data: Any) -> Any:
        # Тут одновременно с разбиением происходит изменение
        # имени поля в snake_case
        # - по хорошему это нужно делать отдельно через алиасы
        data["tariff_options"] = data["tariffOptions"].split(",")
        return data


source_json = """
{
    "tariffOptions": "qwertyTest,qwerty,test"
}
"""

print(repr(Data.model_validate_json(source_json)))

    结论：

    Data(tariff_options=[<TariffOptions.qwerty_test: 'qwertyTest'>, <TariffOptions.qwerty: 'qwerty'>, <TariffOptions.test: 'test'>])

    如果您添加枚举中没有的内容，例如：

    {
    "tariffOptions": "qwertyTest,erhrthrt"
}

    就会出现这样的错误：

    Traceback (most recent call last):
  File "/home/insolor/Projects/stackoverflow/test_pydantic/test.py", line 27, in <module>
    print(repr(Data.model_validate_json(source_json)))
               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/insolor/Projects/stackoverflow/test_pydantic/venv/lib/python3.12/site-packages/pydantic/main.py", line 622, in model_validate_json
    return cls.__pydantic_validator__.validate_json(json_data, strict=strict, context=context)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
pydantic_core._pydantic_core.ValidationError: 1 validation error for Data
tariff_options.1
  Input should be 'qwertyTest', 'qwerty' or 'test' [type=enum, input_value='erhrthrt', input_type=str]
    For further information visit https://errors.pydantic.dev/2.9/v/enum
    • 3

相关问题