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

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 设置说明给定值的用途；在这种情况下:

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

另外，如果给出的项目少于 init 认为应该有的项目，那么将调用 _generate_next_value_ 以尝试提供缺失的部分。

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

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

__init__ 是自动生成的，用于处理非值项，因此它设置了 data 属性

---

对于那些仍在使用 Python 2.7，或者需要他们的代码既适用于 2.7 也适用于 3.x 的人，上面的类应该如下所示(所有更改都在第一 block 中):

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)图书馆。