如何在 Python 中创建 tmp 文件？

Question

我有这个引用文件路径的函数：

some_obj.file_name(FILE_PATH)

其中 FILE_PATH 是文件路径的字符串，即H:/path/FILE_NAME.ext

我想在我的 python 脚本中创建一个包含字符串内容的文件 FILE_NAME.ext：

some_string = 'this is some content'

如何解决这个问题？ Python 脚本将被放置在 Linux 盒子中。

Answer 1

我想你正在寻找这个：http://docs.python.org/library/tempfile.html

import tempfile
with tempfile.NamedTemporaryFile() as tmp:
    print(tmp.name)
    tmp.write(...)

但是：

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

如果您对此感到担忧：

import os, tempfile
tmp = tempfile.NamedTemporaryFile(delete=False)
try:
    print(tmp.name)
    tmp.write(...)
finally:
    tmp.close()
    os.unlink(tmp.name)

Answer 2

python 有一个tempfile module，但一个简单的文件创建也可以解决问题：

new_file = open("path/to/FILE_NAME.ext", "w")

现在您可以使用write 方法对其进行写入：

new_file.write('this is some content')

---

使用tempfile 模块可能如下所示：

import tempfile

new_file, filename = tempfile.mkstemp()

print(filename)

os.write(new_file, "this is some content")
os.close(new_file)

使用mkstemp，您有责任在完成文件后删除该文件。使用其他参数，您可以影响文件的目录和名称。

---

更新

正如Emmet Speer 正确指出的那样，使用mkstemp 时有security considerations，因为客户端代码负责关闭/清理创建的文件。处理它的更好方法是以下 sn-p（取自链接）：

import os
import tempfile

fd, path = tempfile.mkstemp()
try:
    with os.fdopen(fd, 'w') as tmp:
        # do stuff with temp file
        tmp.write('stuff')
finally:
    os.remove(path)

os.fdopen 将文件描述符包装在 Python 文件对象中，当 with 退出时会自动关闭。对 os.remove 的调用会在不再需要时删除文件。