python上下文管理器基础

什么是上下文管理器：

上下文管理器（context manager）。一个上下文管理器是一个对象，它定义了一个运行时的上下文，使用with语句来执行。

代码示范：

 计算 1到1000000的平方，并将其添加到一个空列表的时间，需要多少秒。

import time

class Time():
    def __init__(self):
        self.timein = 0

    def __enter__(self):
        self.timestart = time.perf_counter()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.timeend = time.perf_counter()
        self.timein = self.timeend - self.timestart

with Time() as timer:
    num = []
    for i in range(1000000):
        i = i ** 2
        num.append(i)

print(timer.timein)

可以看出，需要0.3097244秒。那么具体是怎么执行的呢？

我们可以这样思考：

在 for循环还没开始前，用time模块获得一个时间段，将此时间段设置为0，然后执行循环程序。在循环结束，值都添加到列表中后，再获得一个时间段。

用后一个记录的时间 - 刚开始设定为0 的时间 = 所需要的时间。

最后打印，所需要的时间。

可以使用上下文管理器来设定：刚开始设定为0 的时间 和 用后一个记录的时间：

class Time():
    def __init__(self):
        self.timein = 0

    def __enter__(self):
        self.timestart = time.perf_counter()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.timeend = time.perf_counter()
        self.timein = self.timeend - self.timestart

 首先定义一个Time()类, 其中 

def __enter__(self):  # 作为上文的标志

def __exit__(self, exc_type, exc_val, exc_tb):  # 作为下文的标志。

如图中所示 ，写好开始时间 和结束时间。在exit中写出 结束时间 - 开始时间。

然后为循环体：

with Time() as timer:
    num = []
    for i in range(1000000):
        i = i ** 2
        num.append(i)

print(timer.timein)

注意：

用with 来引导 Time（）类 ， Time（）会在开始前，指向在class Time（）中的enter方法。

然后 将返回值指向给timer。然后执行循环体，执行完毕后调用exit方法。最后输出结果。

所以，一定一定要写return self  否则输出的结果为None。