如何在Python中实现可链式堆叠设置的enum.Flag优雅封装类

解决方案 #

方案1：通用可复用的Flag链式构造器 #

这是最贴合需求的实现，不用为每个Flag子类重复写属性，新增标志位无需修改封装逻辑，也规避了property修改状态的问题：

import enum
from typing import TypeVar, Generic, Type

# 泛型支持任意Flag子类
F = TypeVar('F', bound=enum.Flag)

class FlagBuilder(Generic[F]):
    def __init__(self, flag_cls: Type[F]):
        self._flag_cls = flag_cls
        self._flags = flag_cls(0)

    def __getattr__(self, flag_name: str) -> "FlagBuilder[F]":
        # 自动匹配枚举成员，无需手动定义每个flag的属性
        self._flags |= getattr(self._flag_cls, flag_name)
        return self

    def build(self) -> F:
        # 返回原生Flag实例，业务函数直接接收原生枚举类型即可
        return self._flags


# -------------- 业务代码 --------------
class AvailableFlags(enum.Flag):
    flag1 = enum.auto()
    flag2 = enum.auto()
    flag3 = enum.auto() # 新增标志位无需修改Builder代码

# 快捷工厂函数，每次调用生成新的Builder实例，避免状态污染
def flags() -> FlagBuilder[AvailableFlags]:
    return FlagBuilder(AvailableFlags)

def func(setup_flags: AvailableFlags):
    # 可以直接用原生Flag的所有能力，比如判断标志位是否存在
    if AvailableFlags.flag1 in setup_flags:
        print("flag1 已启用")
    print(f"标志值：{setup_flags.value}")


if __name__ == "__main__":
    func(flags().flag1.build())
    func(flags().flag2.build())
    func(flags().flag1.flag2.build())
    func(flags().build())

这个实现的优势：

• 完全通用，所有enum.Flag子类都可以复用这个Builder
• 没有冗余代码，新增标志位只需要在枚举类里加一行定义
• 业务函数直接依赖原生枚举类型，耦合度极低，不需要兼容自定义封装类
• 每次调用flags()生成新实例，不会出现跨调用的状态污染

方案2：更轻量的原生语法简化 #

如果你能接受少量符号，不需要额外封装，原生Flag的位运算已经可以简化到很短：

# 导入时给枚举起短别名
from your_module import AvailableFlags as AF

def func(setup_flags: AF):
    print(setup_flags.value)

# 调用写法
func(AF.flag1)
func(AF.flag1 | AF.flag2)
func(AF(0))

Python 3.11+还支持直接用字符串构造Flag组合，写法更简洁：

func(AF("flag1|flag2"))

内容的提问来源于stack exchange，提问作者Wör Du Schnaffzig