Python类中当acting属性为True时自动拦截以set开头的方法

需求场景 #

现有一个包含多个属性修改方法（以set_开头）的类MyClass，要求当实例的acting属性为True时，自动拦截所有set_开头的方法调用，拦截方式可选择抛出错误或返回False，且无需逐个修改目标方法或手动添加装饰器。

解决方案：重写__getattribute__方法 #

利用Python的__getattribute__魔法方法——它会在实例访问任何属性（包括方法）时触发，我们可以在这个方法中统一实现拦截逻辑。

完整代码示例 #

class MyClass:
    def __init__(self) -> None:
        self.acting = False
        self.att1 = None
        self.att2 = None
        self.att3 = None

    def set_att1(self, value):
        self.att1 = value

    def set_att2(self, value):
        self.att2 = value

    def set_att3(self, value):
        self.att3 = value

    def switch_acting(self):
        self.acting = not self.acting
    
    def __getattribute__(self, name):
        # 获取原本的属性/方法对象，避免递归调用
        original_attr = super().__getattribute__(name)
        
        # 判断是否需要拦截：以set_开头的可调用方法，且acting为True
        if name.startswith('set_') and callable(original_attr) and self.acting:
            # 定义拦截逻辑，这里选择抛出错误，也可改为返回False
            def intercepted_method(*args, **kwargs):
                raise PermissionError("当前处于acting状态，禁止修改属性")
            return intercepted_method
        
        # 无需拦截则返回原属性/方法
        return original_attr

拦截逻辑说明 #

1. __getattribute__会优先于属性访问被调用，通过super().__getattribute__(name)获取原本的属性或方法，避免触发递归调用。
2. 同时满足三个条件时触发拦截：方法名称以set_开头、是可调用的方法、实例的acting状态为True。
3. 符合拦截条件时，返回自定义的拦截函数替代原方法执行；不符合条件则直接返回原属性/方法。

测试验证 #

# 创建实例
obj = MyClass()

# 正常状态下可修改属性
obj.set_att1(10)
print(obj.att1)  # 输出：10

# 开启acting状态
obj.switch_acting()

# 尝试修改属性会触发拦截
try:
    obj.set_att2(20)
except PermissionError as e:
    print(e)  # 输出：当前处于acting状态，禁止修改属性
print(obj.att2)  # 输出：None

可选拦截方式：返回False #

如果不需要抛出错误，只需返回False，可将拦截函数修改为：

def intercepted_method(*args, **kwargs):
    return False

此时调用set_方法会直接返回False，原属性不会被修改。

内容的提问来源于stack exchange，提问作者Davi A. Sampaio