python - 对类的所有成员调用 __exit__

Question

是否有一种 Pythonic 的方式来自动 __exit__ 一个类的所有成员？

class C:
    def __init__(self):
        self.a = open('foo')
        self.b = open('bar')

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_value, traceback):
        # Is it correct to just forward the parameters here?
        self.a.__exit__(self, exc_type, exc_value, traceback)
        self.b.__exit__(self, exc_type, exc_value, traceback)

我可以不在 a 和 b 上手动调用 __exit__ 来执行此操作吗？我是否正确地调用了 __exit__？

假设我拥有的资源不是示例中的文件，并且没有close 或destroy 之类的方法.在 __enter__ 和 __exit__ 之上实现这样的方法可能是一种好的做法吗？

Answer 1

正如我在评论中提到的:

I think most helpfull here will be : contextlib.ExitStack

您可以在 __init__ 中将此对象创建为您自己的类的成员。然后使用 enter_context(cm) 添加所有依赖的上下文管理器，其中 cm 是 context_manager

def __init__(self):
    self.exit_stack = contextlib.ExitStack()
    self.exit_stack.__enter__()
    self.exit_stack.enter_context(open('foo')) 
    ....

要清除所有依赖的上下文，只需调用 __exit__ 这个堆栈的导出。

或者更好的做法是继承 ExitStack 并在 init 中调用 enter_context。