os.walk bez zagłębiania się w poniższe katalogi

Jak ograniczyć os.walkzwracanie plików tylko z katalogu, który go podam?

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

Użyj walklevelfunkcji.

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[:]

Działa tak samo os.walk, ale możesz przekazać mu levelparametr, który wskazuje, jak głęboko zajdzie rekurencja.

Nie używaj os.walk.

Przykład:

import os

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

Myślę, że rozwiązanie jest w rzeczywistości bardzo proste.

posługiwać się

break

aby wykonać tylko pierwszą iterację pętli for, musi istnieć bardziej elegancki sposób.

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

Przy pierwszym wywołaniu os.walk zwraca on tulipany dla bieżącego katalogu, a następnie w następnej pętli wyświetla zawartość następnego katalogu.

Weź oryginalny skrypt i po prostu dodaj przerwę .

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

Sugestia użycia listdirjest dobra. Bezpośrednia odpowiedź na twoje pytanie w Pythonie 2 brzmi root, dirs, files = os.walk(dir_name).next().

Odpowiednia składnia Pythona 3 to root, dirs, files = next(os.walk(dir_name))

Możesz użyć, os.listdir()który zwraca listę nazw (zarówno plików, jak i katalogów) w podanym katalogu. Jeśli potrzebujesz rozróżnić pliki i katalogi, wywołaj os.stat()każdą nazwę.

Jeśli masz bardziej złożone wymagania niż tylko główny katalog (np. Zignoruj ​​katalogi VCS itp.), Możesz również zmodyfikować listę katalogów, aby zapobiec powtarzaniu się przez nie os.walk.

to znaczy:

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()

Uwaga - uważaj, aby zmodyfikować listę, a nie tylko ją ponownie powiązać. Oczywiście os.walk nie wie o zewnętrznym rebindingu.

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

Ten sam pomysł listdir, ale krótszy:

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

Czułem się jak wrzucenie moich 2 pensów.

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

W Pythonie 3 udało mi się to zrobić:

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

Od Pythona 3.5 możesz używać os.scandirzamiast os.listdir. Zamiast łańcuchów otrzymujesz DirEntryw zamian iterator obiektów. Z dokumentów:

Używanie scandir()zamiast tego listdir()może znacznie zwiększyć wydajność kodu, który również wymaga informacji o typie pliku lub atrybutach pliku, ponieważ DirEntryobiekty ujawniają te informacje, jeśli system operacyjny udostępnia je podczas skanowania katalogu. Wszystkie DirEntrymetody mogą wykonywać połączenia systemu, ale is_dir()i is_file()zazwyczaj wymagają jedynie wywołanie systemowe dla dowiązania symboliczne; DirEntry.stat()zawsze wymaga wywołania systemowego w Uniksie, ale wymaga tylko jednego dla dowiązań symbolicznych w Windows.

Możesz uzyskać dostęp do nazwy obiektu, za pośrednictwem DirEntry.namektórej jest odpowiednikiem danych wyjściowych poleceniaos.listdir

Możesz też wykonać następujące czynności:

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

Tak to rozwiązałem

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

...

Podczas korzystania z listdir jest pewien haczyk. Os.path.isdir (identyfikator) musi być ścieżką bezwzględną. Aby wybrać podkatalogi, które robisz:

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

Alternatywą jest przejście do katalogu, aby wykonać test bez funkcji os.path.join ().

Możesz użyć tego fragmentu

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

utwórz listę wykluczeń, użyj fnmatch, aby pominąć strukturę katalogów i wykonać proces

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
            ....

to samo, co w przypadku „obejmuje”:

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

Dlaczego nie po prostu użyć rangei w os.walkpołączeniu zzip ? To nie jest najlepsze rozwiązanie, ale też by działało.

Na przykład w ten sposób:

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

U mnie działa na Pythonie 3.

Poza tym: A przy breakokazji też jest prostszy. (Spójrz na odpowiedź od @Pieter)

Niewielka zmiana w odpowiedzi Alexa, ale używając __next__():

print(next(os.walk('d:/'))[2]) lub print(os.walk('d:/').__next__()[2])

z [2]bycia filew root, dirs, filewspomniano w innych odpowiedzi

zmiany folderu głównego dla każdego katalogu znalezionego przez os.walk. Rozwiązuję to sprawdzanie, czy root == katalog

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