如何在python中将int转换为Enum？

Question

在python 2.7.6中使用新的枚举特性（通过backport enum34）。

给定以下定义，如何将int转换为相应的枚举值？

from enum import Enum

class Fruit(Enum):
    Apple = 4
    Orange = 5
    Pear = 6

我知道我可以手工制作一系列if语句来进行转换，但是有没有一种简单的python方法来转换呢？基本上，我想要一个函数ConvertIntToFruit（int），它返回一个枚举值。

我的用例是我有一个记录的csv文件，在那里我将每个记录读入一个对象。其中一个文件字段是表示枚举的整数字段。在填充对象时，我希望将文件中的整型字段转换为对象中相应的枚举值。

Answer 1

你“调用”了Enum类：

Fruit(5)

要将5转换为Fruit.Orange：

>>> from enum import Enum
>>> class Fruit(Enum):
...     Apple = 4
...     Orange = 5
...     Pear = 6
... 
>>> Fruit(5)
<Fruit.Orange: 5>

从文档的Programmatic access to enumeration members and their attributes部分：

Sometimes it’s useful to access members in enumerations programmatically (i.e. situations where Color.red won’t do because the exact color is not known at program-writing time). Enum allows such access:

>>> Color(1)
<Color.red: 1>
>>> Color(3)
<Color.blue: 3>

在相关说明中：要映射包含枚举成员的名称的字符串值，请使用subscription:

>>> s = 'Apple'
>>> Fruit[s]
<Fruit.Apple: 4>

Answer 2

我认为简单地说就是通过调用EnumType(int_value)将int值转换成Enum，然后访问Enum对象的name：

my_fruit_from_int = Fruit(5) #convert to int
fruit_name = my_fruit_from_int.name #get the name
print(fruit_name) #Orange will be printed here

或作为功能：

def convert_int_to_fruit(int_value):
    try:
        my_fruit_from_int = Fruit(int_value)
        return my_fruit_from_int.name
    except:
        return None

Answer 3

我想要类似的东西，这样我就可以从一个引用访问值对的任何一部分。香草的版本：

#!/usr/bin/env python3


from enum import IntEnum


class EnumDemo(IntEnum):
    ENUM_ZERO       = 0
    ENUM_ONE        = 1
    ENUM_TWO        = 2
    ENUM_THREE      = 3
    ENUM_INVALID    = 4


#endclass.


print('Passes')
print('1) %d'%(EnumDemo['ENUM_TWO']))
print('2) %s'%(EnumDemo['ENUM_TWO']))
print('3) %s'%(EnumDemo.ENUM_TWO.name))
print('4) %d'%(EnumDemo.ENUM_TWO))
print()


print('Fails')
print('1) %d'%(EnumDemo.ENUM_TWOa))

失败将按预期引发异常。

更强大的版本：

#!/usr/bin/env python3


class EnumDemo():


    enumeration =   (
                        'ENUM_ZERO',    # 0.
                        'ENUM_ONE',     # 1.
                        'ENUM_TWO',     # 2.
                        'ENUM_THREE',   # 3.
                        'ENUM_INVALID'  # 4.
                    )


    def name(self, val):
        try:

            name = self.enumeration[val]
        except IndexError:

            # Always return last tuple.
            name = self.enumeration[len(self.enumeration) - 1]

        return name


    def number(self, val):
        try:

            index = self.enumeration.index(val)
        except (TypeError, ValueError):

            # Always return last tuple.
            index = (len(self.enumeration) - 1)

        return index


#endclass.


print('Passes')
print('1) %d'%(EnumDemo().number('ENUM_TWO')))
print('2) %s'%(EnumDemo().number('ENUM_TWO')))
print('3) %s'%(EnumDemo().name(1)))
print('4) %s'%(EnumDemo().enumeration[1]))
print()
print('Fails')
print('1) %d'%(EnumDemo().number('ENUM_THREEa')))
print('2) %s'%(EnumDemo().number('ENUM_THREEa')))
print('3) %s'%(EnumDemo().name(11)))
print('4) %s'%(EnumDemo().enumeration[-1]))

如果未正确使用，这将避免创建异常，而是返回故障指示。一个更像python的方法是返回“None”，但我的特定应用程序直接使用文本。