【Python】修改枚举的取值及链式调用
author: jwensh
date: 2023.02.11
文章目录
- 枚举的取值及链式调用
- 需求背景
- 1. enum.key 即获取值（而不是 enum.key.value）
- 2. 多级链式调用
枚举的取值及链式调用
需求背景
测试过程中需要很多参数化的设置及编程规范要求，希望修改数据不修改代码。最近测试区块链的时候有很多这样的需求，所以这里搞一下
1. enum.key 即获取值（而不是 enum.key.value）
-
因为我不要知道 key 是什么，code review 的时候我看到 key 即可知道其作用
-
使用元类进行继承
from enum import Enum, unique, EnumMetaclass EnumDirectValueMeta(EnumMeta):"""可以解决调用枚举属性时，由类型 enum 变成 string需要枚举类继承: Enum, metaclass=EnumDirectValueMeta"""def __getattribute__(cls, name):value = super().__getattribute__(name)if isinstance(value, cls):value = value.valuereturn value
- 实际的枚举
@unique
class Conservator(Enum, metaclass=EnumDirectValueMeta):ADDRESS_A = "0x0000000000000000000000000000000000000000"PRIVATE_KEY_A = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"ADDRESS_B = "0x0000000000000000000000000000000000000001"PRIVATE_KEY_B = "bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb"if __name__ == '__main__':print(Conservator.ADDRESS_A) # 0x0000000000000000000000000000000000000000
2. 多级链式调用
- 常用用于网络状态相关的定义
from enum import Enumclass DeviceStatus(Enum):UNCONNECTED = (10008, "Check if your device is connected properly")CURRT_PKG = (10001, "got mobile current running package name happened error")def __init__(self, code, msg):self.code = codeself.msg = msgif __name__ == '__main__':print(DeviceStatus.UNCONNECTED.code) #10008print(DeviceStatus.UNCONNECTED.msg) #Check if your device is connected properlyprint(DeviceStatus.CURRT_PKG.msg) # got mobile current running package name happened error