Python笔记-上下文管理器

# with 语句后面的结果对象，需要重写 __enter__ 和 __exit__方法
# 当进入到 with 代码块时，会自动调用 __enter__ 方法里的代码
# 当 with 代码块执行完成以后，会自动调用 __exit__ 方法

class Demo(object):
    def __enter__(self):
        print('__enter__方法被执行了')
        return self
    def __exit__(self, exc_type, exc_val, exc_tb):
        print('__exit__方法被调用了')

def creat_obj():
    x = Demo()
    return x

# y = creat_obj()       y = Demo()
# d = y.__enter__()     d = y.__enter__()
# print(y is d)
with creat_obj() as d:  # as 变量名
    # 变量 d 不是 creat_obj() 的返回结果
    # 它是创建的对象 y 调用 __enter__ 之后的返回结果
    print(d)

结果

__enter__方法被执行了
<__main__.Demo object at 0x018DDCF0>
__exit__方法被调用了

Process finished with exit code 0

try:
    file = open('xxx.txt','r')
except FileNotFoundError:
    print('文件不存在')
else:
    try:
        file.read()
    finally:
        file.close()


try:
    with open('xxx.txt','r') as file:
        file.read()     # 不需要手动的关闭文件
        # file.close()  # with关键字会帮我们关闭文件
except FileNotFoundError:
    print('文件未找到')
# with我们称之为上下文管理器