Python中文件处理时,通常会手动关闭文件,如下:
try:
#write data to here
fp = open('./test.txt')
fp.write('aaa\n')
finally:
file.close()
文本文件处理中,一个文件对象在退出程序之后会自动关闭,写入过的文件总是应该关闭,是因为Python可能会缓存(即出于效率的考虑而把临时文件存储在某处)写入的数据,如果程序因为某些原因崩溃了,那么数据根本就不会被写入文件。为了安全起见,需要在使用完文件后关闭。
事实上,专门有为这种情况而设计的语句,也就是要讲的with语句:
with open('./test.txt') as fp:
fp.write('aaa\n')
with语句的底层,其实是很通用的结构,允许使用上下文管理器(context manager)。上下文管理器实际是指一种支持__enter__和__exit__这两个方法的对象:
enter 方法是不带参数的,它会在进入with语句块时被调用,即将其返回值绑定到as关键字之后的变量上;
exit 方法则带有3个参数:异常类型、异常对象和异常回溯。在离开方法时这个函数会被调用(带有通过参数提供的,可以引发的异常)。如果__exit__返回的是false,那么所有的异常都将不被处理;
代码示例:
class OperateTest(object):
def open(self):
print("此处打开文件了!")
def close(self):
print("此处关闭文件了!")
def operate(self, txt_file):
print("文件处理中!%s" % txt_file)
def __enter__(self):
self.open()
return self
def __exit__(self, exc_type, exc_val, exc_tb):
self.close()
with OperateTest() as fp:
fp.operate("text_file")
执行返回结果:
此处打开文件了!
文件处理中!text_file
此处关闭文件了!
综上,文件是可以被用作上下文管理器。他们的__enter__方法返回文件本省对象,__exit__方法则关闭文件,也就是把我们的定义的方法,添加到系统默认调用的方法中进行二次封装,以此来达到简化代码的效果;