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

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 

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

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

在本例中，我已经提取出的映射到一个类的成员（根据timeit.timeit()，其更快）。

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

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

Answer 1

[只显示相关的片...]

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

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

    def __init__(self, *args): 
     """Patch the embedded MAP dictionary""" 
     self.__class__.__MAPPPING__[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