Скопируйте файлы рекурсивно

Question

def ignore_list(path, files): 
    filesToIgnore = [] 
    for fileName in files: 
     fullFileName = os.path.join(os.path.normpath(path), fileName) 
     if not os.path.isdir(fullFileName) and not fileName.endswith('pyc') and not fileName.endswith('ui') and not fileName.endswith('txt') and not fileName == '__main__.py' and not fileName == 'myfile.bat': 
      filesToIgnore.append(fileName) 

    return filesToIgnore 

# Start of script  
shutil.copytree(srcDir, dstDir, ignore=ignore_list) 

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

Как я могу это сделать?

Answer 1

Что я мог понять из вашего вопроса, вы могли бы просто написать:

if fileName not in fileNameList: 
    filesToIgnore.append(fileName) 

где fileNameList список имен файлов, которые вы хотите скопировать.

Update:

fName, fileExtension = os.path.splitext(fileName) 
if fileName not in fileNameList or fileExtension not in allowedExtensionList: 
    filesToIgnore.append(fileName) 

Answer 2

Вы также можете написать метод, как в следующем.

def ignore_files(path, extensions, wantedfiles): 
     ignores = [] 

     for root, dirs, filenames in os.walk(path): 
      for filename in filenames: 
      if any(filename.endswith(_) for _ in extensions): 
        ignores.append(os.path.join(os.path.normpath(path), filename)) 
      elif filename in wantedfiles: 
        ignores.append(os.path.join(os.path.normpath(path), filename)) 

     return ignores 

# Start of script 
ignore_list = ignore_files(srcDir, ['pyc', 'uid', 'txt']) 
shutil.copytree(srcDir, dstDir, ignore=ignore_list)