如何在 Python 中实例化数据类期间将字符串转换为 Enum

Question

我有一个dataclass dummyClass带有属性虚拟，它只能采用两个值：“ foo”和“ bar” >。我想为此属性使用enum类，以便在使用无效值时会引起错误：

from dataclasses import dataclass, field
from enum import Enum

class DummyAttribute(Enum):
    FOO = "foo"
    BAR = "bar"

@dataclass
class DummyClass:
    dummy: DummyAttribute

问题是我想使用字符串实例化dummyClass （不直接使用dummyAttribute类）。一种方法是使用__ post_init __方法：

@dataclass
class DummyClass:
    dummy: DummyAttribute = field(init=False)
    tmp_dummy: str
    def __post_init__(self):
        self.dummy = DummyAttribute(self.tmp_dummy)

但是，我宁愿避免添加tmp_dummy属性和__ post_init __init __init __方法。是否还有其他简洁的方法将输入字符串投射到dummyattribute枚举？

---

编辑： @chepner的答案帮助我找到了一种更简洁的方法：

from dataclasses import dataclass, field
from enum import Enum

class DummyAttribute(Enum):
    FOO = "foo"
    BAR = "bar"

@dataclass
class DummyClass:
    dummy: str | DummyAttribute
    def __post_init__(self):
        self.dummy = DummyAttribute(self.dummy)

如果有更好的方法，请告诉我！

Answer 1

我认为__ post_init __（您编辑的添加到问题中）是您的最佳选择。 DataClass不支持您想要的价值变压器。 （为此，您可能需要查看Pydantic或attrs软件包。）

from dataclasses import dataclass, InitVar
from enum import Enum

class DummyAttribute(Enum):
    FOO = "foo"
    BAR = "bar"


@dataclass
class DummyClass:
    dummy: DummyAttribute = field(init=False)
    dummyStr: InitVar[str]

    def __post_init__(self, dummyStr):
        self.dummy = DummyAttribute(dummyStr)

Answer 2

作为使用 Enum 的替代方案，如果您真正需要的是 str 类型，那么我建议只使用 str 来代替。

请注意，在这种情况下，typing.Literal 可能会有所帮助：

from dataclasses import dataclass
from typing import Literal


@dataclass
class DummyClass:
    dummy: Literal['foo', 'bar']

    def __post_init__(self):
        # TODO: assert that `self.dummy` is one of valid Literal values
        ...

然后您可以像这样创建实例，如果您传递无效的 Literal<，您的 IDE 会立刻向您发出警告。 /code> 字段的值。

obj1 = DummyClass('bar')
obj2 = DummyClass('blah')  # <-- warning! >:(

如前所述，如果您确实需要验证它是有效的Literal值之一，我建议将该逻辑添加到__post_init__方法中。