python中的原子`ln -sf`（符号链接覆盖现有文件）

Question

我想创建一个符号链接，如果需要，覆盖现有文件或符号链接。

我发现os.path.exists 只为未损坏的符号链接返回True，所以我猜任何测试都必须包含os.path.lexists。

在 python 中实现ln -sf 的最原子方式是什么？ （即，防止文件在删除和符号链接创建之间被另一个进程创建）

---

区分：This question 没有指定原子需求

Answer 1

此代码试图最小化竞争条件的可能性：

import os
import tempfile

def symlink_force(target, link_name):
    '''
    Create a symbolic link link_name pointing to target.
    Overwrites link_name if it exists.
    '''

    # os.replace() may fail if files are on different filesystems
    link_dir = os.path.dirname(link_name)

    while True:
        temp_link_name = tempfile.mktemp(dir=link_dir)
        try:
            os.symlink(target, temp_link_name)
            break
        except FileExistsError:
            pass
    try:
        os.replace(temp_link_name, link_name)
    except OSError:  # e.g. permission denied
        os.remove(temp_link_name)
        raise

注意：

1. 如果功能被中断（例如计算机崩溃），可能会存在一个额外的随机链接到目标。

2. 一个不太可能的竞争条件仍然存在：在随机命名的 temp_link_name 创建的符号链接可能会在替换 link_name 之前被另一个进程修改。

我提出了python issue 来强调os.symlink() 要求目标不存在的问题。

归功于Robert Seimer's input。