如何使用 Python 将临时文件持久保存到磁盘？

Question

我正在尝试使用“tempfile”模块来操作和创建文本文件。文件准备好后，我想将其保存到磁盘。我认为这就像使用'shutil.copy'一样简单。但是，我得到一个“权限被拒绝”的 IOError：

>>> import tempfile, shutil
>>> f = tempfile.TemporaryFile(mode ='w+t')
>>> f.write('foo')
>>> shutil.copy(f.name, 'bar.txt')

Traceback (most recent call last):
  File "<pyshell#5>", line 1, in <module>
    shutil.copy(f.name, 'bar.txt')
  File "C:\Python25\lib\shutil.py", line 80, in copy
    copyfile(src, dst)
  File "C:\Python25\lib\shutil.py", line 46, in copyfile
    fsrc = open(src, 'rb')
IOError: [Errno 13] Permission denied: 'c:\\docume~1\\me\\locals~1\\temp\\tmpvqq3go'
>>> 

这不是在使用“tempfile”库时的意图吗？有一个更好的方法吗？ （也许我忽略了一些非常琐碎的事情）

Answer 1

hop 是对的，而且 dF。错误发生的原因不正确。

由于您尚未致电f.close()，因此该文件未被删除。

doc 的 NamedTemporaryFile 表示：

在命名的临时文件仍处于打开状态时，该名称是否可用于第二次打开文件，因平台而异（在 Unix 上可以这样使用；在 Windows NT 或更高版本上不能）。

对于TemporaryFile：

在 Unix 下，文件的目录项在文件创建后立即被删除。其他平台不支持；您的代码不应依赖于使用此函数创建的临时文件，该文件在文件系统中具有或不具有可见名称。

因此，要保留一个临时文件（在 Windows 上），您可以执行以下操作：

import tempfile, shutil
f = tempfile.NamedTemporaryFile(mode='w+t', delete=False)
f.write('foo')
file_name = f.name
f.close()
shutil.copy(file_name, 'bar.txt')
os.remove(file_name)

Hans Sjunnesson 提供的解决方案也是关闭的，因为copyfileobj 只从类文件对象复制到类文件对象，而不是文件名：

shutil.copyfileobj(fsrc, fdst[, length])

将类文件对象fsrc的内容复制到类文件对象fdst。整数长度（如果给定）是缓冲区大小。特别是，负长度值意味着复制数据而不以块的形式循环源数据；默认情况下，数据以块的形式读取，以避免不受控制的内存消耗。注意如果fsrc对象的当前文件位置不为0，则只会复制当前文件位置到文件末尾的内容。

Answer 2

您使用TemporaryFile 或NamedTemporaryFile 创建的文件在关闭时会自动删除，这就是您收到错误消息的原因。如果你不想要这个，你可以改用mkstemp（参见tempfile的文档）。

>>> import tempfile, shutil, os
>>> fd, path = tempfile.mkstemp()
>>> os.write(fd, 'foo')
>>> os.close(fd)
>>> shutil.copy(path, 'bar.txt')
>>> os.remove(path)

Answer 3

从 python 2.6 开始，您还可以使用 NamedTemporaryFile 并将 delete= 选项设置为 False。这样，即使您关闭临时文件，也可以访问它。

请注意，在 Windows（NT 和更高版本）上，当文件仍处于打开状态时，您无法再次访问该文件。您必须先关闭它，然后才能复制它。在 Unix 系统上不是这样。

Answer 4

在您的示例中，您始终可以使用 shutil.copyfileobj：

new_file = open('bar.txt', 'rw')
shutil.copyfileobj(f, new_file)