如何定义枚举类型中枚举成员的映射？

Question

我有一个这样的 Python 枚举：

from enum import Enum, unique

@unique
class Levels(Enum):
  Unknown = 0
  Warning = 1
  Error =   2

  def __eq__(self, other):    return self.value ==  other.value
  def __ne__(self, other):    return self.value !=  other.value
  def __lt__(self, other):    return self.value <   other.value
  def __le__(self, other):    return self.value <=  other.value
  def __gt__(self, other):    return self.value >   other.value
  def __ge__(self, other):    return self.value >=  other.value

  __MAPPING__ = {
    "warn":   Warning,
    "error":  Error
  }

  @classmethod
  def Parse(cls, value, default=None):
    return cls.__MAPPING__.get(value, default)

在这个例子中，我已经将映射提取到一个类成员中（根据timeit.timeit()，它更快）。

我现在的问题是，dict 值使用原始枚举成员值（整数）而不是创建的枚举成员（EnumMeta）。这是合理的，因为在构造 dict 时不存在。

我如何/在哪里可以挂钩到 Enum 类/EnumMeta 类或我自己的 Levels 类，以使用创建的枚举成员修补字典？

Answer 1

[仅显示相关部分...]

@unique
class Levels(Enum):
    Unknown = 0
    Warning = 1
    Error =   2

    __MAPPING__ = {
        "warn":   Warning,
        "error":  Error
   }

    def __init__(self, *args):
        """Patch the embedded MAP dictionary"""
        self.__class__.__MAPPING__[self._name_] = self

__init__ 在创建后为每个成员调用，因此此时您可以使用成员名称作为键更新__MAPPING__。如果你经常使用这种模式，你应该考虑使用另一个装饰器：

def add_mapping(enum_cls):
    for name in enum_cls.__MAPPING__:
        member = enum_cls.__members__[name]
        enum_cls.__MAPPING__[name] = member
    return enum_cls

并在使用中：

@add_mapping
@unique
class Levels(Enum):
    Unknown = 0
    Warning = ...

Answer 2

以Mr. Furman's 提供above 为基础，使用map_value(name) 方法来获取映射：

add_mapping(enum_cls): 
    enum_cls.__MAPPING__ = {e.value: e for e in enum_cls}
    enum_cls.map_value = lambda name, default=None: enum_cls.__MAPPING__.get(name, default)
    return enum_cls