如何优雅地处理类变量更改？

2024-03-24 11:17:37

优雅地处理类变量更改

在编写代码时，我们经常遇到这样的场景：需要在类变量更改时执行特定的动作。传统上，这可以通过为每个变量编写自定义装饰器或属性方法来实现。然而，当变量数量较多时，这种方法会变得繁琐且难以管理。

更简洁的方法

有几种更简洁的方法可以实现这一目标：

使用 setattr

setattr 是一种特殊方法，每当设置对象的属性时都会被调用。我们可以利用这一点来检查已更改的属性并相应地调用函数。以下是如何使用它：

class MyClass:
    def __init__(self, a, b, c):
        self._vars = {"a": a, "b": b, "c": c}
        self.func1_vars = {"a", "c"}
        self.func2_vars = {"b"}

    def __setattr__(self, name, value):
        if name in self._vars:
            if name in self.func1_vars:
                self.func1()
            elif name in self.func2_vars:
                self.func2()
            self._vars[name] = value
        else:
            raise AttributeError(f"No such attribute: {name}")

    def func1(self):
        print("Function 1 called")

    def func2(self):
        print("Function 2 called")

使用事件监听器

事件监听器允许我们订阅特定事件，例如变量更改事件。当事件发生时，将调用已注册的监听器函数。以下是使用事件监听器的示例：

import threading

class MyClass:
    def __init__(self, a, b, c):
        self.a = a
        self.b = b
        self.c = c
        self.listeners = {"a": [], "b": [], "c": []}

    def add_listener(self, event, listener):
        self.listeners[event].append(listener)

    def remove_listener(self, event, listener):
        self.listeners[event].remove(listener)

    def notify_listeners(self, event, *args, **kwargs):
        for listener in self.listeners[event]:
            listener(*args, **kwargs)

    def __setattr__(self, name, value):
        if name in self.__dict__:
            old_value = self.__dict__[name]
            self.__dict__[name] = value
            self.notify_listeners(name, old_value, value)
        else:
            raise AttributeError(f"No such attribute: {name}")

def func1(old_value, new_value):
    print(f"Variable a changed from {old_value} to {new_value}")

def func2(old_value, new_value):
    print(f"Variable b changed from {old_value} to {new_value}")