我想一次更改几个文件,如果我可以写入所有文件。我想知道是否可以以某种方式将多个打开调用与 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
如果这不可能,那么这个问题的优雅解决方案会是什么样子?
从 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"
大多数时候,您有一组可变文件,但您可能希望一个接一个地打开它们。
只需将 and
替换为 ,
即可:
try:
with open('a', 'w') as a, open('b', 'w') as b:
do_something()
except IOError as e:
print 'Operation failed: %s' % e.strerror
对于一次打开多个文件或长文件路径,将内容分成多行可能很有用。从@Sven Marnach 在评论中建议的 Python Style Guide 到另一个答案:
with open('/path/to/InFile.ext', 'r') as file_1, \
open('/path/to/OutFile.ext', 'w') as file_2:
file_2.write(file_1.read())
从 Python 3.10 开始,Parenthesized context managers 有了一个新功能,它允许如下语法:
with (
open("a", "w") as a,
open("b", "w") as b
):
do_something()
with open("a", "w") as a, open ("b", "w") as b:
不能做的任何事情吗?
从 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]
嵌套 with 语句会做同样的工作,而且在我看来,处理起来更简单。
假设您有 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:
迟到的答案(8 岁),但对于希望将多个文件合并为一个的人,以下功能可能会有所帮助:
def multi_open(_list):
out=""
for x in _list:
try:
with open(x) as f:
out+=f.read()
except:
pass
# print(f"Cannot open file {x}")
return(out)
fl = ["C:/bdlog.txt", "C:/Jts/tws.vmoptions", "C:/not.exist"]
print(multi_open(fl))
2018-10-23 19:18:11.361 PROFILE [Stop Drivers] [1ms]
2018-10-23 19:18:11.361 PROFILE [Parental uninit] [0ms]
...
# This file contains VM parameters for Trader Workstation.
# Each parameter should be defined in a separate line and the
...
不定期副业成功案例分享
with
打开文件的变量列表?ExitStack
。在任何早期版本的 Python 中都没有简单的方法可以做到这一点。