Python中如何自定义Enum枚举类的报错信息并展示合法取值？

可以实现该需求，最简单的方式是重写Enum类的__missing__钩子方法，这是Python官方为Enum类预留的、专门用于处理非法枚举值场景的方法，不需要修改原有调用逻辑，代码侵入性极低。

实现代码 #

单枚举类快速实现 #

如果只需要给单个枚举类加该效果，可以直接在类内实现__missing__方法：

from enum import Enum

class SomeType(Enum):
    TYPEA = 'type_a'
    TYPEB = 'type_b'
    TYPEC = 'type_c'

    @classmethod
    def __missing__(cls, value):
        # 提取所有合法取值，统一格式化加引号
        valid_values = [repr(item.value) for item in cls]
        raise ValueError(f"{repr(value)} is not a valid {cls.__name__}. Valid types are {', '.join(valid_values)}.")

多枚举类复用实现 #

如果有多个枚举类都需要该效果，可以封装通用的基类，后续所有枚举类继承该基类即可自动获得自定义报错能力：

from enum import Enum

class BaseEnum(Enum):
    """通用带非法值提示的枚举基类"""
    @classmethod
    def __missing__(cls, value):
        valid_values = [repr(item.value) for item in cls]
        raise ValueError(f"{repr(value)} is not a valid {cls.__name__}. Valid types are {', '.join(valid_values)}.")

# 业务枚举类直接继承即可
class SomeType(BaseEnum):
    TYPEA = 'type_a'
    TYPEB = 'type_b'
    TYPEC = 'type_c'

class OtherBizType(BaseEnum):
    OPT1 = 'opt1'
    OPT2 = 'opt2'

效果验证 #

执行SomeType('type_o')时，就会抛出预期的报错：

ValueError: 'type_o' is not a valid SomeType. Valid types are 'type_a', 'type_b', 'type_c'.

内容的提问来源于stack exchange，提问作者Cleb