我想一次更改几个文件,如果我可以写入所有文件。我想知道我是否可以将多个打开的调用与with语句结合起来:
try:
with open('a', 'w') as a and open('b', 'w') as b:
do_something()
except IOError as e:
print 'Operation failed: %s' % e.strerror
如果这是不可能的,那么这个问题的优雅解决方案是什么样子的?
我想一次更改几个文件,如果我可以写入所有文件。我想知道我是否可以将多个打开的调用与with语句结合起来:
try:
with open('a', 'w') as a and open('b', 'w') as b:
do_something()
except IOError as e:
print 'Operation failed: %s' % e.strerror
如果这是不可能的,那么这个问题的优雅解决方案是什么样子的?
当前回答
嵌套语句也会起到同样的作用,而且在我看来,处理起来更简单。
假设您有inFile.txt,并希望同时将其写入两个outFile。
with open("inFile.txt", 'r') as fr:
with open("outFile1.txt", 'w') as fw1:
with open("outFile2.txt", 'w') as fw2:
for line in fr.readlines():
fw1.writelines(line)
fw2.writelines(line)
编辑:
我不明白否决的原因。我在发布我的答案之前测试了我的代码,它按预期工作:正如问题所问,它会写入所有outFile。无重复书写或未能书写。所以我很想知道为什么我的答案被认为是错误的、次优的或类似的。
其他回答
对于python 2.6,它将无法工作,我们必须使用以下方式打开多个文件:
with open('a', 'w') as a:
with open('b', 'w') as b:
从Python 2.7(或3.1)开始,您可以编写
with open('a', 'w') as a, open('b', 'w') as b:
do_something()
(历史注释:在早期版本的Python中,有时可以使用contextlib.nested()来嵌套上下文管理器。但是,对于打开多个文件来说,这不会像预期的那样奏效——请参阅链接文档以了解详细信息。)
在罕见的情况下,您希望同时打开可变数量的文件,可以使用contextlib.ExitStack,从Python版本3.3开始:
with ExitStack() as stack:
files = [stack.enter_context(open(fname)) for fname in filenames]
# Do something with "files"
请注意,更常见的情况是,您希望按顺序处理文件,而不是同时打开所有文件,特别是如果您有可变数量的文件:
for fname in filenames:
with open(fname) as f:
# Process f
嵌套语句也会起到同样的作用,而且在我看来,处理起来更简单。
假设您有inFile.txt,并希望同时将其写入两个outFile。
with open("inFile.txt", 'r') as fr:
with open("outFile1.txt", 'w') as fw1:
with open("outFile2.txt", 'w') as fw2:
for line in fr.readlines():
fw1.writelines(line)
fw2.writelines(line)
编辑:
我不明白否决的原因。我在发布我的答案之前测试了我的代码,它按预期工作:正如问题所问,它会写入所有outFile。无重复书写或未能书写。所以我很想知道为什么我的答案被认为是错误的、次优的或类似的。
只需将和替换为,即可完成:
try:
with open('a', 'w') as a, open('b', 'w') as b:
do_something()
except IOError as e:
print 'Operation failed: %s' % e.strerror
从Python 3.3开始,您可以使用contextlib模块中的ExitStack类来安全地打开任意数量的文件。
它可以管理动态数量的上下文感知对象,这意味着如果您不知道要处理多少文件,它将被证明特别有用。
事实上,文档中提到的规范用例是管理动态数量的文件。
with ExitStack() as stack:
files = [stack.enter_context(open(fname)) for fname in filenames]
# All opened files will automatically be closed at the end of
# the with statement, even if attempts to open files later
# in the list raise an exception
如果您对详细信息感兴趣,下面是一个通用示例,以解释ExitStack是如何操作的:
from contextlib import ExitStack
class X:
num = 1
def __init__(self):
self.num = X.num
X.num += 1
def __repr__(self):
cls = type(self)
return '{cls.__name__}{self.num}'.format(cls=cls, self=self)
def __enter__(self):
print('enter {!r}'.format(self))
return self.num
def __exit__(self, exc_type, exc_value, traceback):
print('exit {!r}'.format(self))
return True
xs = [X() for _ in range(3)]
with ExitStack() as stack:
print(len(stack._exit_callbacks)) # number of callbacks called on exit
nums = [stack.enter_context(x) for x in xs]
print(len(stack._exit_callbacks))
print(len(stack._exit_callbacks))
print(nums)
输出:
0
enter X1
enter X2
enter X3
3
exit X3
exit X2
exit X1
0
[1, 2, 3]