如何为枚举对象实现增量函数？ [复制]

Question

我正在使用 Dyson 风扇并正在尝试编写一个增加风扇速度的函数。枚举对象 FanSpeed 具有以下成员。当我执行 [print(i) for i in FanSpeed] 时，我得到：

FanSpeed.FAN_SPEED_1
FanSpeed.FAN_SPEED_2
FanSpeed.FAN_SPEED_3
FanSpeed.FAN_SPEED_4
FanSpeed.FAN_SPEED_5
FanSpeed.FAN_SPEED_6
FanSpeed.FAN_SPEED_7
FanSpeed.FAN_SPEED_8
FanSpeed.FAN_SPEED_9
FanSpeed.FAN_SPEED_10

所以第一个枚举对象（上图）的名称为 FAN_SPEED_1 (FanSpeed.FAN_SPEED_1.name)，值为“0001”(FanSpeed.FAN_SPEED_1.value)，依此类推。

设置速度的函数（本例为5）：

fan.set_configuration(fan_mode=FanMode.FAN, fan_speed=FanSpeed.FAN_SPEED_5)

我不知道如何编写一个获取当前速度然后将其设置为更高级别的函数。我试过了：

1/ 将枚举对象视为字符串，并将最后一个字符替换为更高的数字（字符类型）。但这种方法不适用于枚举对象。

2/ 寻找某种增量或下一个函数，但枚举对象似乎不存在这些函数。

Answer 1

让我们使用 Python 文档中的示例：

from enum import Enum
class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3

current_color = Color.RED
new_color = Color(current_color.value + 1)
>>> new_color == Color.GREEN
True

谢谢，我还没有玩过 Enum 类型，所以这是一次很好的学习体验。

如果您想对非 int 值执行此操作，您可以执行以下操作：

class Color(Enum):
    red = "0001"
    blue = "0002"


def get_next_color(current_color):
    current_color_idx = list(Color).index(current_color)
    next_color_idx = (current_color_idx + 1) % len(Color)
    next_color = list(Color)[next_color_idx]
    return next_color