Linux环境下如何用Python原子重命名文件且不覆盖已有文件

Linux系统中，os.rename()和shutil.move()都会静默覆盖已存在的目标文件，而两步式的「检查存在性+重命名」会引入竞争条件，无法保证原子性。要实现和mv --no-clobber完全一致的原子操作，我们可以直接调用Linux内核提供的renameat2()系统调用——它支持RENAME_NOREPLACE标志，能在目标文件存在时直接失败，且整个操作是内核层面的原子操作。

实现代码 #

import ctypes
from ctypes import c_int, c_char_p, c_uint

# 定义Linux内核常量（来自linux/fs.h）
RENAME_NOREPLACE = 0x1

# 加载系统libc库
libc = ctypes.CDLL('libc.so.6', use_errno=True)

# 声明renameat2函数的参数和返回值类型
# 函数原型：int renameat2(int olddirfd, const char *oldpath, int newdirfd, const char *newpath, unsigned int flags);
libc.renameat2.argtypes = [c_int, c_char_p, c_int, c_char_p, c_uint]
libc.renameat2.restype = c_int

def rename_no_clobber(old_path, new_path):
    # AT_FDCWD表示使用当前工作目录作为文件描述符
    AT_FDCWD = -100
    # 转换路径为字节串（符合C风格字符串要求）
    old_bytes = old_path.encode('utf-8')
    new_bytes = new_path.encode('utf-8')
    
    result = libc.renameat2(AT_FDCWD, old_bytes, AT_FDCWD, new_bytes, RENAME_NOREPLACE)
    if result != 0:
        errno = ctypes.get_errno()
        raise OSError(errno, f"重命名失败: {ctypes.strerror(errno)}")

使用示例 #

try:
    rename_no_clobber("old_file.txt", "new_file.txt")
    print("重命名成功")
except OSError as e:
    if e.errno == 17:  # EEXIST，对应目标文件已存在的错误
        print("目标文件已存在，未执行覆盖")
    else:
        print(f"发生错误: {e}")

关键说明 #

• 原子性保障：renameat2()是单系统调用，内核会一次性完成判断和重命名操作，完全避免了竞争条件。
• 兼容性：该系统调用从Linux 3.15版本开始支持，目前主流Linux发行版均已满足版本要求。
• 错误处理：当目标文件存在时，会抛出OSError，错误号为EEXIST（数值17），可针对性捕获处理覆盖冲突场景。

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