Resolved: How to wrap each method in a class with a context manager?

def wrap_all_methods_in_class_with_chdir_contextmanager(self):
    @contextlib.contextmanager
    def set_directory(path):
        """Sets the cwd within the context
    
        Args:
            path (Path): The path to the cwd
    
        Yields:
            None
        """
    
        origin = os.path.abspath(os.getcwd())
        try:
            os.chdir(path)
            yield
        finally:
                os.chdir(origin)
    
    def wrapper(func):
        def new_func(*args, **kwargs):
            with set_directory(f"{ROOT}/{self.name}"):
                getattr(self,func)(*args, **kwargs)
        return new_func
            
    for func in [func for func in dir(self) if callable(getattr(self, func)) and not func.startswith('__')]:
        setattr(self,func,wrapper(getattr(self,func)))

import contextlib
import os
from functools import wraps
from types import FunctionType

ROOT = "/base"

@contextlib.contextmanager
def set_directory(path):
    """
    Sets the cwd within the context

    Args:
        path (Path): The path to the cwd

    Yields:
        None
    """
    origin = os.path.abspath(os.getcwd())
    try:
        os.chdir(path)
        yield
    finally:
        os.chdir(origin)

def use_chdir_context(f):
    @wraps(f)
    def wrapped(self, *args, **kwargs):
        with set_directory(f"{ROOT}/{self.path}"):
            return f(self, *args, **kwargs)
    return wrapped


class ChdirContextMetaclass(type):
    """
    Wraps all methods in the class with use_chdir_context decorator
    """
    def __new__(metacls, name, bases, class_dict):
        new_class_dict = {}
        for attr_name, attr in class_dict.items():
            if isinstance(attr, FunctionType):
                attr = use_chdir_context(attr)
            new_class_dict[attr_name] = attr
        return type.__new__(metacls, name, bases, new_class_dict)


class Hi(metaclass=ChdirContextMetaclass):
    path = "hi/1/2/3"

    def lovely(self):
        print(os.getcwd())

def use_chdir_context(f):
    @wraps(f)
    def wrapped(self, *args, **kwargs):
        origin = os.path.abspath(os.getcwd())
        try:
            os.chdir(f"{ROOT}/{self.path}")
            return f(self, *args, **kwargs)
        finally:
            os.chdir(origin)
    return wrapped