如何遍历Python Enum Flag的别名成员?
如何遍历Python Enum Flag的别名成员?
直接使用list(Color)只能获取非别名的Flag成员,无法拿到像WHITE这样的别名成员:
from enum import Flag class Color(Flag): RED = 1 BLUE = 2 GREEN = 4 WHITE = RED | BLUE | GREEN # <- 这个别名成员无法通过常规迭代获取 list(Color) # [<Color.RED: 1>, <Color.BLUE: 2>, <Color.GREEN: 4>] # 如何将Color.WHITE及其他别名包含在这个列表中?
根据官方文档说明:
__iter__(self):
返回所有包含的非别名成员
版本3.11变更:迭代时不再返回别名。
解决方案
要获取包括别名在内的所有Flag成员,直接访问枚举类的__members__属性即可,它是一个包含所有成员(别名和非别名)的有序字典,取其values()就能得到完整列表:
from enum import Flag class Color(Flag): RED = 1 BLUE = 2 GREEN = 4 WHITE = RED | BLUE | GREEN # 获取所有成员(含别名) all_members = list(Color.__members__.values()) print(all_members) # 输出: [<Color.RED: 1>, <Color.BLUE: 2>, <Color.GREEN: 4>, <Color.WHITE: 7>]
如果只需要筛选出别名成员,在Python 3.11及以上版本可以直接使用成员的is_alias属性判断:
# 筛选仅别名成员(Python 3.11+) alias_members = [member for member in Color.__members__.values() if member.is_alias] print(alias_members) # 输出: [<Color.WHITE: 7>]
内容的提问来源于stack exchange,提问作者teldios
相关产品推荐
相关产品推荐

