枚举转换
数据库里存类型,用数字会好点,字符串很别扭,而且易错
那么这就存在一个类型转换的问题:
其实转换也很简单,和 golang 差不多, T(value) 就可以把value转化为T
from enum import Enum
class VIP(Enum):
YELLOW = 1
RED = 2
BLACK = 3
PURPLE = 4
# 假设从数据库里取出的是1
a = 1
a = VIP(a)
if a == VIP.YELLOW:
print('黄钻')
elif a == VIP.RED:
print('红钻')
elif a == VIP.BLACK:
print('黑钻')
elif a == VIP.PURPLE:
print('紫钻')
黄钻