如何在对象引用超出作用域时自动触发tidy-up清理方法？

针对你的需求，有两种更符合Python风格的方案，比手动写包装类更简洁可靠：

1. 上下文管理器（Context Manager）+ with语句 #

利用Python的上下文管理器协议，显式定义资源的作用域，退出作用域时自动执行清理逻辑。这种方式符合Python「显式优于隐式」的设计原则，可读性极强。

class ResourceGuard:
    def __init__(self, obj):
        self.obj = obj
        # 自动判断对象是否需要清理（检查是否有tidyup方法）
        self._needs_cleanup = hasattr(obj, 'tidyup')
    
    def __enter__(self):
        # 返回原始对象，使用时完全透明
        return self.obj
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        # 退出with块时自动执行清理（如果需要）
        if self._needs_cleanup:
            self.obj.tidyup()

# 使用示例
with ResourceGuard(get_thing(id)) as thing:
    # 在with块内使用thing，逻辑和直接用get_thing返回的对象完全一致
    thing.process_data()
# 退出with块后，自动调用thing.tidyup()（如果存在）

优势：

• 作用域清晰，一眼就能看到资源的生命周期范围
• 即使代码块内抛出异常，__exit__仍然会执行，保证资源清理
• 无需包装对象，原始对象的所有方法、属性都能直接使用

2. weakref.finalize 绑定垃圾回收终结器 #

如果不想改变调用方式（不需要with语句），可以用weakref.finalize给对象绑定一个清理函数，当对象被Python垃圾回收器回收时，自动执行清理逻辑。

import weakref

def get_managed_thing(id):
    thing = get_thing(id)
    # 仅当对象有tidyup方法时，绑定终结器
    if hasattr(thing, 'tidyup'):
        # 使用弱引用避免阻止对象被回收
        weakref.finalize(thing, thing.tidyup)
    return thing

# 使用示例
thing = get_managed_thing(id)
# 正常使用thing
thing.do_work()
# 当thing超出作用域或被del删除，GC回收时自动调用tidyup()

优势：

• 完全透明，调用方式和直接用get_thing完全一致
• 不需要额外包装类，代码更简洁
• 自动适配不同对象的清理逻辑（无tidyup方法的对象不会被绑定）

对你现有包装类的改进建议 #

如果你坚持使用包装类的方案，可以优化__setattr__的实现，避免将包装类自身的属性设置转发给原始对象：

def get_wrapped_thing(id):
    class TidyUp:
        def __init__(self, obj):
            # 直接用super().__setattr__设置自身属性，避免触发自定义__setattr__
            super().__setattr__('obj', obj)
            super().__setattr__('_needs_cleanup', hasattr(obj, 'tidyup'))
        
        def __getattr__(self, attr):
            return getattr(self.obj, attr)
        
        def __setattr__(self, attr, value):
            # 区分包装类自身属性和原始对象的属性
            if attr in ['obj', '_needs_cleanup']:
                super().__setattr__(attr, value)
            else:
                setattr(self.obj, attr, value)
        
        def __del__(self):
            if self._needs_cleanup:
                self.obj.tidyup()

    thing = get_thing(id)
    return TidyUp(thing)

方案选择建议 #

• 优先选上下文管理器：适合需要严格控制资源生命周期的场景，可读性和可靠性最高
• 其次选weakref.finalize：适合希望完全透明、不改变原有调用习惯的场景
• 包装类方案：适合需要对对象进行额外封装的复杂场景，但代码复杂度更高

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