如何通过点使用 pydantic json 字段

Question

我有这样的模型、枚举、字段：

from pydantic import BaseModel, Json

class SlotActionEnum(Enum):
    NORMAL = 'normal'
    REASK = 'reask'


class ChannelMessage(Json):
    answerText: str
    slot_action: SlotActionEnum = SlotActionEnum.NORMAL 


class Request(BaseModel):
    conversationId: str
    channelMessage: ChannelMessage



o = Request(**{
    "conversationId": "id10",
    "channelMessage": "{\"answerText\": \"sadfg\", \"slot_action\": \"reask\"}"
})

这里有两个问题：

slot_action 不能这样访问：o.channelMessage.slot_action

这会导致 AttributeError：

AttributeError: 'dict' object has no attribute 'slot_action'

如果我通过方括号访问该字段 PyCharm 突出显示：

slot_action 中的值不是 SlotActionEnum，而是像 reask 或 normal
这样的值
slot_action 可以是空字符串。如何将空字符串替换为 normal?

遇到这些问题我应该怎么做？我应该这样做吗：

class ChannelMessage(BaseModel):
    answerText: str
    slot_action: SlotActionEnum = SlotActionEnum.NORMAL

class Request(BaseModel):
    conversationId: str
    channelMessage: ChannelMessage

    def __init__(__pydantic_self__, **data: Any) -> None:
        channel_message = json.loads(data.pop('channelMessage'))
        if channel_message['slot_action'] == '':
            channel_message['slot_action'] = SlotActionEnum.NORMAL.value
        channel_message['slot_action'] = SlotActionEnum(channel_message['slot_action'])
        super().__init__(**data, channelMessage=channel_message)

？好吧，但看起来很丑。

Answer 1

from pydantic import BaseModel, validator
from enum import Enum


class SlotActionEnum(Enum):
    NORMAL = 'normal'
    REASK = 'reask'


class ChannelMessage(BaseModel):
    answerText: str
    slot_action: SlotActionEnum = SlotActionEnum.NORMAL

    @validator("slot_action", pre=True)
    def valid_slot_action(cls, v):
        if v == "":
            return SlotActionEnum.NORMAL
        return v


class Request(BaseModel):
    conversationId: str
    channelMessage: ChannelMessage

    @validator("channelMessage", pre=True)
    def valid_channel_message(cls, v):
        if isinstance(v, str):
            return ChannelMessage.parse_raw(v)
        return v


o = Request(**{
    "conversationId": "id10",
    "channelMessage": "{\"answerText\": \"sadfg\", \"slot_action\": \"reask\"}"
})

如何通过点使用 pydantic json 字段

How to work with pydantic json field through dot

python

pydantic