Python 枚举组合

Question

Python 枚举组合

pythonenumsmetaprogrammingpython-3.5metaclass

7

我希望创建一个新的枚举（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)])))

我的问题是: 是否有一种高级的方法来实现这个目标，以便有一个适当的类定义呢？比如覆盖一些元类方法等等。我想要像这样的东西，以便还可以提供docstring：

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

有什么想法吗？谢谢！

- waszil

你不需要使用OrderedDict，你可以直接使用列表推导式。 - jdehesa

我知道它是这样工作的，问题在于我正在使用PyCharm，如果只给出一个列表，它会发出警告：“期望类型为'Integral'，而得到了'List[Tuple[Any, Any]]'”，而OrderedDict解决了这个问题。但是确实，即使没有那个也可以工作。无论如何，我仍然无法像类定义那样处理它。 - waszil

可能是如何扩展Python Enum？的重复问题。 - John Crawford

2个回答

0

这个库显式地防止这样做：

只有在枚举没有定义任何成员的情况下才允许对枚举进行子类化。

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

因此，我找到了一个Stackoverflow答案，使用了与你几乎相同的解决方法。我认为这是唯一的办法。

- Kruupös

是的，看起来这似乎是唯一的方法。同时我尝试了类似于这篇文章中的做法，即对EnumMeta元类进行子类化并玩弄其__new__方法，但最终所有尝试都遇到了相同的问题：TypeError: Cannot extend enumerations。 - waszil

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Ethan Furman · Accepted Answer

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标准库Enum、enum34回溯和高级枚举(aenum)库的作者。