python - 继承上下文管理器的pythonic方式是什么

Question

Python 教我们使用 __enter__ 和 __exit__ 对对象进行清理。如果我需要创建一个使用对象的对象必须使用上下文管理器怎么办？想象一下:

from database1 import DB1
from database2 import DB2

通常，它们会这样使用:

with DB1() as db1, DB2() as db2:
    db1.do_stuff()
    db2.do_other_stuff()

无论发生什么，db1 和 db2 都将运行它们的 __exit__ 函数，并清理连接、刷新等。

当我将所有这些都放在一个类中时，我会怎么做？ 这对吗？ 这显然是不对的，db1 和 db2 的上下文管理器在 block 的末尾运行，如注释中所指出的.

class MyApp(object):
    def __enter__(self):
        with DB1() as self.db1, DB2() as self.db2:
            return self
    def __exit__(self, type, value, traceback):
        self.db1.__exit__(self, type, value, traceback)
        self.db2.__exit__(self, type, value, traceback)

我什至考虑过做这样的事情: 这看起来是个好主意，实际上(经过一些清理之后):

class MyApp(object):
    def __init__(self):
        self.db1 = DB1()
        self.db2 = DB2()
    def __enter__(self):
        self.db1.__enter__()
        self.db2.__enter__()
        return self
    def __exit__(self, type, value, traceback):
        try:
            self.db1.__exit__(self, type, value, traceback)
        except:
            pass
        try:
            self.db2.__exit__(self, type, value, traceback)
        except:
            pass

编辑:修复了代码。

Answer 1

我会选择第二种解决方案，但也会处理数据库错误:

import sys

class MyApp(object):
    def __init__(self):
        self.db1 = DB1()
        self.db2 = DB2()
    def __enter__(self):
        self.db1.__enter__()
        try:
            self.db2.__enter__()
        except:
            self.db1.__exit__(None, None, None) # I am not sure with None
            raise
        return self
    def __exit__(self, type, value, traceback):
        try:
            self.db1.__exit__(self, type, value, traceback)
        finally:
            self.db2.__exit__(self, type, value, traceback)

第一个在 __enter__ 中调用 __exit__ 因为 with - 所以，不起作用。

编辑:另请查看 answer by @Ming .在许多情况下，它更短。