Pydantic 枚举字段未转换为字符串

Question

我试图将一个类中的一个字段限制为一个枚举。但是，当我尝试从课堂中取出字典时，它不会转换为字符串。相反，它保留了枚举。我检查了pydantic documentation，但找不到与我的问题相关的任何内容。

这段代码代表了我实际需要的内容。

from enum import Enum
from pydantic import BaseModel

class S(str, Enum):
    am='am'
    pm='pm'

class K(BaseModel):
    k:S
    z:str

a = K(k='am', z='rrrr')
print(a.dict()) # {'k': <S.am: 'am'>, 'z': 'rrrr'}

我正在尝试让 .dict() 方法返回 {'k': 'am', 'z': 'rrrr'}

Answer 1

你需要使用model config的use_enum_values选项：

use_enum_values

是否使用枚举的value 属性填充模型，而不是原始枚举。如果您想稍后序列化model.dict()，这可能很有用（默认：False）

from enum import Enum
from pydantic import BaseModel

class S(str, Enum):
    am='am'
    pm='pm'

class K(BaseModel):
    k:S
    z:str

    class Config:  
        use_enum_values = True  # <--

a = K(k='am', z='rrrr')
print(a.dict())