Python：上下文管理之 __enter__, __exit__

我们常用文件打开的时候用with语句来管理文件上下文

with open(somefile) as file:

 我们如何实现这种上下文管理的类呢

__enter__ 方法是一个在类中定义的特殊方法，用于在进入上下文管理器时执行一些操作。它通常与 with 语句一起使用，用于初始化资源或执行其他必要的操作。

以下是一个示例，展示了如何在类中使用 __enter__ 方法和 with 语句：

class MyResource:
    def __enter__(self):
        # 在进入上下文管理器时执行的操作
        print("Entering the context")

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 在离开上下文管理器时执行的操作
        print("Exiting the context")

# 使用 with 语句创建上下文管理器
with MyResource() as resource:
    # 在上下文中执行操作
    print("Performing some actions")

在上面的示例中，我们定义了一个名为 MyResource 的类，并在其中实现了 __enter__ 和 __exit__ 两个特殊方法。

在 __enter__ 方法中，我们定义了进入上下文管理器时要执行的操作，这里只是简单地打印一条消息。

在 __exit__ 方法中，我们定义了离开上下文管理器时要执行的操作，这里也只是简单地打印一条消息。

然后，我们使用 with 语句创建了一个上下文管理器，并将其赋值给 resource 变量。在 with 语句块中，我们可以执行一些操作，这些操作将在进入和离开上下文管理器时执行相应的操作。

当执行程序时，输出将如下所示：

Entering the context
Performing some actions
Exiting the context

通过使用 __enter__ 方法和 with 语句，可以确保在进入和离开上下文管理器时执行必要的操作，例如资源的初始化和清理。这样可以更方便地管理资源，并确保资源的正确使用。