上下文管理器

[with语句]的强大是因为有上下文管理器。

什么是上下文管理器?

一个类只要实现了__enter__()__exit__()这两个方法,通过该类创建的对象,就称之为上下文管理器。

案例

需求:自定义上下文管理器类,模拟文件操作

# 1. 定义一个File类
class File(object):
    def __init__(self, file_name, file_model):
        self.file_name = file_name
        self.file_mode = file_model
# 2. 实现__enter__()和__exit__()方法
    def __enter__(self):
        print('这是上文')
        self.file = open(self.file_name, self.file_mode)
        return self.file

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('这是下文')
        self.file.close()
# 3. 然后使用with语句完成操作文件
with File('1.txt', 'r') as f:
    file_data = f.read()
    print(file_data)