Python中使用枚举类
开发中我们经常定义常量,其实有更好的方法:为这样的枚举类型定义一个class类型,然后,每个常量都是class的一个唯一实例。
Python中提供了Enum类来实现这个功能:
from enum import Enum
Numbers = Enum('Name', ('乔峰', '鸠摩智', '段誉', '阿紫', '虚竹','慕容复'))
for name, member in Numbers.__members__.items():
print(name, '=>', member, ',', member.value)
print(Numbers.慕容复.value) # 值默认从1 开始
print(Numbers['虚竹']) # 用数组下标的形式访问属性
print(Numbers(2)) # 直接使用索引形式访问
输出:
乔峰 => Name.乔峰 , 1
鸠摩智 => Name.鸠摩智 , 2
段誉 => Name.段誉 , 3
阿紫 => Name.阿紫 , 4
虚竹 => Name.虚竹 , 5
慕容复 => Name.慕容复 , 6
6
Name.虚竹
Name.鸠摩智
上面我们就获得了Numbers类型的枚举类,可以直接使用: Numbers.乔峰 来引用一个常量,或者枚举它的所有成员。
value属性则是自动赋给成员的int常量,默认从1开始计数。
可见,我们既可以用成员名称引用枚举常量,又可以直接根据value的值获得枚举常量。我们再看一个例子:
from enum import Enum
class Gender(Enum):
Male = "男"
Female = "女"
class Person(object):
def __init__(self, name, gender):
self.name = name
self.gender = gender
p = Person('乔峰', Gender.Male)
if p.gender == Gender.Male: # 输出 纯爷们
print('纯爷们!')
else:
print('俺娘说了!')