Python枚举成员添加额外字段：实现多值互访的最优方案

需求明确 #

• 同一行定义枚举成员的标识符（如CAT）、索引值（如1）、自定义名称（如"Cat"）
• 不拆分枚举类，实现三者间的互访：
  • 通过自定义名称（如"Dog"）获取索引值（2）
  • 通过索引值（2）获取自定义名称（"Dog"）
  • 通过标识符（DOG）直接访问索引值和自定义名称
• 支持正常遍历枚举成员

符合Python风格的实现方案 #

利用Python原生enum.Enum的扩展能力，通过重写__init__方法解析成员的元组值，将索引值和自定义名称分别绑定为成员属性，再实现对应的类方法完成互访逻辑：

from enum import Enum, unique

@unique
class Animals(Enum):
    CAT     = 1, "Cat"
    DOG     = 2, "Dog"
    SHARK   = 3, "custom shark name"

    def __init__(self, value, label):
        # 将索引值绑定为成员的核心value属性
        self._value_ = value
        # 将自定义名称绑定为独立的label属性
        self.label = label

    @classmethod
    def get_value_by_label(cls, label):
        """通过自定义名称获取索引值"""
        for member in cls:
            if member.label == label:
                return member.value
        raise ValueError(f"No value found for label '{label}' in {cls.__name__}")

    @classmethod
    def get_label_by_value(cls, value):
        """通过索引值获取自定义名称"""
        try:
            return cls(value).label
        except ValueError:
            raise ValueError(f"No label found for value '{value}' in {cls.__name__}")

# 测试示例
if __name__ == "__main__":
    # 通过标识符直接访问
    print(Animals.DOG.value)          # 输出: 2
    print(Animals.DOG.label)          # 输出: Dog
    print(Animals.DOG.name)           # 输出: DOG（原生Enum的name属性为成员标识符）

    # 双向互访
    print(Animals.get_value_by_label("Dog"))  # 输出: 2
    print(Animals.get_label_by_value(2))      # 输出: Dog

    # 遍历所有枚举成员
    print("\n遍历结果:")
    for animal in Animals:
        print(f"标识符: {animal.name}, 索引值: {animal.value}, 自定义名称: {animal.label}")

关键说明 #

1. __init__方法的作用：
   原生Enum中如果成员赋值为元组，默认会把整个元组作为value。重写__init__后，我们将元组拆分，把第一个元素设为真正的索引值（self._value_是Enum的内置属性，用于存储成员的核心值），第二个元素作为自定义名称label。
2. @unique装饰器：确保所有成员的索引值唯一，避免重复冲突。
3. 互访逻辑优化：
   通过索引值获取自定义名称时，直接利用Enum原生的cls(value)语法获取成员，再取其label属性，比遍历更高效。
4. 原生属性保留：成员的name属性仍为定义时的标识符（如DOG），不破坏原生Enum的行为规范，用单独的label属性存储自定义名称更清晰。

扩展简化（可选） #

如果需要更简洁的调用方式，可以给类方法起短名称，或者实现__getitem__支持下标访问：

@classmethod
def __getitem__(cls, item):
    try:
        # 优先通过标识符获取
        return super().__getitem__(item)
    except KeyError:
        # 失败则尝试通过自定义名称获取成员
        for member in cls:
            if member.label == item:
                return member
        raise KeyError(f"No member found for '{item}' in {cls.__name__}")

之后就可以用Animals["Dog"]直接获取Animals.DOG成员。

内容的提问来源于stack exchange，提问作者fillo