将成员添加到 Python 枚举

Question

我知道这不是标准用例，但我需要动态地将元素添加到 Python 中的 IntEnum 派生类。请注意，使用功能 API 动态创建 Enum 是不够的。我需要向现有枚举添加元素。我该怎么做？

背景：对于那些想知道为什么有人要这样做的人。我正在包装一个库，枚举的值是在库中定义的。我可以使用库 API 查询名称和值。但是我不能在初始化时这样做，因为它取决于库根据用户请求动态加载的组件。我可以在启动时加载所有组件并使用功能 API 在导入时创建枚举，但这很耗时并且有副作用。

Answer 1

这是来自aenum library¹ 的extend_enum 函数的作业。

---

一对夫妇样本Enums：

from aenum import Enum

class Color(Enum):
    black = 0

class ColorHelp(Enum):
    _init_ = 'value __doc__'
    black = 0, 'the absence of color'

extend_enum 在行动：

from aenum import extend_enum

extend_enum(Color, 'white', 1)
print Color, list(Color)
print repr(Color.black), Color.black, repr(Color.white), Color.white
print

extend_enum(ColorHelp, 'white', 1, 'the presence of every color')
print ColorHelp, list(ColorHelp)
print repr(ColorHelp.black), ColorHelp.black, ColorHelp.black.__doc__, repr(ColorHelp.white), ColorHelp.white, ColorHelp.white.__doc__

这给了我们：

<enum 'Color'> [<Color.black: 0>, <Color.white: 1>]
<Color.black: 0> Color.black <Color.white: 1> Color.white

<enum 'ColorHelp'> [<ColorHelp.black: 0>, <ColorHelp.white: 1>]
<ColorHelp.black: 0> ColorHelp.black the absence of color <ColorHelp.white: (1, 'the presence of every color')> ColorHelp.white None

---

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

Answer 2

枚举是不可变的，这才是重点。您可以创建一个新的枚举来替换原来的枚举：

from enum import Enum

names = [m.name for m in ExistingEnum] + ['newname1', 'newname2']
ExistingEnum = Enum('ExistingEnum', names)

但任何现有的引用（例如，在其他模块中）将继续使用旧定义。

names 可以是：

• 包含成员名称的字符串，以空格或逗号分隔。值从start 递增 1（可以设置为关键字参数，默认为 1）。
• 一个可迭代的成员名称（如上面的代码）。值从 start 开始加 1。
• （成员名称、值）对的可迭代。
• 成员名称 -> 值对的映射。