如何通过内部类方法更新Python enum Flag实例内容？

核心原因 #

Python标准库的Flag（以及所有枚举类型）实例属于不可变对象，和整数、字符串特性一致：当你在方法中执行self ^= flag这类操作时，本质是创建了一个新的MyFlags实例，并将方法内部的局部变量self指向这个新实例，但外部变量（比如x）仍引用原来的旧实例，所以方法执行结束后外部看不到任何变化。

解决方案：用可变包装类封装Flag实例 #

既然Flag本身不可变，我们可以创建一个可变包装类，将Flag实例作为内部状态存储，以此实现类似set.update()的原地修改效果：

from enum import Flag, auto

class MyFlags(Flag):
    Null = auto()
    A = auto()
    B = auto()
    C = auto()
    D = auto()

    def get_flag_list(self):
        return [flag for flag in MyFlags if flag in self]

class MutableFlag:
    def __init__(self, flag_instance):
        if not isinstance(flag_instance, MyFlags):
            raise TypeError(f"必须传入MyFlags类型的实例")
        self._flag = flag_instance

    def update(self, given_flags):
        if not isinstance(given_flags, (MyFlags, MutableFlag)):
            raise TypeError(f"update方法需要传入MyFlags或MutableFlag类型的对象")
        # 统一处理传入的是Flag实例还是包装类实例
        target_flag = given_flags._flag if isinstance(given_flags, MutableFlag) else given_flags
        # 原地更新内部状态：合并所有给定的标志位
        self._flag |= target_flag

    def __repr__(self):
        return repr(self._flag)

    # 按需添加模拟Flag位运算的方法（可选）
    def __or__(self, other):
        if isinstance(other, (MyFlags, MutableFlag)):
            other_flag = other._flag if isinstance(other, MutableFlag) else other
            return MutableFlag(self._flag | other_flag)
        return NotImplemented

使用示例 #

# 创建可变包装实例
x = MutableFlag(MyFlags.A | MyFlags.B)
y = MyFlags.B | MyFlags.C | MyFlags.D

print(x)  # 输出: MyFlags.A|B
x.update(y)
print(x)  # 输出: MyFlags.A|B|C|D

补充说明 #

如果不想额外引入包装类，另一种思路是让update方法返回新实例并重新赋值（即x = x.update(y)），但这不符合你“避免返回值赋值”的需求。包装类是最贴合需求的方案，既保留了Flag的原有功能，又实现了原地修改的特性。

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