Python 中枚举的常见做法是什么？

Question

如何在 Python 中实现枚举类型（在某些语言中拼写为 enum）？ 获得此功能的常见做法是什么？

Answer 1

class Materials:
    Shaded, Shiny, Transparent, Matte = range(4)

>>> print Materials.Matte
3

更新：对于 Python 3.4+：

从 Python 3.4+ 开始，您现在可以使用 Enum （或 IntEnum 对于具有 int 值的枚举） enum 模块。 使用 enum.auto 来自动增加值：

import enum


class Materials(enum.IntEnum):
    Shaded = 1
    Shiny = enum.auto()
    Transparent = 3
    Matte = enum.auto()


print(Materials.Shiny == 2)  # True
print(Materials.Matte == 4)  # True

Answer 2

我已经多次看到这种模式：

>>> class Enumeration(object):
        def __init__(self, names):  # or *names, with no .split()
            for number, name in enumerate(names.split()):
                setattr(self, name, number)

>>> foo = Enumeration("bar baz quux")
>>> foo.quux
2

您也可以只使用类成员，尽管您必须提供自己的编号：

>>> class Foo(object):
        bar  = 0
        baz  = 1
        quux = 2

>>> Foo.quux
2

如果您正在寻找更健壮的东西（稀疏值、特定于枚举的异常等）， 尝试这个食谱。

Answer 3

我不知道为什么 Python 本身不支持枚举。
我发现模拟它们的最好方法是覆盖 _ str _ 和 _ eq _ 这样你就可以比较它们，当你使用 print() 时，你会得到字符串而不是数值。

class enumSeason():
    Spring = 0
    Summer = 1
    Fall = 2
    Winter = 3
    def __init__(self, Type):
        self.value = Type
    def __str__(self):
        if self.value == enumSeason.Spring:
            return 'Spring'
        if self.value == enumSeason.Summer:
            return 'Summer'
        if self.value == enumSeason.Fall:
            return 'Fall'
        if self.value == enumSeason.Winter:
            return 'Winter'
    def __eq__(self,y):
       return self.value==y.value

用法：

>>> s = enumSeason(enumSeason.Spring)

>>> print(s)

Spring

Answer 4

你也许可以使用继承结构，尽管我玩得越多，我就越觉得肮脏。

class AnimalEnum:
  @classmethod
  def verify(cls, other):
    return issubclass(other.__class__, cls)


class Dog(AnimalEnum):
  pass

def do_something(thing_that_should_be_an_enum):
  if not AnimalEnum.verify(thing_that_should_be_an_enum):
    raise OhGodWhy