Получить уникальный путь файла с числовым суффиксом в скобках

Я ищу функцию, которая может обрабатывать случаи, когда путь к входному файлу уже содержит числовой суффикс в скобках. Текущая реализация моей функции get_unique_path увеличивает числовой суффикс на единицу, пока не найдет путь, который не существует. Однако этот подход не работает, когда путь к входному файлу уже содержит числовой суффикс.

Например, если путь к входному файлу example(1).txt, функция неправильно добавит ещё одну пару скобок и число, в результате получится example(1)(2).txt. Вместо этого я хочу, чтобы функция проверяла базовый путь на наличие существующих скобок и увеличивала числовой суффикс соответственно.

Вот моя текущая реализация:

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

Я хотел бы изменить эту функцию, чтобы обработать случаи, когда путь к входному файлу уже содержит числовой суффикс в скобках. Может кто-нибудь предложить решение?

Очевидный первый шаг — проверить, содержит ли file_path уже число в скобках, и удалить его, если это так. Мы можем использовать re модуль для выполнения этой операции.

Одним из возможных вариантов реализации является:

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

Вот несколько простых тестов:

>>> get_unique_path('/tmp/foo')
'/tmp/foo(1)'
>>> # создаём /tmp/foo(1)
>>> open('/tmp/foo(1)', 'w').close()
>>> # проверяем снова
>>> get_unique_path('/tmp/foo')
'/tmp/foo(2)'
>>> # передаем имя файла, которое уже содержит число
>>> get_unique_path('/tmp/foo(1)')
'/tmp/foo(2)'

Чтобы обработать случаи, когда путь к входному файлу уже содержит числовой суффикс в скобках, вы можете использовать регулярные выражения для извлечения базового пути и числового суффикса.

Используйте re.search(r'(.*)\((\d+)\)(\.[^.]+)$', file_path), чтобы сопоставить путь к входному файлу. Затем увеличьте числовой суффикс и создайте новый путь. Если совпадение не найдено, используйте оригинальную логику, чтобы добавить числовой суффикс и увеличивать его, пока не будет найден путь, который не существует.

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

    ...
else:
    ...