Python枚举类问题：turn_right方法无法更新实例值的原因及解决

先看你给出的测试代码：

from enum import Enum
class Direction(Enum):
    NORTH = 0
    EAST = 1
    SOUTH = 2
    WEST = 3
    def turn_right(self):
        self = Direction((self.value + 1) % 4)

d = Direction.EAST
print(d)
d.turn_right()
print(d)
d.turn_right()
print(d)

问题根源：Enum实例是不可变的，且方法内的self是局部引用 #

Python的Enum类设计出来的实例就是不可变对象，一旦创建就不能修改其值。而且在类的方法里，self只是原实例的一个局部引用——你在turn_right里给self赋值新的Direction实例，只是改变了这个局部变量的指向，完全不会影响外部变量d原本指向的那个Enum实例。这就是为什么每次打印都是Direction.EAST的原因。

能不能实现原地修改？ #

很遗憾，原生Enum实例做不到真正的原地修改，因为这违背了Enum的不可变设计原则。不过我们可以换个思路，用两种方式保持你想要的d.turn_right()调用方式，同时达到预期效果：

方案1：用一个包装类持有Enum实例

我们可以创建一个包装类，内部存储Direction Enum的实例，然后给包装类实现turn_right方法来修改内部的引用，同时重写__repr__方法，让打印的时候和原Enum实例表现一致：

from enum import Enum

class Direction(Enum):
    NORTH = 0
    EAST = 1
    SOUTH = 2
    WEST = 3

class MutableDirection:
    def __init__(self, direction):
        self.direction = direction
    
    def turn_right(self):
        self.direction = Direction((self.direction.value + 1) % 4)
    
    def __repr__(self):
        return repr(self.direction)
    
    # 按需添加原Enum的特性，比如.value属性
    @property
    def value(self):
        return self.direction.value

# 调用方式和你的预期完全一致
d = MutableDirection(Direction.EAST)
print(d)  # 输出: Direction.EAST
d.turn_right()
print(d)  # 输出: Direction.SOUTH
d.turn_right()
print(d)  # 输出: Direction.WEST

方案2：放弃原生Enum，实现自定义可变枚举类

如果你不想用包装类，也可以自己实现一个类似Enum功能的可变类，这样就能直接在方法里修改实例的状态：

class Direction:
    NORTH = 0
    EAST = 1
    SOUTH = 2
    WEST = 3

    def __init__(self, value):
        self.value = value
    
    def turn_right(self):
        self.value = (self.value + 1) % 4
    
    def __repr__(self):
        # 模拟Enum的字符串表示格式
        name = next(k for k, v in vars(Direction).items() if v == self.value)
        return f"Direction.{name}"

# 调用方式不变
d = Direction(Direction.EAST)
print(d)  # 输出: Direction.EAST
d.turn_right()
print(d)  # 输出: Direction.SOUTH
d.turn_right()
print(d)  # 输出: Direction.WEST

关于返回新实例的替代方案 #

你提到的让turn_right返回新实例的方案其实是更符合Python Enum设计的做法，因为它遵循了不可变对象的使用习惯，代码也更简洁：

from enum import Enum
class Direction(Enum):
    NORTH = 0
    EAST = 1
    SOUTH = 2
    WEST = 3
    def turn_right(self):
        return Direction((self.value + 1) % 4)

d = Direction.EAST
print(d)  # Direction.EAST
d = d.turn_right()
print(d)  # Direction.SOUTH
d = d.turn_right()
print(d)  # Direction.WEST

这种方式更贴合Python的Immutable设计哲学，只是需要每次重新赋值变量。如果一定要保持d.turn_right()的调用方式（不需要重新赋值），那前面的包装类方案是更合适的选择。

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