首页 > > 网络编程 > 其它 >

Python中使用枚举类

2019-04-11 10:20:49来源：博客园阅读 ()

开发中我们经常定义常量，

其实有更好的方法：为这样的枚举类型定义一个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('俺娘说了!')