Python3中枚举仅含enum与bool字段的dataclass/NamedTuple全实例方案问询

首先明确：你不需要更换dataclass或NamedTuple的选型，这两类都完全适配你的场景。目前没有专门做该需求的独立成熟库，但可以基于Python标准库用极少量代码实现完全符合你预期的功能，不需要重复造复杂轮子。

通用实现方案 #

核心逻辑是用itertools.product生成所有字段可选值的笛卡尔积，配合标准库的类型自省接口获取字段的所有可选值，兼容dataclass和NamedTuple两种类型。
实现代码如下：

import itertools
from dataclasses import is_dataclass, fields
from enum import Enum
from typing import Type, TypeVar, Iterator

T = TypeVar('T')

class DataTable:
    def __init__(self, cls: Type[T]) -> None:
        self.cls = cls
        self.field_values = []
        # 处理dataclass类型
        if is_dataclass(cls):
            for f in fields(cls):
                if issubclass(f.type, Enum):
                    self.field_values.append(list(f.type))
                elif f.type is bool:
                    self.field_values.append([True, False])
                else:
                    raise TypeError(f"不支持的字段类型 {f.type}")
        # 处理NamedTuple类型
        elif hasattr(cls, "_fields") and hasattr(cls, "__annotations__"):
            for field_name in cls._fields:
                f_type = cls.__annotations__[field_name]
                if issubclass(f_type, Enum):
                    self.field_values.append(list(f_type))
                elif f_type is bool:
                    self.field_values.append([True, False])
                else:
                    raise TypeError(f"不支持的字段类型 {f_type}")
        else:
            raise TypeError("仅支持dataclass和NamedTuple作为入参")
    
    def __iter__(self) -> Iterator[T]:
        for item in itertools.product(*self.field_values):
            yield self.cls(*item)

使用示例 #

和你期望的调用方式完全一致：

from enum import Enum, auto
from typing import NamedTuple

class MyEnum(Enum):
    v1 = auto()
    v2 = auto()
    v3 = auto()

class MyStateDefinition(NamedTuple):
    a: MyEnum
    b: bool

table = DataTable(MyStateDefinition)
for item in table:
    print(item.a, item.b)

上述代码会遍历全部6个互不相等的实例，符合你的预期。

可选的第三方工具 #

如果你已有测试库hypothesis的依赖，可以直接用它的类型生成能力实现全量遍历，无需自行实现：

from hypothesis import given, strategies as st

# 强制全量遍历所有可能的实例
@given(st.from_type(MyStateDefinition).unique())
def process_state(state):
    # 你的业务逻辑
    print(state.a, state.b)

该方案仅适合测试场景，生产环境优先用标准库实现的版本，无额外依赖、性能更稳定。

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