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

一个不创建枚举的实例.该Signal(foo)语法用于按值访问Enum成员,这些成员不打算使用它们auto().

但是,可以使用字符串来访问Enum成员,就像dict使用方括号访问a中的值一样:

Signal[brain_detected_colour] is Signal.red

另一种可能性是将字符串与nameEnum成员进行比较:

# Bad practice:
brain_detected_colour is Signal.red.name

但是在这里,我们不测试Enum成员之间的身份,而是比较字符串,因此最好使用相等性测试:

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

(由于字符串实习,字符串之间的身份比较有效,最好不要依赖.感谢@mwchase和@Chris_Rands让我意识到这一点.)

另一种可能性是在创建枚举时明确将成员值设置为其名称:

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

(有关将此方法自动化的方法,请参阅此答案.)

然后,Signal(brain_detected_colour) is Signal.red将是有效的.

Answer 2

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

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在文档¹的部分中:

>>> 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与Pythons一样长到2.7).

²披露:我是Python stdlibEnum,enum34backport和Advanced Enumeration(aenum) 库的作者.