我们想将数据写入文件中,但只有当文件不存在时才执行写入操作。
解决方案
这个问题可以通过 open() 函数中鲜为人知的 x 模式替代常见的 w 模式来解决。例如:
>>> with open('somefile', 'wt') as f:
... f.write('Hello\n')
...
>>> with open('somefile', 'xt') as f:
... f.write('Hello\n')
...
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
FileExistsError: [Errno 17] File exists: 'somefile'
>>>
如果文件是二进制模式的,用 xb 代替 xt 模式即可。
讨论
这里的示例代码以一种非常优雅的方式解决了一个常会在写文件时出现的问题(即,意外地覆盖了某个已存在的文件)。另一种解决方案是首先像这样检查文件是否已存在:
>>> import os
>>> if not os.path.exists('somefile'):
... with open('somefile', 'wt') as f:
... f.write('Hello\n')
... else:
... print('File already exists!')
...
File already exists!
>>>
显然,使用 x 模式更加简单直接。需要注意的是,x 模式是 Python 3 中对 open() 函数的扩展。在早期的 Python 版本或者在 Python 的实现中用到的底层 C 函数库里都不存在这样的模式。