一种无需覆盖即可创建文件和目录的方法

Question

你知道当你下载一些东西并且下载文件夹包含一个同名的文件时，而不是覆盖它或抛出错误，文件最终会在末尾附加一个数字吗？比如我要下载my_file.txt，但是目标文件夹中已经存在，那么新的文件将命名为my_file(2).txt。如果我再试一次，它会是my_file(3).txt。

我想知道 Python 3.x 中是否有办法检查它并获得一个唯一的名称（不一定要创建文件或目录）。我目前正在这样做：

import os
def new_name(name, newseparator='_')
    #name can be either a file or directory name

    base, extension = os.path.splitext(name)
    i = 2
    while os.path.exists(name):
        name = base + newseparator + str(i) + extension
        i += 1

    return name

在上面的示例中，如果 cwd 中已经存在 my_file.txt，则运行 new_file('my_file.txt') 将返回 my_file_2.txt。 name 也可以包含完整或相对路径，它也可以。

Answer 1

我会使用PathLib 并按照以下方式做一些事情：

from pathlib import Path 

def new_fn(fn, sep='_'):
    p=Path(fn)
    if p.exists():
        if not p.is_file(): 
            raise TypeError
        np=p.resolve(strict=True)
        parent=str(np.parent)
        extens=''.join(np.suffixes)  # handle multiple ext such as .tar.gz
        base=str(np.name).replace(extens,'')
        i=2
        nf=parent+base+sep+str(i)+extens    
        while Path(nf).exists():
            i+=1
            nf=parent+base+sep+str(i)+extens    
        return nf   
    else:       
        return p.parent.resolve(strict=True) / p 

这仅处理写入的文件，但同样的方法适用于目录（您稍后添加）。我将把它作为一个项目留给读者。

Answer 2

另一种获得新名称的方法是使用内置的tempfile 模块：

from pathlib import Path
from tempfile import NamedTemporaryFile

def new_path(path: Path, new_separator='_'):
    prefix = str(path.stem) + new_separator
    dir = path.parent
    suffix = ''.join(path.suffixes)

    with NamedTemporaryFile(prefix=prefix, suffix=suffix, delete=False, dir=dir) as f:
        return f.name

如果你在下载目录中执行这个函数，你会得到类似的东西：

>>> new_path(Path('my_file.txt'))
'/home/krassowski/Downloads/my_file_90_lv301.txt'

90_lv301 部分是由 Python 的 tempfile 模块在内部生成的。

注意：使用delete=False 参数，该函数将创建（并且不删除）具有新名称的空文件。如果您不想以这种方式创建空文件，只需删除 delete=False，但保留它会阻止其他人在您下一次操作之前创建具有该名称的新文件（尽管他们仍然可以覆盖它）。

简单地说，如果您（或最终用户）要同时运行您的程序两次，使用delete=False 可以防止并发问题。