python 上下文管理器

import time

class Timer:

    def __init__(self):
        self.res = 0

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

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.end = time.perf_counter()
        self.res = self.end - self.start
        return False

def main():

    # 上下文管理器，程序运行完自动执行 instence.close()
    with open("mydata.txt","w") as instence:
        instence.write("hello,this is a test text")

    # start = time.perf_counter()
    # nums = []
    # for n in range(1000000):
    #     nums.append(n ** 2)
    # end = time.perf_counter()
    # res = end - start
    # print(res)

    with Timer() as timer:
        nums = []
        for n in range(1000000):
            nums.append(n ** 2)
    print(timer.res)


if __name__ == "__main__":
    main()