Python上下文管理器怎样工作 - 问答

Python上下文管理器是一种特殊的对象，它允许你在执行代码块之前和之后执行一些操作

上下文管理器的工作原理是通过实现两个特殊的方法：__enter__() 和 __exit__()。当你使用 with 语句时，这两个方法会按照以下顺序被调用：

__enter__()：在执行 with 语句块中的代码之前调用。这个方法应该返回一个对象，这个对象将赋值给 as 子句中的变量（如果有的话）。
with 语句块中的代码：在这里执行你需要的操作。
__exit__(exc_type, exc_value, traceback)：在执行完 with 语句块中的代码之后调用。这个方法用于处理在 with 语句块中发生的异常。如果 __exit__() 方法返回 True，则异常会被忽略；如果返回 False 或者没有返回值，则异常会被重新抛出。

下面是一个简单的上下文管理器的例子，用于计算代码块的执行时间：

import time

class Timer:
    def __enter__(self):
        self.start = time.time()
        return self

    def __exit__(self, exc_type, exc_value, traceback):
        end = time.time()
        print(f"Elapsed time: {end - self.start} seconds")

with Timer() as t:
    # 在这里执行你需要的操作
    time.sleep(2)

在这个例子中，Timer 类实现了上下文管理器。当你使用 with Timer() as t: 语句时，__enter__() 方法会被调用，记录开始时间，并将 self 赋值给变量 t。然后执行 with 语句块中的代码（在这里是 time.sleep(2)）。最后，__exit__() 方法会被调用，计算并打印出经过的时间。

0 赞

0 踩