Làm thế nào để chuyển đổi int sang Enum trong python?

Sử dụng tính năng Enum mới (thông qua backport enum34 ) với python 2.7.6.

Với định nghĩa sau, làm cách nào tôi có thể chuyển đổi một int thành giá trị Enum tương ứng?

from enum import Enum

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

Tôi biết tôi có thể tạo thủ công một loạt câu lệnh if để thực hiện chuyển đổi nhưng có cách nào dễ dàng để chuyển đổi không? Về cơ bản, tôi muốn một hàm ConvertIntToFruit (int) trả về giá trị enum.

Trường hợp sử dụng của tôi là tôi có một tệp bản ghi csv nơi tôi đang đọc từng bản ghi thành một đối tượng. Một trong các trường tệp là trường số nguyên đại diện cho một kiểu liệt kê. Khi tôi đang điền đối tượng, tôi muốn chuyển đổi trường số nguyên đó từ tệp thành giá trị Enum tương ứng trong đối tượng.

Bạn 'gọi' Enumlớp:

Fruit(5)

biến 5thành Fruit.Orange:

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

Từ quyền truy cập có lập trình đến các thành viên điều tra và phần thuộc tính của họ trong tài liệu:

Đôi khi, việc truy cập các thành viên trong bảng liệt kê theo chương trình Color.redsẽ rất hữu ích (tức là các tình huống sẽ không thực hiện được vì không xác định được màu chính xác tại thời điểm viết chương trình). Enumcho phép truy cập như vậy:

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

Trong một lưu ý liên quan: để ánh xạ một giá trị chuỗi có chứa tên của một thành viên enum, hãy sử dụng đăng ký:

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

Tôi nghĩ nó nói một cách đơn giản là chuyển đổi intgiá trị thành Enumbằng cách gọi EnumType(int_value), sau đó truy cập namevào Enumđối tượng:

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

Hoặc như một chức năng:

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

Tôi muốn một cái gì đó tương tự để tôi có thể truy cập một trong hai phần của cặp giá trị từ một tham chiếu duy nhất. Phiên bản vani:

#!/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))

Sự thất bại ném ra một ngoại lệ như mong đợi.

Một phiên bản mạnh mẽ hơn:

#!/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]))

Khi không được sử dụng đúng cách, điều này sẽ tránh tạo ra một ngoại lệ và thay vào đó, trả lại một chỉ báo lỗi. Một cách Pythonic khác để làm điều này là trả lại "Không có" nhưng ứng dụng cụ thể của tôi sử dụng trực tiếp văn bản.