如何将字符串与Python枚举进行比较

Question

我刚刚在python中发现了一个Enum基类的存在，我试图想象它对我有什么用处。

假设我定义了红绿灯状态：

from enum import Enum, auto

class Signal(Enum):
    red = auto()
    green = auto()
    orange = auto()

假设我从程序中的某个子系统接收信息，以表示颜色名称的字符串的形式，例如brain_detected_colour = "red"。

如何将此字符串与我的红绿灯信号进行比较？

显然，brain_detected_colour is Signal.red是False，因为Signal.red不是字符串。

Signal(brain_detected_colour) is Signal.red失败，出现ValueError: 'red' is not a valid Signal。

Answer 1

可以让auto()返回枚举成员的名称作为其值（位于^{} section of the docs^1中）：

>>> class AutoName(Enum):
...     def _generate_next_value_(name, start, count, last_values):
...         return name
...

>>> class Ordinal(AutoName):
...     NORTH = auto()
...     SOUTH = auto()
...     EAST = auto()
...     WEST = auto()
...

>>> list(Ordinal)
[<Ordinal.NORTH: 'NORTH'>, <Ordinal.SOUTH: 'SOUTH'>, <Ordinal.EAST: 'EAST'>, <Ordinal.WEST: 'WEST'>]

---

¹这需要Python 3.6版本，或者aenum2.0²（aenum与2.7版本的Python一起工作）。

²披露：我是Python stdlib ^{}、^{} backport和Advanced Enumeration (^{})库的作者。