有没有办法指定python枚举的默认值？

Question

考虑下面的枚举：

class MyEnum(IntEnum): 

    A = 0 
    B = 1 
    C = 2 

我如何可以指定一个默认值。我希望能够做到：

my_val = MyEnum() 

，并有my_val是<MyEnum.A: 0>

这可能吗？我试过定制__new__,__init__,__call__但我无法让它工作。

Answer 1

MyEnum(..)由EnumMeta.__call__处理。你需要重写方法：

from enum import EnumMeta, IntEnum 


class DefaultEnumMeta(EnumMeta): 
    default = object() 

    def __call__(cls, value=default, *args, **kwargs): 
     if value is DefaultEnumMeta.default: 
      # Assume the first enum is default 
      return next(iter(cls)) 
     return super().__call__(value, *args, **kwargs) 
     # return super(DefaultEnumMeta, cls).__call__(value, *args, **kwargs) # PY2 


class MyEnum(IntEnum, metaclass=DefaultEnumMeta): 
    # __metaclass__ = DefaultEnumMeta # PY2 with enum34 
    A = 0 
    B = 1 
    C = 2 


assert MyEnum() is MyEnum.A 
assert MyEnum(0) is MyEnum.A 
assert MyEnum(1) is not MyEnum.A 

Answer 2

为什么不使用standard syntax？

my_val = MyEnum.A 

---

如果你真的想这样做，你可能需要编写自己的枚举元覆盖类。您可以看到implementation in cpython这个例子，这样您就可以为该枚举的值映射分配一个默认值，该值等于该值的第一个值。

Answer 3

如果不控制Enum S，然后执行：

my_val = list(MyEnum)[0] 

，或者，如果Enum可能是空的：

my_val = len(MyEnum) and list(MyEnum)[0] or None 

如果您确实控制了Enum s，然后添加一个get_default方法：

class MyEnum(Enum): 
    def get_default(self): 
     return self.A 

，或者更简单，让default是一个别名：

class MyEnum(Enum): 
    A = 0 
    B = 1 
    C = 2 
    default = A 

你也可以换第一种方法中的函数：

def default_member(enum): 
    return list(enum)[0]