【Python】修改枚举的取值及链式调用

author: jwensh
date: 2023.02.11

枚举的取值及链式调用

需求背景

测试过程中需要很多参数化的设置及编程规范要求，希望修改数据不修改代码。最近测试区块链的时候有很多这样的需求，所以这里搞一下

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