上下文管理器
[with语句]的强大是因为有上下文管理器。
什么是上下文管理器?
一个类只要实现了__enter__()
和__exit__()
这两个方法,通过该类创建的对象,就称之为上下文管理器。
__exter__()
表示上文方法,需要返回一个操作文件对象__exit()
表示下文方法,with语句执行完成会自动执行,即使出现异常也会执行该方法
案例
需求:自定义上下文管理器类,模拟文件操作
# 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)