os.walk без копания в каталогах ниже

Question

Как ограничить os.walk только для возврата файлов в каталог, который я ему предоставляю?

def _dir_list(self, dir_name, whitelist): 
    outputList = [] 
    for root, dirs, files in os.walk(dir_name): 
     for f in files: 
      if os.path.splitext(f)[1] in whitelist: 
       outputList.append(os.path.join(root, f)) 
      else: 
       self._email_to_("ignore") 
    return outputList 

Answer 1

Используйте функцию walklevel.

import os 

def walklevel(some_dir, level=1): 
    some_dir = some_dir.rstrip(os.path.sep) 
    assert os.path.isdir(some_dir) 
    num_sep = some_dir.count(os.path.sep) 
    for root, dirs, files in os.walk(some_dir): 
     yield root, dirs, files 
     num_sep_this = root.count(os.path.sep) 
     if num_sep + level <= num_sep_this: 
      del dirs[:] 

Он работает так же, как os.walk, но вы можете передать ему параметр level, который указывает, насколько глубоко рекурсии будет идти.

Answer 2

Вы можете использовать os.listdir(), который возвращает список имен (для обоих файлов и каталогов) в заданной директории. Если вам нужно различать файлы и каталоги, вызовите os.stat() по каждому имени.

Answer 3

Не используйте os.walk.

Пример:

import os 

root = "C:\\" 
for item in os.listdir(root): 
    if os.path.isfile(os.path.join(root, item)): 
     print item 

Answer 4

Предложение использовать listdir является хорошим. Прямой ответ на ваш вопрос: root, dirs, files = os.walk(dir_name).next()

Answer 5

Если у вас есть более сложные требования, чем только верхний каталог (например, игнорировать VCS-dirs и т. Д.), Вы также можете изменить список каталогов, чтобы предотвратить их повторение через os.walk.

т.е.

def _dir_list(self, dir_name, whitelist): 
    outputList = [] 
    for root, dirs, files in os.walk(dir_name): 
     dirs[:] = [d for d in dirs if is_good(d)] 
     for f in files: 
      do_stuff() 

Примечание - будьте осторожны, чтобы мутировать список, а не просто перепривязывают его. Очевидно, os.walk не знает о внешнем восстановлении.

Answer 6

Вы также можете сделать следующее:

for path, subdirs, files in os.walk(dir_name): 
    for name in files: 
     if path == ".": #this will filter the files in the current directory 
      #code here 

Answer 7

Я думаю, что решение на самом деле очень просто.

использование

break 

только сделать первую итерацию цикла, должна быть более изящным способом.

for root, dirs, files in os.walk(dir_name): 
    for f in files: 
     ... 
     ... 
    break 
... 

Первый раз, когда вы звоните os.walk, она возвращает тюльпаны для текущего каталога, а затем на следующем цикле содержимое следующего каталога.

Возьмите оригинальный сценарий и просто добавьте перерыв.

def _dir_list(self, dir_name, whitelist): 
    outputList = [] 
    for root, dirs, files in os.walk(dir_name): 
     for f in files: 
      if os.path.splitext(f)[1] in whitelist: 
       outputList.append(os.path.join(root, f)) 
      else: 
       self._email_to_("ignore") 
     break 
    return outputList 

Answer 8

Та же идея с listdir, но короче:

[f for f in os.listdir(root_dir) if os.path.isfile(os.path.join(root_dir, f))] 

Answer 9

Это, как я решил, что

if recursive: 
    items = os.walk(target_directory) 
else: 
    items = [next(os.walk(target_directory))] 

... 

Answer 10

Существует поймать при использовании listdir. Os.path.isdir (идентификатор) должен быть абсолютным путем.Для того, чтобы подобрать подкаталоги вы делаете:

for dirname in os.listdir(rootdir): 
    if os.path.isdir(os.path.join(rootdir, dirname)): 
    print("I got a subdirectory: %s" % dirname) 

Альтернативой является изменение в каталог, чтобы сделать тестирование без os.path.join().

Answer 11

В Python 3, я был в состоянии сделать это:.

import os 
dir = "/path/to/files/" 

#List all files immediately under this folder: 
print (next(os.walk(dir))[2]) 

#List all folders immediately under this folder: 
print (next(os.walk(dir))[1]) 

Answer 12

for path, dirs, files in os.walk('.'): 
    print path, dirs, files 
    del dirs[:] # go only one level deep 

Answer 13

Вы можете использовать этот фрагмент

for root, dirs, files in os.walk(directory): 
    if level > 0: 
     # do some stuff 
    else: 
     break 
    level-=1 

Answer 14

Войлок как бросание моих 2 пенсов в

baselevel = len(rootdir.split("\\")) 
for subdirs, dirs, files in os.walk(rootdir): 
    curlevel = len(subdirs.split("\\")) 
    if curlevel <= baselevel + 1: 
     [do stuff] 

Answer 15

создать список исключений, использовать fnmatch, чтобы пропустить структуру каталогов d сделать процесс

excludes= ['a\*\b', 'c\d\e'] 
for root, directories, files in os.walk('Start_Folder'): 
    if not any(fnmatch.fnmatch(nf_root, pattern) for pattern in excludes): 
     for root, directories, files in os.walk(nf_root): 
      .... 
      do the process 
      .... 

же, как для «включает в себя»:

if **any**(fnmatch.fnmatch(nf_root, pattern) for pattern in **includes**):