枚举函数将值映射到相关值

Question

目前，我有一个表示名称/数字映射的枚举。然而，每个 EventType 还具有额外的“相关属性”（例如状态代码和消息类型）。

class EventType(Enum):
  CANCELLED = 0
  ONTIME = 1
  DELAYED = 2

  def get_status(self):
    if self == EventType.CANCELLED:
        return "statuscode1"
    elif self == EventType.DELAYED:
        return "statuscode2"
    else:
        return "statuscode3"

  def get_message_type(self):
    if self == EventType.CANCELLED:
        return "messagetype1"
    elif self == EventType.DELAYED:
        return "messagetype2"
    else:
        return "messagetype3"

除了创建上面的方法和大量的 if 链检查自身之外，有没有更简洁的重构方法来返回状态代码和消息类型？几乎就像CANCELLED = (0, statuscode1, messagetype1).. 我怎样才能在枚举中表示这个概念？枚举甚至是正确的方法吗？

Answer 1

如果您希望Enum 成员的值为0、1 或2，则需要覆盖__new__，或使用aenum。

使用aenum 库¹：

from aenum import MultiValueEnum

class EventType(MultiValueEnum):
    _init_ = 'value status message'
    CANCELLED = 0, 'status1', 'message1'
    ONTIME = 1, 'status2', 'message2'
    DELAYED = 2, 'status3', 'message3'

并在使用中：

>>> print(EventType.ONTIME)
EventType.ONTIME
>>> print(EventType.ONTIME.value)
1
>>> print(EventType.ONTIME.status)
status2
>>> print(EventType.ONTIME.message)
message2

请参阅this answer，了解如何使用 stdlib 枚举（它使用覆盖 __new__ 技术）执行此操作的示例。

---

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

Answer 2

在 Python 3.4 中，有一个新的Enum data type，您可以将其用作：

class EventType(Enum):

    def __init__(self, id, code, message):
        self.id = id
        self.code = code
        self.message = message

    CANCELLED = 1, 'code1', 'message1'
    ONTIME = 2, 'code2', 'message2'
    DELAYED = 3, 'code3', 'message3

要使用，只需：

EventType.CANCELLED.code # returns 'message1'

如果只是你描述的用例，你也可以根据需要使用命名元组：

from collections import namedtuple

    Event = namedtuple('event', ['id', 'code', 'message'])

    class EventType:
        CANCELLED = Event(1, 'code1', 'message1')
        ONTIME = Event(2, 'code2', 'message2')
        DELAYED = Event(3, 'code3', 'message3')

要使用这个：

EventType.CANCELLED # return event(id=1, code='code1', message='message1')
EventType.CANCELLED.message # return message1

Answer 3

您可以使用元组作为Enum 对象的值，并将状态和消息字符串与整数代码一起包括在内。然后您可以添加方法（或property 描述符）以通过value 属性获取元组的各个部分：

class EventType(Enum):
   CANCELLED = 0, "status1", "message1"
   ONTIME = 1, "status2", "message2"
   DELAYED = 2, "status3", "message3"

   @property
   def code(self):
       return self.value[0]

   @property
   def status(self):
       return self.value[1]

   @property
   def message(self):
       return self.value[2]

print(EventType.CANCELLED.message) # prints "message3"