Início / unix / Perguntas / 784570
Accepted
ImportanceLow7312
Asked: 2024-10-06 05:36:58 +0800 CST

Obter caminho de arquivo exclusivo com sufixo numérico entre parênteses

  • 772

Estou procurando uma função que possa lidar com casos em que o caminho do arquivo de entrada já contém um sufixo numérico entre parênteses. A implementação atual da minha get_unique_pathfunção incrementa o sufixo numérico em um até encontrar um caminho que não existe. No entanto, essa abordagem falha quando o caminho do arquivo de entrada já contém um sufixo numérico.

Por exemplo, se o caminho do arquivo de entrada for example(1).txt, a função anexará incorretamente outro conjunto de parênteses e um número, resultando em example(1)(2).txt. Em vez disso, quero que a função verifique o caminho base para parênteses existentes e incremente o sufixo numérico de acordo.

Aqui está minha implementação atual:

def get_unique_path(file_path: str, extension: str) -> Optional[str]:
    base_path = os.path.splitext(file_path)[0]
    attempt = 1
    while True:
        path = f"{base_path}({attempt + 1}).{extension}"
        
        if not os.path.exists(path):
            return path
        
        attempt += 1

Gostaria de modificar esta função para lidar com casos em que o caminho do arquivo de entrada já contém um sufixo numérico entre parênteses. Alguém pode sugerir uma solução?

python

2 respostas

  1. Best Answer

    O primeiro passo óbvio é verificar se o file_pathjá contém um número entre parênteses e removê-lo se tiver. Podemos usar o remódulo para executar essa operação.

    Uma implementação possível é:

    import re
from pathlib import Path

re_marker = re.compile(r"(?P<prefix>.*)\(\d+\)$")


def get_unique_path(file_path: str | Path) -> str:
    file_path = Path(file_path)
    parent = file_path.parent
    base = file_path.stem
    if mo := re_marker.match(base):
        base = mo.group("prefix")

    count = 1
    while True:
        new_path = parent / f"{base}({count}){file_path.suffix}"
        if not new_path.exists():
            return str(new_path)
        count += 1

    Aqui estão alguns testes simples:

    >>> get_unique_path('/tmp/foo')
'/tmp/foo(1)'
>>> # create /tmp/foo(1)
>>> open('/tmp/foo(1)', 'w').close()
>>> # check again
>>> get_unique_path('/tmp/foo')
'/tmp/foo(2)'
>>> # pass in a filename that already contains a number
>>> get_unique_path('/tmp/foo(1)')
'/tmp/foo(2)'
    • 1

  2. Para lidar com casos em que o caminho do arquivo de entrada já contém um sufixo numérico entre parênteses, você pode usar expressões regulares para extrair o caminho base e o sufixo numérico.

    Use re.search(r'(.*)\((\d+)\)(\.[^.]+)$', file_path)para corresponder ao caminho do arquivo de entrada. Em seguida, incremente o sufixo numérico e construa o novo caminho. Se nenhuma correspondência for encontrada, use a lógica original para anexar um sufixo numérico e incrementá-lo até que um caminho que não existe seja encontrado.

    if match:
    base_path = match.group(1)
    count = int(match.group(2))
    file_extension = match.group(3)

    ...
else:
    ...
    • 0

relate perguntas