python - 在脚本评估期间删除临时文件

Question

我正在使用 python 包 cdo ，它严重依赖 tempfile用于存储中间结果。创建的临时文件非常大，当运行更大的计算时，我遇到了 /tmp 目录被填满并且脚本因 disk full 失败的问题错误(我们谈论的是 10 到 100 GB)。我找到了解决该问题的方法，方法是创建一个本地文件夹，例如 $HOME/tmp 然后执行

import tempfile
tempfile.tempdir='$HOME/tmp'

在导入 cdo 模块之前。虽然这对我有用，但如果我也希望其他人使用我的脚本，那就有点麻烦了。因此我想知道是否有一种更优雅的方法来解决问题，例如，定期告诉 tmpfile 清除所有临时文件(通常只有在脚本完成后才会这样做)。从我的角度来看，这是可能的，因为我正在运行一个长循环，每次迭代都会生成一个命名文件，并且在该迭代期间创建的所有临时文件随后都将被丢弃。

Answer 1

作为examples显示:您可以使用 tempfile在上下文管理器中:

with tempfile.TemporaryFile() as fp:
    fp.write(b'Hello world!')
    fp.seek(0)
    fp.read()

这样，当上下文退出时，它们就会被删除。

...您对 cdo 如何使用临时文件有那么大的控制权吗？