os.walk, не заглиблюючись у каталоги нижче

Як обмежувати 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

Використовуйте 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параметр, який вказує, наскільки глибока буде проходити рекурсія.

Не використовуйте os.walk.

Приклад:

import os

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

Я думаю, що рішення насправді дуже просте.

використання

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

Пропозиція використовувати listdirхороша. Пряма відповідь на ваше запитання в Python 2 - цеroot, dirs, files = os.walk(dir_name).next() .

Еквівалентний синтаксис Python 3 є root, dirs, files = next(os.walk(dir_name))

Ви можете використовувати, os.listdir()що повертає список імен (як для файлів, так і для каталогів) у заданій теці. Якщо вам потрібно розрізняти файли та каталоги, зателефонуйте os.stat()до кожного імені.

Якщо у вас є більш складні вимоги, ніж просто головна директорія (наприклад, ігноруйте файли VCS і т. Д.), Ви також можете змінити список каталогів, щоб запобігти тому, щоб 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 не знає про зовнішнє відсікання.

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

Ця ж ідея listdir, але коротша:

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

Відчув, як кинув мою 2 пенси.

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

У 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] )

Оскільки Python 3.5 ви можете використовувати os.scandirзамість os.listdir. Замість рядків ви отримуєте ітератор DirEntryоб'єктів натомість. З документів:

Використання scandir()замість listdir()може значно збільшити продуктивність коду, який також потребує інформації про тип файлу або атрибути файлів, оскільки DirEntryоб’єкти викривають цю інформацію, якщо операційна система надає її під час сканування каталогу. Всі DirEntryметоди можуть виконати системний виклик, але is_dir()і , як is_file()правило , потрібно тільки системний виклик для символічних посилань; DirEntry.stat()завжди вимагає системного виклику в Unix, але вимагає лише одного для символічних посилань у Windows.

Ви можете отримати доступ до імені об'єкта, за допомогою DirEntry.nameякого потім еквівалентний вихідos.listdir

Ви також можете зробити наступне:

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

Ось як я це вирішив

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

...

Існує улов при використанні 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 ().

Ви можете використовувати цей фрагмент

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

створіть список виключень, використовуйте fnmatch, щоб пропустити структуру каталогу та виконати процес

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**):

Чому б просто не використовувати rangeта os.walkпоєднати з zip? Це не найкраще рішення, але працювало б теж.

Наприклад так:

# your part before
for count, (root, dirs, files) in zip(range(0, 1), os.walk(dir_name)):
    # logic stuff
# your later part

Для мене працює на python 3.

Також: A breakпростіше занадто btw. (Подивіться на відповідь від @Pieter)

Невелика зміна відповіді Алекса, але використовуючи __next__():

print(next(os.walk('d:/'))[2]) або print(os.walk('d:/').__next__()[2])

з [2]будучи fileв root, dirs, fileзгаданих в інших відповідях

зміни кореневої папки для кожного каталогу, знайденого os.walk. Я вирішую цю перевірку, чи root == каталог

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for root, dirs, files in os.walk(dir_name):
        if root == dir_name: #This only meet parent folder
            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

import os

def listFiles(self, dir_name):
    names = []
    for root, directory, files in os.walk(dir_name):
        if root == dir_name:
            for name in files:
                names.append(name)
    return names