python常用模块 enum

Posted 2022-10-01 风流 少年

from enum import Enum, unique

# 枚举中的key不能相同，value可以相同，如果希望value不相同可以使用@unique来限制
# IntEnum 表示value值只能是整型数字或者整型字符串
@unique
class Weekday(Enum):
    monday = 1
    tuesday = 2
    wednesday = 3
    thursday = 4
    friday = 5
    saturday = 6
    sunday = 7


# 为枚举项添加其他属性，可以为同一个枚举项添加多个属性
Weekday.monday.label = '星期一'
Weekday.tuesday.label = '星期二'
Weekday.wednesday.label = '星期三'
Weekday.thursday.label = '星期四'
Weekday.friday.label = '星期五'
Weekday.saturday.label = '星期六'
Weekday.sunday.label = '星期日'


# 枚举可以使用==或者is关键字进行比较
monday = Weekday['monday']
monday2 = Weekday(1)

# <enum 'Weekday'> monday 1 星期一 True True
print(type(Weekday.monday),
      Weekday.monday.name,
      Weekday.monday.value,
      Weekday.monday.label,
      monday == monday2,
      monday is monday2)