with上下文管理

两种方式

• 上下文管理器本质就是能够支持with操作。

• 任何实现了 enter() 和 exit() 魔法方法的对象都可称之为上下文管理器，上下文管理器对象可以使用 with 关键字。显然，文件（file）对象也实现了上下文管理器协议。

第一方式

简单版的文件操作功能（非完美版）

# 上下文管理 器实现了__enter__ 与 __exit__这个两个魔法方法就是上下文管理器
# 上下文管理 作用就是可以使用with xxx


class MyOpen(object):
    def __init__(self):
        print("初始化")
        # 得到操作符
        f = open("./test1.py", 'r')
        self.f = f

    # 这个返回操作对象f
    def __enter__(self):
        print("enter打印了")
        return self.f

    # 退出以后
    def __exit__(self, exc_type, exc_val, exc_tb):
        print("exit打印了")
        # 关闭资源
        self.f.close()


with MyOpen() as f:
    content = f.read()

print(content)

结果

初始化    ======>  MyOpen（）创建时，__init__方法执行了
enter打印了   =====>  __enter__方法，把初始化的对象返回给了 f
exit打印了   =======>  执行到__exxit__方法，我们把文件关闭
abcd

文件操作功能（完美版）(实现传参功能)

# 上下文管理 器实现文件操作


class MyOpen(object):
    def __init__(self, file_path, mode):
        f = open(file_path, mode)
        self.f = f

    # 返回操作的对象
    def __enter__(self):
        return self.f

    # 关闭
    def __exit__(self, exc_type, exc_val, exc_tb):
        print("退出")
        self.f.close()


with MyOpen("./test1.py", 'r') as f:
    f.write("abcd")

第二种方式（写入地方有异常不会关闭）

from contextlib import contextmanager

# 使用装饰器
@contextmanager
def my_open(file_path, mode):
    # 初始化
    f = open(file_path, mode)
    
    # 返回操作对象
    yield f
    
    # 关闭
    print("退出")
    f.close()


with my_open("./test1.py", 'r') as f:
    f.write("abcd")

两种方法对比