intflag - python enum to string

Convertir cadena a Enum en Python

Question

Me pregunto cuál es la forma correcta de convertir (deserializar) una cadena a una clase Enum de Python. Parece que getattr(YourEnumType, str) hace el trabajo, pero no estoy seguro de si es lo suficientemente seguro.

Solo para ser más específico, me gustaría convertir una cadena de ''debug'' en un objeto Enum como este:

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

Answer 1

Esta funcionalidad ya está integrada en Enum [1]:

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

[1] Documentos oficiales: Enum programmatic access

Answer 2

Mi solución al problema similar a Java. Espero que ayude a alguien ...

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:
return mm
sim = SignInMethod.value_of(''EMAIL'')
print("""TEST
1). {0}
2). {1}
3). {2}
""".format(sim, sim.name, isinstance(sim, SignInMethod)))

Answer 3

Otra alternativa (especialmente útil si sus cadenas no se asignan 1-1 a sus casos enum) es agregar un staticmethod a su Enum , por ejemplo:

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

Entonces puede hacer question_type = QuestionType.from_str(''singleSelect'')

Answer 4

Solo quiero notificar que esto no funciona en Python 3.6

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

Tendrás que dar los datos como una tupla como esta

MyEnum((''aaa'',))

EDITAR: Esto resulta ser falso. Créditos a un comentarista por señalar mi error

Answer 5

Una mejora a la respuesta de @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

Answer 6

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)

¿O necesita convertir una cadena a Enum conocida ?

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

O:

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