Python中的原子文件替换

import os

def atomic_file_replace(src_file, dst_file):
    # 首先，将源文件重命名为一个临时文件
    temp_file = dst_file + ".tmp"
    os.rename(dst_file, temp_file)

    try:
        # 然后，将目标文件重命名为目标文件的最终名称
        os.rename(src_file, dst_file)
    except Exception as e:
        # 如果发生错误，将临时文件重命名回目标文件的最终名称
        os.rename(temp_file, dst_file)
        raise e
    else:
        # 如果成功，则删除临时文件
        os.remove(temp_file)