下午小伙伴问了一个有趣的问题, 怎么用 Python 的 with 语句同时打开多个文件?
首先, Python 本身是支持同时在 with 中打开多个文件的
with open('a.txt', 'r') as a, open('b.txt', 'r') as b:
print(a.read())
print(b.read())
当然, 小伙伴的问题不可能这么简单, 他需要从打开一个列表中的所有文件, 而不是打开固定的一个文件, 这时候上面的语法显然就没法满足要求了. 幸运的是 with 语句是遵循协议的, 我们只要自定义出__enter__方法和__exit__方法就可以使用 with 语句打开自定义的类.比如我们定义以下类, 用于同时打开多个文件:
#!/usr/bin/env python
# coding: utf-8
class open_many:
def __init__(self, files=None, mode='r'):
if files is None:
self._files = []
else:
self._files = files
self.mode = mode
self.fds = [] # fd is short for file descriptor
def __enter__(self):
print('-->enter')
for f in self._files:
print('-->opening file')
self.fds.append(open(f, self.mode))
return self.fds
def __exit__(self, exc_type, exc_val, traceback):
print('-->exit')
for fd in self.fds:
print('-->closing file')
fd.close()
if exc_type == ValueError:
print('-->exception: ' + str(exc_val))
return True
if __name__ == '__main__':
print('')
with open_many(['a.txt', 'b.txt'], 'r') as files:
for f in files:
print f.read()
print('')
with open_many() as files:
raise ValueError('captured')
print('')
with open_many() as files:
raise Exception('uncaptureable')
运行结果如下:
其中__enter__方法会被进入 with 语句时调用, 返回值会作为 as 后面变量的值. __exit__方法的三个参数分别是发生的异常的类型, 异常变量, 和发生异常时的 traceback. 如果在__exit__中能够处理异常, 那么应该返回True, 否则应该返回 False. 需要注意的是, 并不需要再次把异常抛出. 这里处理的异常实在 with 语句块中发生的异常
显然为了使用 with 还需要定义一个类有点过于复杂了, Python 还提供了另一种方法, 使用 contextlib.contextmanager 装饰器和生成器来实现
#!/usr/bin/env python
# coding: utf-8
from contextlib import contextmanager
@contextmanager
def open_many(files=None, mode='r'):
if files is None:
files = []
try: #相当于__enter__
fds = []
for f in files:
fds.append(open(f, mode))
yield fds
except ValueError as e:
print(e)
finally: #相当于__exit__
for fd in fds:
fd.close()
if __name__ == '__main__':
with open_many(['a.txt', 'b.txt'], 'r') as files:
for f in files:
print(f.read())
此外, contextlib 中还提供了 closing 语句, 使用这个语句可以自动调用类的 close 方法.
#!/usr/bin/env python
# coding: utf-8
from contextlib import closing
class Connection:
def open(self):
print('opened')
def close(self):
print('closed')
if __name__ == '__main__':
with closing(Connection()) as con:
con.open()
# do stuff with con
如果觉得《pythonwith open 打开多个文件_Python中使用with语句同时打开多个文件》对你有帮助，请点赞、收藏，并留下你的观点哦！
