如何使用上下文管理器批量包装Python类的所有方法

你写的代码存在不少硬伤，直接运行大概率出问题：

• 丢失返回值：内部的new_func调用原方法后没有返回执行结果，所有被包装的方法只要有返回值，最终都会返回None
• 调用时机不匹配：这个包装函数把self作为第一个参数，只能在实例生成之后（比如__init__方法内部）调用，根本没法在类定义末尾执行——类定义阶段还没有创建任何实例，不存在self对象
• 闭包逻辑有bug：循环遍历拿到的func是方法名字符串，你传给wrapper的是已经取到的方法对象，但内部又通过getattr(self, func)取方法，这里会触发循环变量晚绑定问题——所有包装后的方法执行时，都会拿到循环最后一轮的方法名，最终全部调用最后一个遍历到的方法
• 包装范围不可控：用dir(self)遍历会拿到实例所有可调用属性，包括继承来的方法、动态挂载的非类方法属性、甚至已经被包装过的方法，重复调用会导致上下文多层嵌套，很容易出现预期外的行为
• 异常风险：上下文管理器finally块下的路径切回逻辑缩进错误，极端异常场景下可能无法执行到切回原路径的逻辑，会打乱整个进程的工作目录

推荐用类装饰器实现，在类定义完成后统一在类层面完成方法包装，不需要每个实例初始化时重复执行遍历逻辑，性能更好也更容易维护：

import os
import contextlib
from functools import wraps

@contextlib.contextmanager
def set_directory(path):
    """上下文管理器：进入时切换到指定路径，退出时自动切回原工作目录"""
    origin = os.path.abspath(os.getcwd())
    try:
        os.chdir(path)
        yield
    finally:
        os.chdir(origin)

# 替换成你实际的根目录路径
ROOT = "/your/actual/root/path"

def wrap_chdir_context(exclude_methods=None):
    """类装饰器：为类的普通实例方法自动包装工作目录切换逻辑
    Args:
        exclude_methods: 不需要包装的方法名列表，默认跳过所有双下划线开头的特殊方法
    """
    if exclude_methods is None:
        exclude_methods = []
    
    def decorator(cls):
        for method_name, method in vars(cls).items():
            # 跳过特殊方法、指定排除的方法、非可调用对象、静态方法/类方法
            if (method_name.startswith("__")
                or method_name in exclude_methods
                or not callable(method)
                or isinstance(method, (staticmethod, classmethod))):
                continue
            
            @wraps(method)
            def wrapped(self, *args, **kwargs):
                target_path = os.path.join(ROOT, self.name)
                with set_directory(target_path):
                    # 必须返回原方法的执行结果
                    return method(self, *args, **kwargs)
            
            setattr(cls, method_name, wrapped)
        return cls
    return decorator

使用时直接在类定义上添加装饰器即可，不需要在类末尾额外写调用逻辑，还可以灵活指定不需要包装的方法：

@wrap_chdir_context(exclude_methods=["check_config"])
class YourClass:
    def __init__(self, name):
        self.name = name
    
    def foo(self):
        # 执行到这里时已经自动切换到 ROOT/self.name 目录
        print("当前工作目录：", os.getcwd())
    
    def bar(self, num):
        # 方法返回值可以正常拿到
        return num * 2

如果确实需要在实例层面动态控制包装逻辑，可以把类似的包装逻辑放到__init__方法中，但要注意用functools.wraps保留原方法元信息，同时不要用dir(self)遍历，避免误包装无关属性。

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