Python枚举组合

Question

我想基于两个现有的类创建一个新的 Enum (IntEnum) 类。有一个可行的解决方案，如下所示：

from enum import unique, IntEnum
from itertools import chain
from collections import OrderedDict

@unique
class FirstEnumClass(IntEnum):
    a = 1
    b = 2

@unique
class SecondEnumClass(IntEnum):
    c = 3
    d = 4

# here a combined class is created:
CombinedEnumClass = unique(IntEnum('CombinedEnumClass', OrderedDict([(i.name, i.value) for i in chain(FirstEnumClass, SecondEnumClass)])))

我的问题：有没有一种奇特的方法来实现这一点，以便有一个适当的类定义？喜欢覆盖一些元类方法，还是这样？我想要这样的东西，这样也可以给出文档字符串：

@unique
class CombinedEnumClass(IntEnum):
    """ docstring """
    # magic needed here

有什么想法吗？谢谢！

Answer 1

Python 2

使用vars() 的技巧可以完成：

class CombinedEnum(IntEnum):
    """ doc string """
    cls = vars()
    for member in chain(list(FirstEnumClass), list(SecondEnumClass)):
        cls[member.name] = member.value
    del member, cls

print(list(CombinedEnum))

之所以有效，是因为：

• vars() 返回当前命名空间
• 修改命名空间会修改类

我们删除member 和cls，因为我们不希望他们成为会员。

---

Python 3

以上内容（还）不适用于 Python3 3.4 和 3.5 中的新 Enum（不确定 3.6）。解决方法是改用aenum，这让我们可以告诉Enum 忽略某些名称：

from aenum import IntEnum

class CombinedEnum(IntEnum):
    """ doc string """
    _ignore_ = 'member cls'
    cls = vars()
    for member in chain(list(FirstEnumClass), list(SecondEnumClass)):
        cls[member.name] = member.value

---

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

Answer 2

图书馆prevents explicitly 这样做：

仅当枚举不包含子类时才允许对枚举进行子类化 定义任何成员。

允许定义成员的枚举的子类化将导致 违反类型和实例的一些重要不变量。在 另一方面，允许共享一些常见行为是有意义的 在一组枚举之间。

因此，我发现Stackoverflow answer 使用的解决方法与您几乎相同。我认为这是唯一的方法。