如何在Python中实现临时文件操作成功后重命名为正式文件

原代码报错原因 #

你遇到的报错本质来自两个问题：

1. 运行时修改NamedTemporaryFile实例的delete属性不会对所有Python版本生效，上下文管理器退出时仍会尝试删除原路径的文件，而你已经将文件重命名，原路径不存在就会抛出异常。
2. 你在文件仍处于打开状态时执行重命名操作，Windows环境下会直接触发文件占用错误。

标准实现方案 #

方案1：使用NamedTemporaryFile自行管理生命周期 #

初始化时就关闭自动删除逻辑，用try-finally块控制临时文件清理，同时将重命名操作放在文件关闭后执行，避免跨平台兼容问题：

import os
import tempfile

temp_file_path = None
try:
    # 初始化时指定delete=False，关闭上下文的自动删除逻辑
    with tempfile.NamedTemporaryFile(mode='wb', dir=os.getcwd(), delete=False) as f:
        temp_file_path = f.name
        # 执行你的文件操作
        f.write(b'foobar')
        # 强制刷入磁盘，避免内容留在内核缓存导致写入丢失
        f.flush()
        os.fsync(f.fileno())
    # 文件已关闭，执行重命名
    os.rename(temp_file_path, 'foobar')
    # 重命名成功，标记不需要删除临时文件
    temp_file_path = None
finally:
    # 操作失败时清理临时文件
    if temp_file_path is not None and os.path.exists(temp_file_path):
        os.unlink(temp_file_path)

方案2：使用mkstemp+os.fdopen获取文件对象 #

你提到的mkstemp返回的文件描述符，可以通过os.fdopen()直接转为Python文件对象，实现自定义的生命周期管理：

import os
import tempfile

temp_file_path = None
try:
    fd, temp_file_path = tempfile.mkstemp(dir=os.getcwd())
    # 将文件描述符转为文件对象，with块结束会自动关闭文件描述符
    with os.fdopen(fd, 'wb') as f:
        f.write(b'foobar')
        f.flush()
        os.fsync(f.fileno())
    os.rename(temp_file_path, 'foobar')
    temp_file_path = None
except Exception:
    # 异常时清理临时文件后重新抛出原异常，不会吞错
    if temp_file_path is not None and os.path.exists(temp_file_path):
        os.unlink(temp_file_path)
    raise

两个方案都满足原子性要求：只要重命名操作完成，正式文件就是完整的；操作任何一步失败都会自动清理临时文件，不会残留垃圾文件。

内容的提问来源于stack exchange，提问作者petersohn