Python 2：使用字符串解释进行枚举的最优雅/ pythonic方式是什么？

Question

我想要一个带有预定义单字符常量的枚举（适合存储在数据库中）和字符串解释。 这是我在想的：

class Fruits(Enum):
    APPLE = 'A'
    PEAR = 'P'
    BANANA = 'B'
    def __unicode__(self):
        if self == APPLE: return "Crunchy Apple"
        if self == PEAR: return "Sweet Pear"
        if self == BANANA: return "Long Banana"

但

fruit = Fruits.APPLE
print fruit.__unicode__()

给

AttributeError: 'unicode' object has no attribute '__unicode__'

此外，必须有一种更优雅的方式

怎么做得更好？

Answer 1

一对观察：

• 你不应该直接调用__dunder__方法; 而是使用匹配命令： unicode而不是__unicode__

• 我无法复制您的问题

使用stdlib Enum （3.4+）或enum34 backport （Python 2.x），你将不得不这么做 - 制作你自己的基础Enum类：

class EnumWithDescription(Enum):
    def __new__(cls, value, desc):
        member = object.__new__(cls)
        member._value_ = value
        member.description = desc
        return member
    def __unicode__(self):
        return self.description

class Fruits(EnumWithDescription):
    _order_ = 'APPLE PEAR BANANA'   # if using Python 2.x and order matters
    APPLE = 'A', 'Crunchy Apple'
    PEAR = 'P', 'Sweet Pear'
    BANANA = 'B', 'Long Banana'

并在使用中：

>>> fruit = Fruits.APPLE
>>> unicode(fruit)
u'Crunchy Apple'

---

如果您可以使用aenum library ^1，您将有更轻松的时间：

from aenum import Enum

class Fruits(Enum, init='value description'):
    APPLE = 'A', 'Crunchy Apple'
    PEAR = 'P', 'Sweet Pear'
    BANANA = 'B', 'Long Banana'
    def describe(self):
        return self.description

并在使用中：

fruit = Fruits.APPLE
fruit.describe()

请注意，由于unicode是Python 3中的默认值，因此我将名称更改为describe 。

---

¹披露：我是Python stdlib Enum ， enum34 backport和Advanced Enumeration（ aenum ）库的作者。

Answer 2

enum34模块有你想要的。

from enum import Enum

class Fruits(Enum):
    apple = 'A'
    pear = 'P'
    banana = 'B'

fruit = Fruits.apple

print fruit.value
>> 'A'

使用整数可能更好

from enum import Enum

class Fruits(Enum):
    apple = 1
    pear = 2
    banana = 3

fruit = Fruits.apple

print fruit.value
>> 1

并使用以下方法重新创建对象（例如，从数据库中获取）：

fruit = Fruits(1)