如何将字符串与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

一个人不会创建instance of an Enum。 Signal(foo) 语法用于按值访问 Enum 成员，当它们是 auto() 时不打算使用。

但是，可以使用字符串到access Enum members，就像使用方括号访问dict 中的值一样：

Signal[brain_detected_colour] is Signal.red

另一种可能性是将字符串与枚举成员的name 进行比较：

# Bad practice:
brain_detected_colour is Signal.red.name

但在这里，我们不是在测试 Enum 成员之间的身份，而是比较字符串，所以最好使用相等测试：

# Better practice:
brain_detected_colour == Signal.red.name

（感谢string interning，字符串之间的身份比较有效，最好不要依赖它。感谢@mwchase 和@Chris_Rands 让我意识到这一点。）

另一种可能性是在创建 Enum 时将成员值显式设置为其名称：

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

（请参阅this answer 了解自动化的方法。）

那么，Signal(brain_detected_colour) is Signal.red 将是有效的。

Answer 2

更好的做法是从str 继承Signal：

class Signal(str, Enum):
    red = 'red'
    green = 'green'
    orange = 'orange'

brain_detected_colour = 'red'
brain_detected_colour == Signal.red  # direct comparison

Answer 3

可以让auto() 返回枚举成员的名称作为其值（在auto section of the docs¹:

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 或 aenum 2.0² 版本（aenum 可用于 2.7 之前的 Python）。

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

Answer 4

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

    def equals(self, string):
       return self.name == string

brain_detected_colour = "red"

if Signal.red.equals(brain_detected_colour):
   #something awesome