Как предотвратить перезапись shutil.move файла, если он уже существует? - PullRequest
Новая
       78

Как предотвратить перезапись shutil.move файла, если он уже существует?

0 голосов
/ 28 мая 2020

Я использую этот код Python в Windows: 

shutil.move(documents_dir + "\\" + file_name, documents_dir + "\\backup\\"
            + subdir_name + "\\" + file_name)

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

, например, file_name = foo.pdf

, а в папке backup будет foo.pdf, foo(1).pdf, foo(2).pdf и c. или аналогично, например, с тире foo-1.pdf, foo-2.pdf et c.

1 Ответ

1 голос
/ 28 мая 2020

Вы можете просто проверить с помощью os.path.exists() по мере продвижения. 

import os
import shutil

file_name = 'test.csv'
documents_dir = r'C:\BR\Test'
subdir_name = 'test'

# using os.path.join() makes your code easier to port to another OS
source = os.path.join(documents_dir, file_name)
dest = os.path.join(documents_dir, 'backup', subdir_name, file_name)

num = 0
# loop until we find a file that doesn't exist
while os.path.exists(dest):
    num += 1

    # use rfind to find your file extension if there is one
    period = file_name.rfind('.')
    # this ensures that it will work with files without extensions
    if period == -1:
        period = len(file_name)

    # create our new destination
    # we could extract the number and increment it
    # but this allows us to fill in the gaps if there are any
    # it has the added benefit of avoiding errors 
    # in file names like this "test(sometext).pdf"
    new_file = f'{file_name[:period]}({num}){file_name[period:]}'

    dest = os.path.join(documents_dir, 'backup', subdir_name, new_file)

shutil.move(source, dest)

Или, поскольку это, вероятно, используется в al oop, вы можете просто вставить его в функцию. 

import os
import shutil

def get_next_file(file_name, dest_dir):
    dest = os.path.join(dest_dir, file_name)
    num = 0

    while os.path.exists(dest):
        num += 1

        period = file_name.rfind('.')
        if period == -1:
            period = len(file_name)

        new_file = f'{file_name[:period]}({num}){file_name[period:]}'

        dest = os.path.join(dest_dir, new_file)

    return dest

file_name = 'test.csv'
documents_dir = r'C:\BR\Test'
subdir_name = 'test'

source = os.path.join(documents_dir, file_name)

dest = get_next_file(file_name, os.path.join(documents_dir, 'backup', subdir_name))

shutil.move(source, dest)
...