Python 子进程优雅读取临时文件的最佳实践

2024-03-19 23:23:06

在子进程中优雅地读取临时文件

在编程中，有时我们需要将数据存储在临时文件中，以便另一个进程可以访问它。在 Python 中，可以使用 NamedTemporaryFile 类创建临时文件。但是，在 Windows 上，如果临时文件在写入后被刷新但没有关闭，子进程将无法打开它进行读取。

解决方案：使用 `mkstemp`

一种更优雅的方法是使用 mkstemp 函数，它返回一个文件符和一个文件名。我们可以使用 os.fdopen() 函数将文件符包装在文件对象中，如下所示：

import tempfile

fd, temp_filename = tempfile.mkstemp()
with os.fdopen(fd, 'w') as f:
    f.write(someStuff)

# ...run the subprocess and wait for it to complete...

os.close(fd)
os.remove(temp_filename)

使用此方法，无需关闭文件即可将临时文件传递给子进程，从而简化了代码并避免了潜在的错误来源。此外，它在所有平台上都可靠地工作，而无需担心特定于操作系统的细微差别。

替代方案：使用 `TemporaryDirectory` 上下文管理器

另一个替代方案是使用 TemporaryDirectory 上下文管理器，它创建一个临时目录并在退出时自动删除该目录。我们可以将临时文件写入该目录，然后传递目录路径给子进程。

from tempfile import TemporaryDirectory

with TemporaryDirectory() as temp_dir:
    temp_file = os.path.join(temp_dir, 'temp_file')
    with open(temp_file, 'w') as f:
        f.write(someStuff)

# ...run the subprocess and wait for it to complete...

这种方法的优点是它不需要显式删除临时文件，因为它会在上下文管理器退出时自动删除。