Python的Enum类型通过巧妙结合魔术方法__setattr__和元类机制，实现了枚举成员的只读访问。本文将深入探讨这两个核心概念，揭示Enum如何在其元类EnumType中重写__setattr__，从而在类级别阻止对枚举成员的重新赋值，确保枚举类型的数据完整性和不可变性。
1. Python的魔术方法 (__setattr__)
Python语言提供了一系列特殊的“魔术方法”（Magic Methods），也常被称为“双下划线方法”或“dunders”，它们允许开发者自定义对象的行为。这些方法在特定操作发生时被Python解释器自动调用。例如，__str__方法在调用内置的str()函数时被触发。
其中一个关键的魔术方法是__setattr__(self, name, value)。当尝试给一个对象的属性赋值时，如果该对象定义了__setattr__方法，Python就会调用它来处理赋值操作。这使得我们可以在属性赋值前后执行自定义逻辑，例如验证、日志记录或阻止赋值。
以下是一个简单的示例，展示了__setattr__如何在实例级别工作：
立即学习“Python免费学习笔记（深入）”；
from typing import Any
class Talkative:
def __setattr__(self, name: str, value: Any) -> None:
"""
当尝试设置Talkative实例的属性时，此方法会被调用。
"""
print(f"尝试设置属性 {name!r} on {self!r}: {value!r}")
# 调用父类的__setattr__来实际完成属性赋值，否则属性将不会被设置
super().__setattr__(name, value)
# 创建Talkative类的实例
obj = Talkative()
# 尝试设置实例属性
obj.a = "xyz"
obj.b = -1输出：
尝试设置属性 'a' on <__main__.Talkative object at 0x...>: 'xyz' 尝试设置属性 'b' on <__main__.Talkative object at 0x...>: -1
从输出可以看出，每次为obj实例设置属性时，__setattr__方法都被成功调用。然而，需要注意的是，这种__setattr__的自定义行为仅适用于Talkative类的实例。如果尝试直接在Talkative类对象本身上设置属性，__setattr__将不会被触发：
Talkative.idea = "lightbulb" # 不会触发Talkative实例的__setattr__
这是因为Python中的类本身也是对象，它们是type类的实例。要控制类对象本身的属性设置行为，我们需要引入元类的概念。
2. 元类：类的类
在Python中，一切皆对象，包括类。一个类是其元类的一个实例。通常情况下，我们定义的类都是type类的实例，type是Python的默认元类。元类允许我们自定义类的创建过程，从而影响类对象本身的结构和行为。
为了在类级别上实现类似__setattr__的控制，我们需要定义一个自定义的元类。这个元类将继承自type，并在其中实现我们所需的魔术方法。
下面是一个自定义元类的示例，它在类对象上实现了__setattr__：
from typing import Any
class TalkativeType(type):
"""
自定义元类，用于控制类对象的属性设置。
"""
def __setattr__(cls, name: str, value: Any) -> None:
"""
当尝试设置使用此元类的类对象的属性时，此方法会被调用。
注意参数名为cls，表示操作的是类对象本身。
"""
print(f"尝试设置类属性 {name!r} on {cls!r}: {value!r}")
super().__setattr__(cls, name, value)
class Talkative(metaclass=TalkativeType):
"""
使用自定义元类TalkativeType的类。
"""
pass
# 尝试设置Talkative类对象的属性
Talkative.q = "bert"输出：
尝试设置类属性 'q' on <class '__main__.Talkative'>: 'bert'
在这个例子中，Talkative类通过metaclass=TalkativeType指定了它的元类。当尝试给Talkative类（而不是它的实例）设置属性q时，TalkativeType元类中的__setattr__方法被调用，从而允许我们在类级别上拦截并处理属性赋值操作。
3. Enum的实现机制：魔术方法与元类的结合
Python的enum模块正是利用了上述魔术方法和元类机制，来实现其枚举成员的只读特性。Enum类的定义中指定了一个名为EnumType的元类。这个EnumType元类正是阻止枚举成员被重新赋值的关键。
我们可以查看Python enum.py 模块的源码（例如，https://www.php.cn/link/ac5dd1eff7e0349bfd4b10e182577707），找到EnumType的定义：
class EnumType(type):
"""
Metaclass for Enum
"""
# ... 其他方法 ...
def __setattr__(cls, name, value):
"""
阻止重新赋值Enum成员。
"""
# 获取当前类的成员映射
member_map = cls.__dict__.get('_member_map_', {})
# 如果尝试设置的属性名是已存在的枚举成员
if name in member_map:
raise AttributeError(f'无法重新赋值成员 {name!r}')
# 否则，允许通过父类（type）的__setattr__进行赋值
super().__setattr__(cls, name, value)
# ...
class Enum(metaclass=EnumType):
"""
Base class for creating enumerated constants.
"""
# ...从源码中可以看出：
- Enum类通过metaclass=EnumType声明其元类为EnumType。
- EnumType元类重写了__setattr__方法。
- 在EnumType的__setattr__方法中，首先会检查待设置的属性name是否已经存在于cls（即当前的枚举类）的_member_map_中。_member_map_是一个内部字典，用于存储枚举成员的名称到其对应值的映射。
- 如果name是已存在的枚举成员，EnumType的__setattr__方法会立即抛出一个AttributeError，明确指出“无法重新赋值成员”。
- 如果name不是已存在的枚举成员（例如，尝试添加一个全新的类属性），则调用super().__setattr__(cls, name, value)，允许正常地在枚举类上设置该属性。
这种机制确保了一旦枚举成员被定义，它们就不能被修改或重新赋值，从而维护了枚举类型的不可变性和数据完整性。
总结与注意事项
通过深入理解Python的魔术方法__setattr__和元类机制，我们清晰地看到了Enum类型如何强制实现其成员的只读访问。
- 魔术方法__setattr__ 提供了在属性赋值时执行自定义逻辑的能力。
- 元类 允许我们在类级别上，而非仅仅实例级别，控制类的行为和属性操作。
- EnumType元类 结合了这两个概念，通过在其__setattr__方法中检查_member_map_并阻止对现有枚举成员的修改，成功地实现了Enum的只读特性。
这种设计模式不仅是Python语言强大和灵活性的体现，也为我们提供了在自定义类行为时，利用底层机制实现复杂特性的范例。在日常开发中，当我们定义需要不可变常量集合时，Enum无疑是最佳选择，而其背后的实现原理则进一步增强了我们对其行为的理解和信心。
