手动调用 __enter__ 和 __exit__

我用谷歌搜索过但没有运气。所以让我们想象一下我有MySQL使用的连接器类__enter__ and __exit__函数（最初与with语句）来连接/断开与数据库的连接。

让我们有一个使用其中 2 个连接的类（例如用于数据同步）。注意：这不是我的现实生活场景，但这似乎是最简单的例子.

使其协同工作的最简单方法是这样的类：

class DataSync(object):

    def __init__(self):
        self.master_connection = MySQLConnection(param_set_1)
        self.slave_connection = MySQLConnection(param_set_2)

    def __enter__(self):
            self.master_connection.__enter__()
            self.slave_connection.__enter__()
            return self

    def __exit__(self, exc_type, exc, traceback):
            self.master_connection.__exit__(exc_type, exc, traceback)
            self.slave_connection.__exit__(exc_type, exc, traceback)

    # Some real operation functions

# Simple usage example
with DataSync() as sync:
    records = sync.master_connection.fetch_records()
    sync.slave_connection.push_records(records)

Q: 可以打电话吗（有什么问题吗）__enter__/__exit__像这样手动？

Pylint 1.1.0 没有对此发出任何警告，我也没有找到任何关于它的文章（开头的谷歌链接）。

那么调用呢：

try:
    # Db query
except MySQL.ServerDisconnectedException:
    self.master_connection.__exit__(None, None, None)
    self.master_connection.__enter__()
    # Retry

这是一个好/坏的做法吗？为什么？

你的第一个例子不是一个好主意：

1. 如果发生什么情况slave_connection.__enter__抛出异常：

   • master_connection获取其资源
   • slave_connection fails
   • DataSync.__enter__传播异常
   • DataSync.__exit__不运行
   • master_connection永远不会被清理干净！
   • 潜在的坏事

2. 如果发生什么情况master_connection.__exit__抛出异常？

   • DataSync.__exit__提前完成
   • slave_connection永远不会被清理干净！
   • 潜在的坏事

contextlib.ExitStack可以在这里提供帮助：

def __enter__(self):
    with ExitStack() as stack:
        stack.enter_context(self.master_connection)
        stack.enter_context(self.slave_connection)
        self._stack = stack.pop_all()
    return self

def __exit__(self, exc_type, exc, traceback):
    self._stack.__exit__(self, exc_type, exc, traceback)

问同样的问题：

1. 如果发生什么情况slave_connection.__enter__抛出异常：

   • 退出 with 块，并且stack清理master_connection
   • 一切都好！

2. 如果发生什么情况master_connection.__exit__抛出异常？

   • 没关系，slave_connection在调用之前被清理
   • 一切都好！

3. 好的，如果会发生什么slave_connection.__exit__抛出异常？

   • ExitStack确保致电master_connection.__exit__无论从属连接发生什么情况
   • 一切都好！

打电话没有什么问题__enter__直接调用它，但如果您需要在多个对象上调用它，请确保正确清理！