介绍
enum是一个用来枚举的模块创建枚举类型
import enum
# 创建一个类，继承自enum下的Enum
class Color(enum.Enum):
red = 1
green = 2
blue = 3
yellow = 4
pink = 5
cyan = 6
# 下面便可以通过名称直接获取成员
print(Color["red"], type(Color["red"])) # Color.red
print(Color.red, type(Color.red)) # Color.red
# 打印的是类型
# 那么如何获取里面的值呢？
# 调用name获取名称，调用value获取值
print(Color.red.name, type(Color.red.name)) # red
print(Color.red.value) # 1
# 也可以通过迭代来获取值
for c in Color:
print(c.name, c.value)
"""
red 1
green 2
blue 3
yellow 4
pink 5
cyan 6
""" 名称或者值相同
import enum
# 如果我定义了重复的key(仮)
try:
class Color(enum.Enum):
red = 1
green = 2
blue = 3
yellow = 4
pink = 5
cyan = 6
cyan = 5
except Exception as e:
print(e) # Attempted to reuse key: 'cyan'
# 提示我们key重复了import enum
# 如果我定义了重复的value(仮)
class Color(enum.Enum):
red = 1
green = 2
blue = 3
yellow = 4
pink = 5
cyan = 6
purple = 6
# 这是通过value来获取成员
# 获取成员可以通过Color.key | Color[key]的方式，还可以使用Color(value)的方式
print(Color(6).name) # cyan
print(Color.purple.name) # cyan
# 可以看到只打印了cyan
# 如果值相同的话，那么后者相当于前者的别名,两者指向的内容是一样的
for c in Color:
print(c)
"""
Color.red
Color.green
Color.blue
Color.yellow
Color.pink
Color.cyan
"""
# purple依旧没有打印
# 那如果我就想把值相同的也打印出来，要怎么做呢？
# 枚举内部有一个__members__属性，相当于一个字典
for c in Color.__members__.items():
print(c[0], c[1])
"""
red Color.red
green Color.green
blue Color.blue
yellow Color.yellow
pink Color.pink
cyan Color.cyan
purple Color.cyan
"""
# 第一个元素是key，str类型。第二个元素是枚举类型。
# 可以看到最后一个，即便key是purple，但是对应的枚举类型依旧是Color.cyan
for c in Color.__members__.items():
print(c[0], c[1], c[1].name, c[1].value)
"""
red Color.red red 1
green Color.green green 2
blue Color.blue blue 3
yellow Color.yellow yellow 4
pink Color.pink pink 5
cyan Color.cyan cyan 6
purple Color.cyan cyan 6
"""
# 最后一个打印的仍是cyan
# 成员之前也可以进行比较，但只支持is和==操作符，不支持大小比较
print(Color.green is Color.red) # False
print(Color.green is Color.green) # True
print(Color.cyan is Color.purple) # True
print(Color.cyan == Color.purple) # True成员的值是否可变
import enum
class Color(enum.Enum):
red = 1
green = 2
blue = 3
yellow = 4
pink = 5
cyan = 6
purple = 6
try:
Color.red.name = "xxx"
except Exception as e:
print(e) # can't set attribute
try:
Color.red.value = "xxx"
except Exception as e:
print(e) # can't set attribute
# 可以看到是无法赋值的
import enum
class Color(enum.Enum):
red = 1
green = 2
blue = 3
yellow = 4
pink = 5
cyan = []
purple = []
print(Color.cyan is Color.purple) # True
Color.cyan.value.append(123)
print(Color.cyan is Color.purple) # True
"""
可以看到，尽管无法重新赋值，但是本地修改还是可以的，如果是可变类型的话
但是两个还是一样的，因为purple是cyan的别名，当在创建这个类的时候，由于都是空列表，所以认为两者是一样
因此当我对cyan.value进行append的时候，会影响purple
"""
print(Color.cyan.value) # [123]
print(Color.purple.value) # [123]
Color.purple.value.append(456)
# 通过反过来也是一样的
print(Color.cyan.value) # [123, 456]
print(Color.purple.value) # [123, 456]强制让成员的值不一样
枚举类，key重复是不允许的，但是值重复是可以的，可如果我也不允许值重复呢？只需要加上一个装饰器即可
import enum
try:
@enum.unique
class Color(enum.Enum):
red = 1
green = 2
blue = 3
yellow = 4
pink = 5
cyan = []
purple = []
except Exception as e:
print(e) # duplicate values found in : purple -> cyan
# 提示我们purple和cyan的value重复了 比较成员的值
之前说过，枚举成员是不支持大小比较的，但如果我想支持呢？
import enum
# 换一种继承的类，改成IntEnum
class Color(enum.IntEnum):
red = 1
green = 2
blue = 3
yellow = 4
pink = 5
cyan = "6"
# 此时里面的成员要是int类型,或者能将字符串转成int
print(Color.red < Color.blue) # True
print(Color.red + Color.cyan) # 7