默认情况下，如何让 Enum 中的元素评估为它们的值？

Question

Is there anyway I can have an Enum like below

class Key(Enum):
    CUSTOMER_ID = 'c_id'
    PHONE_NUMBER = 'phone'

such that calling Key.CUSTOMER_ID evaluates to 'c_id' by default, without the need of calling CUSTOMER_ID.value ?

Is there any method I can override that changes what the object returned when evaluated is?

Answer 1

When you want the Enum members to evaluate as their value, you need to mixin their value type:

class Key(str, Enum):             # note that `str` is before `Enum`
    CUSTOMER_ID = 'c_id'
    PHONE_NUMBER = 'phone'

and in use:

>>> Key.PHONE_NUMBER == 'phone'
True