如何将枚举名称用作其他信息的值

Question

我对此进行了研究并找到了类似的答案，但它们在这种情况下不起作用，因为 auto() 的解析早于 __new__ 被调用，并且只有当整个值为 _auto_null 时。

基本上，我想要的是以下内容：

class MyEnum(str, Enum):
    one = '1 data'
    two = '2 data'

    def __new__(cls, data):
        member = str.__new__(cls, <NEED NAME HERE>)
        member.data = data
        member._value_ = <NEED NAME HERE>

assert MyEnum.one == 'one'
assert MyEnum.one.value == 'one'
assert MyEnum.one.data == '1 data'

但是，名称没有传递到__new__，所以没有什么可以填写<NEED NAME HERE>。

然后，我尝试通过以下方式使用auto()：

class MyEnumBase(Enum):
    def _generate_next_value_(name, start, count, last_values):
        return name

class MyEnum(str, MyEnumBase):
    one = '1 data'
    two = '2 data'

    def __new__(cls, data):
        member = str.__new__(cls, auto())
        member.data = data
        member._value_ = auto()

assert MyEnum.one == 'one'
assert MyEnum.one.value == 'one'
assert MyEnum.one.data == '1 data'

这不起作用，因为auto() 仅在定义成员时用作唯一值时才被解析，如one = auto()。因此，似乎甚至不可能执行 one = auto(), '1 data' 然后在 __new__ 函数中获取这两个参数。

class MyEnum(str, MyEnumBase):
    one = auto(), '1 data'
    two = auto(), '2 data'

    def __new__(cls, name, data):
        member = str.__new__(cls, name)
        member.data = data
        member._value_ = name

看来我唯一的选择是硬编码名称：

one = 'one', '1 data'

并定义__new__ 函数以获取两个参数：

def __new__(cls, name, data):
    ...

我错过了什么吗？有一个更好的方法吗？构建此信息的更好方法？

Answer 1

这绝对是高级行为，所以要使用它，您需要 aenum¹ 库。

该代码将如下所示：

from aenum import Enum

class MyEnum(str, Enum, init='value data'):

    def __new__(cls, name, *args, **kwds):
        obj = str.__new__(cls, name)
        obj._value_ = name
        return obj

    def _generate_next_value_(name, start, count, last_values, *args, **kwds):
        return (name, ) + args

    one = '1 data'
    two = '2 data'

init 设置说明了给定值的用途；在这种情况下：

• 每个成员的value
• 每个成员的data 属性

此外，如果提供的项目少于init 所说的应该有，则将调用_generate_next_value_ 以尝试提供缺失的部分。

在这种情况下，_generate_next_value 将name 添加到给定的data，然后将其传递给__new__。

__new__ 使用 name 作为它的值并忽略其余部分

__init__是自动生成来处理非值项的，所以它设置了data属性

---

对于那些仍在使用 Python 2.7，或者需要他们的代码在 2.7 和 3.x 上工作的人，上面的类应该如下所示（所有更改都在第一个块中）：

class MyEnum(str, Enum):
    _order_ = 'one two'          # only if order actually matters
    _settings_ = AutoValue
    _init_ = 'value data'

    def __new__(cls, name, *args, **kwds):
        obj = str.__new__(cls, name)
        obj._value_ = name
        return obj

    def _generate_next_value_(name, start, count, last_values, *args, **kwds):
        return (name, ) + args

    one = '1 data'
    two = '2 data'

---

¹ 披露：我是Python stdlib Enum、enum34 backport 和Advanced Enumeration (aenum) 库的作者。