Ubah string menjadi Enum dengan Python

Saya bertanya-tanya apa cara yang benar untuk mengubah (deserializing) string ke kelas Enum Python. Sepertinya getattr(YourEnumType, str)melakukan pekerjaan, tapi saya tidak yakin apakah itu cukup aman.

Untuk lebih spesifik, saya ingin mengonversi 'debug'string ke objek Enum seperti ini:

class BuildType(Enum):
    debug = 200
    release = 400

Fungsi ini sudah ada di dalam untuk Enum [1]:

>>> from enum import Enum
>>> class Build(Enum):
...   debug = 200
...   build = 400
... 
>>> Build['debug']
<Build.debug: 200>

[1] Dokumen resmi: Enum programmatic access

Alternatif lain (terutama berguna jika string Anda tidak peta 1-1 untuk kasus enum Anda) adalah menambahkan staticmethoduntuk Anda Enum, misalnya:

class QuestionType(enum.Enum):
    MULTI_SELECT = "multi"
    SINGLE_SELECT = "single"

    @staticmethod
    def from_str(label):
        if label in ('single', 'singleSelect'):
            return QuestionType.SINGLE_SELECT
        elif label in ('multi', 'multiSelect'):
            return QuestionType.MULTI_SELECT
        else:
            raise NotImplementedError

Maka Anda bisa melakukannya question_type = QuestionType.from_str('singleSelect')

def custom_enum(typename, items_dict):
    class_definition = """
from enum import Enum

class {}(Enum):
    {}""".format(typename, '\n    '.join(['{} = {}'.format(k, v) for k, v in items_dict.items()]))

    namespace = dict(__name__='enum_%s' % typename)
    exec(class_definition, namespace)
    result = namespace[typename]
    result._source = class_definition
    return result

MyEnum = custom_enum('MyEnum', {'a': 123, 'b': 321})
print(MyEnum.a, MyEnum.b)

Atau Anda perlu mengonversi string menjadi Enum yang dikenal ?

class MyEnum(Enum):
    a = 'aaa'
    b = 123

print(MyEnum('aaa'), MyEnum(123))

Atau:

class BuildType(Enum):
    debug = 200
    release = 400

print(BuildType.__dict__['debug'])

print(eval('BuildType.debug'))
print(type(eval('BuildType.debug')))    
print(eval(BuildType.__name__ + '.debug'))  # for work with code refactoring

Solusi mirip Java saya untuk masalah ini. Semoga ini bisa membantu seseorang ...

    from enum import Enum, auto


    class SignInMethod(Enum):
        EMAIL = auto(),
        GOOGLE = auto()

        @staticmethod
        def value_of(value) -> Enum:
            for m, mm in SignInMethod.__members__.items():
                if m == value.upper():
                    return mm


    sim = SignInMethod.value_of('EMAIL')
    print("""TEST
    1). {0}
    2). {1}
    3). {2}
    """.format(sim, sim.name, isinstance(sim, SignInMethod)))

Peningkatan jawaban @rogueleaderr:

class QuestionType(enum.Enum):
    MULTI_SELECT = "multi"
    SINGLE_SELECT = "single"

    @classmethod
    def from_str(cls, label):
        if label in ('single', 'singleSelect'):
            return cls.SINGLE_SELECT
        elif label in ('multi', 'multiSelect'):
            return cls.MULTI_SELECT
        else:
            raise NotImplementedError

Saya hanya ingin memberi tahu ini tidak bekerja di python 3.6

class MyEnum(Enum):
    a = 'aaa'
    b = 123

print(MyEnum('aaa'), MyEnum(123))

Anda harus memberikan data sebagai tuple seperti ini

MyEnum(('aaa',))

EDIT: Ini ternyata salah. Penghargaan kepada komentator karena menunjukkan kesalahan saya