如何为Python Flag枚举类添加自定义属性，替代字典映射？

问题背景 #

已知普通Enum类型可通过元组和__init__方法携带自定义属性，示例如下：

from enum import Enum

class ActivityType(Enum):
    NEXT = 0, "N"
    JOIN = 3, "J"
    DIVIDE_REAR = 1, "DR"
    DIVIDE_FRONT = 2, "DF"

    def __init__(self, xml_code, label):
        self.xml_code = xml_code
        self.label = label

assert ActivityType.JOIN.label == "J"
assert ActivityType.DIVIDE_FRONT.xml_code == 2

现在需要为可组合的Flag枚举PowerType实现类似功能：给每个成员绑定自定义子码，按成员定义顺序组合子码生成XML编码。当前通过字典映射实现，但希望去掉字典，直接通过成员属性完成。

尝试的写法因将元组作为成员值，导致auto()处理异常、位运算失效：

from enum import Flag, auto

class PowerType(Flag):
    NONE = 0, ""
    AC_OVERHEAD = auto(), "O"
    DC_3RAIL = auto(), "3"
    DIESEL = auto(), "D"

    def __init__(self, flag, xml_code):
        self.flag = flag
        self._xml_subcode = xml_code

    def xml_code(self):
        return "".join(pt._xml_subcode for pt in PowerType if pt & self)

解决方案 #

核心要点：Flag枚举的成员值必须是合法整数（2的幂次，保证位运算正常），自定义属性通过__init__方法单独初始化，不能将元组直接作为成员值。

方法1：手动指定Flag数值（直观易维护） #

手动为每个成员指定2的幂次作为数值，元组第二个元素为自定义子码，通过__init__初始化属性：

from enum import Flag

class PowerType(Flag):
    NONE = 0, ""
    AC_OVERHEAD = 1, "O"
    DC_3RAIL = 2, "3"
    DIESEL = 4, "D"

    def __init__(self, value, xml_subcode):
        self._value_ = value  # 显式绑定枚举数值，确保位运算正常
        self._xml_subcode = xml_subcode

    def xml_code(self):
        # 按成员定义顺序筛选并拼接子码
        return "".join(pt._xml_subcode for pt in PowerType if pt & self)

# 验证逻辑
assert (PowerType.DIESEL | PowerType.DC_3RAIL).xml_code() == "3D"
assert (PowerType.AC_OVERHEAD | PowerType.DC_3RAIL).xml_code() == "O3"
assert PowerType.NONE.xml_code() == ""

方法2：使用auto()配合重写_generate_next_value_（灵活扩展） #

如果不想手动计算2的幂次，可重写_generate_next_value_方法自动生成合法的Flag数值，同时保留自定义子码：

from enum import Flag, auto

class PowerType(Flag):
    NONE = 0, ""
    AC_OVERHEAD = auto(), "O"
    DC_3RAIL = auto(), "3"
    DIESEL = auto(), "D"

    @classmethod
    def _generate_next_value_(cls, name, start, count, last_values):
        # 生成2的幂次作为Flag数值，忽略auto()的占位值
        return 1 << count if count > 0 else 0

    def __init__(self, _, xml_subcode):
        # 忽略元组中被替换的auto()占位值，仅初始化子码属性
        self._xml_subcode = xml_subcode

    def xml_code(self):
        return "".join(pt._xml_subcode for pt in PowerType if pt & self)

# 验证逻辑
assert (PowerType.DIESEL | PowerType.DC_3RAIL).xml_code() == "3D"
assert (PowerType.AC_OVERHEAD | PowerType.DC_3RAIL).xml_code() == "O3"

关键说明 #

• Flag枚举的核心是位运算，因此成员值必须是整数（2的幂次），不能用元组直接作为值，否则会破坏位运算逻辑。
• 两种方法均通过__init__将自定义子码绑定到成员属性，避免了字典映射的冗余。

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