安全地创建一个文件，当且仅当它不存在于python中
我希望根据文件是否已经存在写入文件，只有在文件不存在的情况下才写入（实际上，我希望不断尝试文件，直到找到不存在的文件为止）。
以下代码显示了一种潜在攻击者可以插入符号链接的方式，正如本文中对文件测试和正在编写的文件之间的建议。 如果代码以足够高的权限运行，则可能会覆盖任意文件。
有什么办法可以解决这个问题吗？
import os
import errno
file_to_be_attacked = 'important_file'
with open(file_to_be_attacked, 'w') as f:
f.write('Some important content!n')
test_file = 'testfile'
try:
with open(test_file) as f: pass
except IOError, e:
# symlink created here
os.symlink(file_to_be_attacked, test_file)
if e.errno != errno.ENOENT:
raise
else:
with open(test_file, 'w') as f:
f.write('Hello, kthxbye!n')
编辑 ：另请参阅Dave Jones的答案：从Python 3.3开始，您可以使用x标志open()来提供此功能。
下面的原始答案
是的，但不使用Python的标准open()调用。 您需要使用os.open() ，它允许您为基础C代码指定标志。
特别是，你想使用O_CREAT | O_EXCL O_CREAT | O_EXCL 。 在我的Unix系统的O_EXCL下open(2)的手册页：
确保此调用创建文件：如果此标志与O_CREAT一起指定，且路径名已存在，则open()将失败。 如果未指定O_CREAT则O_EXCL的行为未定义。
如果指定了这两个标志，则不会遵循符号链接：如果路径名是符号链接，则无论符号链接指向何处， open()都会失败。
在内核2.6或更高版本上使用NFSv3或更高版本时，仅在NFS上支持O_EXCL 。 在没有提供NFS O_EXCL支持的环境中，依赖它执行锁定任务的程序将包含竞争条件。
所以这不是完美的，但AFAIK它是最接近你可以避免这种竞争条件。
编辑：使用os.open()而不是open()的其他规则仍然适用。 特别是，如果要使用返回的文件描述符进行读取或写入，则还需要O_RDONLY ， O_WRONLY或O_RDWR标志之一。
所有的O_*标志都在Python的os模块中，所以你需要import os并使用os.O_CREAT等。
例：
import os
import errno
flags = os.O_CREAT | os.O_EXCL | os.O_WRONLY
try:
file_handle = os.open('filename', flags)
except OSError as e:
if e.errno == errno.EEXIST: # Failed as the file already exists.
pass
else: # Something unexpected went wrong so reraise the exception.
raise
else: # No exception, so the file must have been created successfully.
with os.fdopen(file_handle, 'w') as file_obj:
# Using `os.fdopen` converts the handle to an object that acts like a
# regular Python file object, and the `with` context manager means the
# file will be automatically closed when we're done with it.
file_obj.write("Look, ma, I'm writing to a new file!")
作为参考，Python 3.3在open()函数中实现了一个新的'x'模式来覆盖这个用例（仅创建，如果文件存在则失败）。 请注意， 'x'模式是自行指定的。 使用'wx'导致ValueError因为'w'是多余的（如果调用成功，您可以做的唯一事情就是写入文件;如果调用成功则不会存在）：
>>> f1 = open('new_binary_file', 'xb')
>>> f2 = open('new_text_file', 'x')
对于Python 3.2及更低版本（包括Python 2.x），请参阅接受的答案。
如果一个文件不存在，该代码将很容易创建一个文件。
import os
if not os.path.exists('file'):
open('file', 'w').close()
上一篇: Safely create a file if and only if it does not exist with python