Python：如何确保在程序完成时删除文件？

Question

程序完成后，我有几个绝对必须删除的文件。我真的不想使用tempfile.TemporaryFile因为文件必须传递很多，并且用于os.system()调用中的命令输出重定向。所以我会使用tempfile.mkstemp，但这些不会被自动删除。我考虑将我的main方法包装在try-finally中，如下所示：

if __name__ == "__main__":
    try:
        main()
    finally:
        os.remove(tempfile)

但这看起来很值得怀疑。还有更好的方法吗？

Answer 1

我的问题已在上述评论中得到解答，但我找到了另一个简洁的解决方案，我将留待此处以供将来参考：atexit