Konvertieren Sie einen String in Python in Enum

Ich frage mich, wie man einen String richtig in eine Python-Enum-Klasse konvertiert (deserialisiert). Scheint getattr(YourEnumType, str)den Job zu erledigen , aber ich bin mir nicht sicher, ob es sicher genug ist.

Um genauer zu sein, möchte ich einen 'debug'String wie folgt in ein Enum-Objekt konvertieren :

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

Diese Funktionalität ist bereits in Enum [1] integriert:

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

[1] Offizielle Dokumente: Enum programmatic access

Eine andere Alternative (besonders nützlich, wenn Ihre Zeichenfolgen Ihren Aufzählungsfällen nicht 1-1 zuordnen) besteht darin, a staticmethodzu Ihren hinzuzufügen Enum, z.

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

Dann können Sie tun 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)

Oder müssen Sie einen String in eine bekannte Enum konvertieren ?

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

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

Oder:

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

Meine Java-ähnliche Lösung für das Problem. Hoffe es hilft jemandem ...

    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)))

Eine Verbesserung der Antwort von @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

Ich möchte nur benachrichtigen, dass dies in Python 3.6 nicht funktioniert

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

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

Sie müssen die Daten als solches Tupel angeben

MyEnum(('aaa',))

EDIT: Dies stellt sich als falsch heraus. Dank an einen Kommentator für den Hinweis auf meinen Fehler