Chuyển chuỗi thành Enum bằng Python

Question

Tôi tự hỏi cách chính xác để chuyển đổi (deserializing) một chuỗi thành lớp Enum của Python là gì. Có vẻ như getattr(YourEnumType, str) thực hiện công việc, nhưng tôi không chắc liệu nó có đủ an toàn hay không.

Chỉ cần được cụ thể hơn, tôi muốn chuyển đổi một chuỗi 'debug' đến một đối tượng Enum như thế này:

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

Answer 1

Chức năng này đã được tích hợp sẵn trong Enum [1]:

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

[1] tài liệu chính thức: Enum programmatic access

Answer 2

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) 

Hoặc bạn cần phải chuyển đổi chuỗi để biết Enum?

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

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

Hoặc:

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 

Answer 3

Một lựa chọn khác (đặc biệt là hữu ích nếu các chuỗi của bạn không ánh xạ 1-1 đến các trường hợp enum của bạn) là thêm staticmethod để Enum của bạn, ví dụ:

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 

Sau đó, bạn có thể làm question_type = QuestionType.from_str('singleSelect')