如何在父ENUM类中添加BLANK成员？解决代码检查器报错问题

需求说明 #

给所有子类继承的父ENUM类ENUM_Blank添加BLANK成员，解决代码检查器的报错问题。当前父类包含FromStringToENUM类方法，该方法会在输入列表长度大于1时添加cls.BLANK；所有子类均继承该方法并自行定义了BLANK成员，代码运行正常，但检查器因父类未定义BLANK而报错。需要实现两个目标：

• 让代码检查器识别父类存在BLANK成员
• 强制所有子类必须包含BLANK成员（子类间无需支持比较）

现有问题代码 #

from enum import Enum

class ENUM_Blank(Enum):
    @classmethod
    def FromStringToENUM(cls, stringList: list[str]) ->list[ENUM_Blank]:
        fixedList = []
        # 原代码存在参数错误：stringConvert应为stringList
        if len(stringConvert) > 1: fixedList.append(cls.BLANK)
        [...]
        return fixedList

class COLOR(ENUM_Blank): # 继承方法
    BLANK = 'Blank'
    RED = 'Red'
    GREEN = 'Green'

class ANIMAL(ENUM_Blank): # 继承方法
    BLANK = 'Blank'
    CAT = 'Cat'
    DOG = 'Dog'

解决方案 #

通过在父类定义BLANK成员解决检查器报错，同时利用元类强制子类必须自行定义BLANK成员，避免子类直接继承父类的BLANK而不符合需求。

完整实现代码 #

from enum import Enum, EnumMeta

class BlankEnumMeta(EnumMeta):
    def __new__(metacls, clsname, bases, namespace):
        # 仅检查非父类的子类是否定义BLANK
        if 'BLANK' not in namespace and clsname != 'ENUM_Blank':
            raise TypeError(f"Enum子类 {clsname} 必须定义BLANK成员")
        return super().__new__(metacls, clsname, bases, namespace)

class ENUM_Blank(Enum, metaclass=BlankEnumMeta):
    BLANK = 'Blank'

    @classmethod
    def FromStringToENUM(cls, stringList: list[str]) -> list['ENUM_Blank']:
        fixedList = []
        if len(stringList) > 1:
            fixedList.append(cls.BLANK)
        # 补充字符串转枚举的核心逻辑（示例）
        for s in stringList:
            try:
                fixedList.append(cls(s))
            except ValueError:
                # 可选：处理无法转换的字符串
                pass
        return fixedList

class COLOR(ENUM_Blank):
    BLANK = 'Blank'
    RED = 'Red'
    GREEN = 'Green'

class ANIMAL(ENUM_Blank):
    BLANK = 'Blank'
    CAT = 'Cat'
    DOG = 'Dog'

效果说明 #

1. 解决检查器报错：父类ENUM_Blank明确定义了BLANK成员，代码检查器会识别cls.BLANK的合法性，不再报错。
2. 强制子类定义BLANK：如果子类未定义BLANK，元类BlankEnumMeta会在子类创建时抛出TypeError，确保所有子类都必须包含自己的BLANK成员。
3. 子类独立BLANK：子类的BLANK会覆盖父类的定义，保证每个子类的BLANK是自身枚举的成员，符合业务需求。

内容的提问来源于stack exchange，提问作者Ghosty.Arcane