Python中如何将Enum类成员直接暴露到包层级访问？

不需要硬编码每个成员，可通过在包的__init__.py中动态遍历所有Enum类的成员，将其添加到包的顶层命名空间，同时保留原Enum类的定义。

具体实现步骤 #

1. 整理包结构 #

将Enum类统一放在单独的文件中（比如enums.py），包结构如下：

mypackage/
├── __init__.py
└── enums.py

2. 在enums.py中保留原Enum定义 #

from enum import Enum

class enum_class1(Enum):
    class1_sym1: int = 0
    class1_sym2: int = 1
    class1_sym3: int = 2

class enum_class2(Enum):
    class2_sym1: int = 0
    class2_sym2: int = 1
    class2_sym3: int = 2

3. 在__init__.py中动态导出成员 #

from enum import Enum
from .enums import *

# 遍历所有导入的对象，筛选自定义Enum子类
for name, obj in list(locals().items()):
    # 排除Enum基类本身，只处理我们定义的Enum类
    if isinstance(obj, type) and issubclass(obj, Enum) and obj is not Enum:
        # 将每个Enum成员添加到包的顶层命名空间
        for enum_member in obj:
            locals()[enum_member.name] = enum_member

效果验证 #

现在即可直接从包名访问Enum成员，同时原Enum类仍可正常访问：

import mypackage as p

# 直接访问成员（符合需求）
state = p.class2_sym1
print(state)  # 输出: enum_class2.class2_sym1

# 仍可通过原Enum类访问成员
print(p.enum_class1.class1_sym1)  # 输出: enum_class1.class1_sym1

注意事项 #

• 成员唯一性：确保所有Enum成员的name（如class1_sym1）在包内唯一，否则后导入的重名成员会覆盖前面的。
• 性能影响：动态遍历的开销极小，对包的导入速度几乎无影响。

内容的提问来源于stack exchange，提问作者Maxime Debarbat