Как сделать валидацию в использование Pydantic с заменой значения из словаря?

Question

[domanskiy]

Делаю небольшой парсер.
Есть yml (yml_catalog из 1С для яндекса с описанием продуктов)
Продукты из yml парсятся и по API загружаются в woocommerce. С этим вопросов нет.
Но есть момент. В yml есть key": "производитель".
В woocommerce есть key": "производитель", но там список другой.
Сейчас я просто сравнением значение из yml меняю на значение из woocommerce.
Вопрос, можно ли это (замена значения) сделать прогоняя json через pydantic???

Answer 1

[Влад Григорьев]

с живым примером было бы проще, но так можно через @validator обрабатывать значения
Пример реализации, само собой можно более сложную обработку делать

from pydantic import BaseModel, validator

keys = {
    '1': 'one'
}


class User(BaseModel):
    name: str = ''
    key: str = ''

    @validator('key')
    def update_key(cls, value):
        return keys.get(value, value)


user_1 = User.parse_obj({'name': 'Test User', 'key': '1'})
user_2 = User.parse_obj({'name': 'Test User', 'key': '2'})

print(user_1)
print(user_2)

name='Test User' key='one'
name='Test User' key='2'

Comments

[domanskiy]

Спасибо. Этого вполне хватит.

[domanskiy]

Можете помочь еще?
Вот что нужно

Кусок кода классов pydantic

class ProductMetaData(BaseModel):
    key: str = Field(..., description='Ключ мета.')
    value: Any = Field(..., description='Значение мета.')

class Product(BaseModel):
    meta_data: List[ProductMetaData] = Field(..., description='метаданные продукта')

А вот что приходит на вход:

"meta_data": [{"key": "галерея", "value": ["1378"]},
                      {"key": "производитель", "value": "1337"},
                      {"key": "тип_кондиционера", "value": "настенная сплит-система"},
                      {"key": "площадь", "value": "84"},
                      {"key": "мощность_в_режиме_охлаждения", "value": "8000"},
                      {"key": "мощность_в_режиме_обогрева", "value": "8500"},
                      {"key": "потребляемая_мощность_при_обогреве", "value": "2640"},
                      {"key": "потребляемая_мощность_при_охлаждении", "value": "2840"},
                      {"key": "инверторный", "value": "0"},
                      ...

Для key = производитель, если value = 1337, то на выходе нужно 49
Для key = тип_кондиционера, если value = "настенная сплит-система", то на выходе value = "Настенные кондиционеры"
Как бы это сказать. Тут сопастовление value из списка как в вашем ответе, но в зависимости от key. Наверное для каждого key нужно отдельный список.

[Влад Григорьев]

domanskiy,Если прям сильно не заморачиваться и в лоб

Как-то так

from typing import Any, List

from pydantic import BaseModel, Field, validator

tr_producers = {
    '1337': '49'
}
tr_names = {
    'настенная сплит-система': 'Настенные кондиционеры'
}

transformer = {
    'производитель': tr_producers,
    'тип_кондиционера': tr_names
}
meta_data = [
    {"key": "галерея", "value": ["1378"]},
    {"key": "производитель", "value": "1337"},
    {"key": "тип_кондиционера", "value": "настенная сплит-система"},
    {"key": "площадь", "value": "84"},
    {"key": "мощность_в_режиме_охлаждения", "value": "8000"},
    {"key": "мощность_в_режиме_обогрева", "value": "8500"},
    {"key": "потребляемая_мощность_при_обогреве", "value": "2640"},
    {"key": "потребляемая_мощность_при_охлаждении", "value": "2840"},
    {"key": "инверторный", "value": "0"}
]


class ProductMetaData(BaseModel):
    key: str = Field(..., description='Ключ мета.')
    value: Any = Field(..., description='Значение мета.')

    @validator('value')
    def update_data(cls, value, values):
        tr = transformer.get(values['key'])
        if tr:
            return tr.get(value, value)
        return value


class Product(BaseModel):
    meta_data: List[ProductMetaData] = Field(..., description='метаданные продукта')


print(Product.parse_obj({'meta_data': meta_data}).json(indent=4, ensure_ascii=False))

Результат

{
    "meta_data": [
        {
            "key": "галерея",
            "value": [
                "1378"
            ]
        },
        {
            "key": "производитель",
            "value": "49"
        },
        {
            "key": "тип_кондиционера",
            "value": "Настенные кондиционеры"
        },
        {
            "key": "площадь",
            "value": "84"
        },
        {
            "key": "мощность_в_режиме_охлаждения",
            "value": "8000"
        },
        {
            "key": "мощность_в_режиме_обогрева",
            "value": "8500"
        },
        {
            "key": "потребляемая_мощность_при_обогреве",
            "value": "2640"
        },
        {
            "key": "потребляемая_мощность_при_охлаждении",
            "value": "2840"
        },
        {
            "key": "инверторный",
            "value": "0"
        }
    ]
}

[domanskiy]

Влад Григорьев, Спасибо большое.