python 3.x 枚举

枚举Enumerate

• 枚举用来表示不同类型
• 枚举的本质是类
• 如果想让类称为枚举类型的话,该类需要继承Enum

 

枚举类和普通类相比的优势:

1. 枚举类型值不能被二次赋值
2. 枚举类型key不能重复

from enum import Enum


class VIP(Enum):
    YELLOW = 1
    GREEN = 2
    BLACK = 3
    RED = 4


# 输出结果: VIP.YELLOW,而不是值1
# 通常关注枚举的名字,名字相对而言是有意义的
print(VIP.YELLOW)
print(type(VIP.YELLOW))  # <enum 'VIP'>

# 获取枚举标签所对应的值 1
print(VIP.YELLOW.value)
print(type(VIP.YELLOW.value))  # <class 'int'>

# 获取枚举下面标签的名字 YELLOW
print(VIP.YELLOW.name)
print(type(VIP.YELLOW.name))  # <class 'str'>

print(VIP["YELLOW"])  # VIP.YELLOW

# 枚举可以被遍历
# 输出结果:
#       VIP.YELLOW
#       VIP.GREEN
#       VIP.BLACK
#       VIP.RED
for v in VIP:
    print(v)

 

枚举的比较运算 :

1. 两个枚举类型之间可以进行等值比较 如: VIP.YELLOW == VIP.GREEN
   • 枚举之间的等值比较,不仅仅比较值是否相等,而且判断在内存空间地址值是否相等.
2. 两个枚举类型之间不可以进行大小比较 如: VIP.YELLOW > VIP.GREEN 报错
3. 两个枚举类型之间可以进行身份比较 如 : VIP.YELLOW is VIP.GREEN

from enum import Enum


class VIP(Enum):
    YELLOW = 1
    GREEN = 2
    BLACK = 3
    RED = 4


class VIP1(Enum):
    YELLOW = 1
    GREEN = 2
    BLACK = 3
    RED = 4


# 枚举之间等值比较
print(VIP.YELLOW == VIP.BLACK)  # False
print(VIP.YELLOW == VIP.YELLOW)  # True
print(VIP.YELLOW == 1)  # False

# 枚举之间身份比较
print(VIP.YELLOW is VIP.GREEN)  # False
print(VIP.YELLOW is VIP.YELLOW)  # True

# 两个枚举类中的成员进行比较
# 枚举之间的等值比较,不仅仅比较值是否相等,而且判断在内存空间地址值是否相等.
print(VIP.YELLOW == VIP1.YELLOW)  # False

 

枚举注意事项

• 枚举不能实例化,因为枚举是用单例模式来实现的
• 枚举类中的成员名key不能重复,否则报错.
• 枚举类中的成员取值value可以相同,但是除第一个外,剩下都是第一个成员名的别名

from enum import Enum


class VIP(Enum):
    YELLOW = 1
    GREEN = 1
    BLACK = 3
    RED = 4
    BLUE = 1


# GREEN就是YELLOW的别名
print(VIP.GREEN)  # 输出结果是VIP.YELLOW而不是VIP.GREEN
print(VIP.BLUE)  # VIP.YELLOW

# 枚举类中的成员有三个取值相同,此时在遍历一次枚举VIP
# 输出结果: 第一个取值1的枚举被打印出来,而其他两个被忽略.
#           VIP.YELLOW
#           VIP.BLACK
#           VIP.RED
for v in VIP:
    print(v)

# 遍历枚举类VIP所有的成员,包括别名(取值相同的枚举成员) 使用枚举中提供的内置变量,来获取枚举中所有的成员.
# 输出结果:
#           YELLOW
#           GREEN
#           BLACK
#           RED
#           BLUE
for v in VIP.__members__:
    print(v)

# 输出结果:  使用VIP___member__.items()遍历枚举,返回结果是元组,第一个时枚举标签的名称,第二个是枚举标签及对应的一个取值.
#           ('YELLOW', <VIP.YELLOW: 1>)
#           ('GREEN', <VIP.YELLOW: 1>)
#           ('BLACK', <VIP.BLACK: 3>)
#           ('RED', <VIP.RED: 4>)
#           ('BLUE', <VIP.YELLOW: 1>)
for v in VIP.__members__.items():
    print(v)

 

 

枚举转换

• 枚举类名称(具体数值)  如  VIP(1) --> VIP.YELLOW

"""
    枚举转换
        在服务器端开发时,少不了与数据库打交道,如果想在数据库中存储枚举值,
        通常在数据库中存储的是枚举的具体数值,除了在数据库中存取枚举的具体数值用来代表枚举类型之外,
        还可以选择存储枚举的标签名,把标签的名字以字符串的形式存储到数据库中某一个字段中,
        用来代表某个枚举类型.建议使用枚举的具体数值,来存储枚举类型.因为数字更加的简洁,并且占用数据库的空间较小

    代码中分类,建议使用枚举标签来分类,便于代码的可读性.
"""
from enum import Enum

""" 需求 : 从数据库中获取的分类是具体的数字,将数字转换成代码中对应的枚举类型 """


class VIP(Enum):
    YELLOW = 1
    GREEN = 2
    BLACK = 3
    RED = 4
    BLUE = 5


# 将具体的数字转化成对应的枚举类型
# VIP(a)看上去像是强制类型转换,实际上不是,而是对应关系.
a = 1
print(VIP(a))  # VIP.YELLOW

b = 5
print(VIP(b))  # VIP.BLUE

 

""" 当一个类继承IntEnum类,那么该类下的所有枚举类型都是number类型或者可以强制转换成number类型的数据.
    枚举类中的具体数值,强制必须是int类型,包括可强转成数值int的类型,使用IntEnum,
    给枚举标签赋值其他类型具体值,报错
"""
from enum import IntEnum


class VIP(IntEnum):
    YELLOW = 1
    GREEN = 2
    BLACK = "str"  # 报错:ValueError: invalid literal for int() with base 10: 'str'
    RED = 4
    BLUE = 5


class VIP1(IntEnum):
    YELLOW = 1
    GREEN = 2
    BLACK = "3"  # 不会报错,因为"3"可以被强转转换成int类型
    RED = 4
    BLUE = 5

 

 如果想限制每一个不同的枚举类型不能取相同的值的方法:

• 将枚举类class上添加装饰器@unique来检查枚举中是否有赋值相同的枚举类型,有则报错.

""" 限制每一个枚举标签,不能取相同的值,否则报错,需要给枚举类添加装饰器@unique """
from enum import Enum, unique


# 不会报错
class VIP(Enum):
    YELLOW = 1
    GREEN = 1
    BLACK = 3
    RED = 4
    BLUE = 5


# 报错:
# ValueError: duplicate values found in <enum 'VIP'>: GREEN -> YELLOW
# 在<enum 'VIP'>中发现的重复值:GREEN ->YELLOW
@unique
class VIP1(Enum):
    YELLOW = 1
    GREEN = 1
    BLACK = 3
    RED = 4
    BLUE = 5

 

 

 

********