Python：无需全局变量或返回值，更新替换类实例的其他方法

问题背景 #

现有如下代码场景，希望在装饰器捕获到异常时，自动将main函数中的旧CLS实例替换为新实例，且不使用全局变量或返回实例的方式：

def main():
    # ... 其他逻辑
    cls = CLS()
    result = func1(cls, some_data)
    # 期望此处的 cls 已经是新实例，而非旧实例

@decorator
def func1(*args, **kwargs):
    # ... 函数逻辑

现有装饰器代码存在一些问题（比如While大小写错误、cls1未定义等），原始代码如下：

def decorator(func):
    @wraps(func)
    def wrapper(cls, *args, **kwargs):
        connection = cls.connection
        data = func(*args, **kwargs)
        while True: 
            try:
                result = connection(data)
            except:
                cls = CLS()
                connection = cls1.connection
        return result
    return wrapper

可行解决方案（无需全局变量/返回实例） #

方法1：用可变容器封装实例 #

Python中不可变类型的变量引用无法在函数内部直接修改外部绑定，但可变容器（如列表、字典）可以。我们可以把cls放在列表里传入：

from functools import wraps

class CLS:
    def __init__(self):
        self.connection = self._mock_connection
    
    def _mock_connection(self, data):
        # 模拟可能抛出异常的连接逻辑
        if hasattr(self, 'fail'):
            delattr(self, 'fail')
            raise ConnectionError("连接失败")
        return f"处理数据: {data}"

def decorator(func):
    @wraps(func)
    def wrapper(cls_container, *args, **kwargs):
        data = func(*args, **kwargs)
        while True: 
            cls = cls_container[0]
            try:
                result = cls.connection(data)
                break
            except ConnectionError:
                new_cls = CLS()
                cls_container[0] = new_cls  # 修改容器内的实例引用
        return result
    return wrapper

@decorator
def func1(some_data):
    return some_data

def main():
    cls = CLS()
    cls.fail = True  # 触发第一次连接失败
    cls_container = [cls]
    result = func1(cls_container, "测试数据")
    print(result)  # 输出：处理数据: 测试数据
    print(cls is cls_container[0])  # 输出：False，原实例已被替换

if __name__ == "__main__":
    main()

方法2：让CLS实例支持自我重置 #

如果可以修改CLS类，给它添加重置方法，异常时直接修改当前实例的状态，效果等价于替换实例：

from functools import wraps

class CLS:
    def __init__(self):
        self._reset_connection()
    
    def _reset_connection(self):
        # 初始化/重置连接状态
        self.connection = self._mock_connection
        self._fail_triggered = False
    
    def _mock_connection(self, data):
        if not self._fail_triggered:
            self._fail_triggered = True
            raise ConnectionError("连接失败")
        return f"处理数据: {data}"

def decorator(func):
    @wraps(func)
    def wrapper(cls, *args, **kwargs):
        data = func(*args, **kwargs)
        while True: 
            try:
                result = cls.connection(data)
                break
            except ConnectionError:
                cls._reset_connection()  # 重置当前实例状态
        return result
    return wrapper

@decorator
def func1(some_data):
    return some_data

def main():
    cls = CLS()
    result = func1(cls, "测试数据")
    print(result)  # 输出：处理数据: 测试数据
    print(cls._fail_triggered)  # 输出：False，说明实例状态已重置

if __name__ == "__main__":
    main()

方法3：直接更新实例的__dict__属性 #

如果无法修改CLS类，可直接用新实例的属性覆盖旧实例的__dict__，复用原对象内存地址：

from functools import wraps

class CLS:
    def __init__(self):
        self.connection = self._mock_connection
        self.instance_id = id(self)
    
    def _mock_connection(self, data):
        if self.instance_id == id(self):
            raise ConnectionError("连接失败")
        return f"处理数据: {data}"

def decorator(func):
    @wraps(func)
    def wrapper(cls, *args, **kwargs):
        data = func(*args, **kwargs)
        while True: 
            try:
                result = cls.connection(data)
                break
            except ConnectionError:
                new_cls = CLS()
                cls.__dict__.update(new_cls.__dict__)  # 用新实例属性覆盖旧实例
        return result
    return wrapper

@decorator
def func1(some_data):
    return some_data

def main():
    cls = CLS()
    original_id = id(cls)
    result = func1(cls, "测试数据")
    print(result)  # 输出：处理数据: 测试数据
    print(id(cls) == original_id)  # 输出：True，对象地址不变但属性已替换

if __name__ == "__main__":
    main()

注意事项 #

• 不要裸捕Exception，应捕获具体异常类型（如ConnectionError），避免隐藏未知错误。
• 方法1需修改调用处代码（将实例放入容器），但无需改动CLS类。
• 方法2的自我重置方式最优雅，但需要有权限修改CLS类实现。
• 方法3的__dict__更新方式对定义了__slots__的类无效，因为__slots__会禁止实例生成__dict__。

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