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

Question

给定以下枚举:

class MyEnum(IntEnum):

    A = 0
    B = 1
    C = 2

如何指定默认值。我希望能够做到:

my_val = MyEnum()

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

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

Answer 1

falsetru's answer如果需要满足 0 参数情况，这将是可行的方法。

如果只是在给定值不存在时返回默认值，我们可以覆盖 Enum 类中的 _missing_ 钩子(Hook)(自 Python 3.6 起):

from enum import IntEnum


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

    @classmethod
    def _missing_(cls, value):
        return cls.A

assert MyEnum(0) is MyEnum.A
assert MyEnum(1) is MyEnum.B
assert MyEnum(-1) is MyEnum.A
assert MyEnum(None) is MyEnum.A