os.walk senza scavare nelle directory sottostanti

Come posso limitare os.walka restituire solo i file nella directory che fornisco?

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

Usa la walklevelfunzione.

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

Funziona esattamente come os.walk, ma puoi passargli un levelparametro che indica quanto in profondità andrà la ricorsione.

Non utilizzare os.walk.

Esempio:

import os

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

Penso che la soluzione sia in realtà molto semplice.

uso

break

per fare solo la prima iterazione del ciclo for, ci deve essere un modo più elegante.

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

La prima volta che chiami os.walk, restituisce tulipani per la directory corrente, quindi al ciclo successivo il contenuto della directory successiva.

Prendi il copione originale e aggiungi solo una pausa .

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

Il suggerimento da utilizzare listdirè buono. La risposta diretta alla tua domanda in Python 2 è root, dirs, files = os.walk(dir_name).next().

La sintassi equivalente di Python 3 è root, dirs, files = next(os.walk(dir_name))

È possibile utilizzare os.listdir()which restituisce un elenco di nomi (sia per i file che per le directory) in una determinata directory. Se hai bisogno di distinguere tra file e directory, chiama os.stat()ogni nome.

Se hai requisiti più complessi rispetto alla sola directory principale (ad esempio, ignora le directory VCS, ecc.), Puoi anche modificare l'elenco delle directory per evitare che os.walk ricorra attraverso di esse.

vale a dire:

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

Nota: fai attenzione a modificare l'elenco, piuttosto che ricollegarlo. Ovviamente os.walk non conosce il rebinding esterno.

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

La stessa idea con listdir, ma più breve:

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

Mi è sembrato di buttare i miei 2 pence.

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

In Python 3, sono stato in grado di farlo:

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

A partire da Python 3.5 puoi usare os.scandirinvece di os.listdir. Invece di stringhe ottieni DirEntryin cambio un iteratore di oggetti. Dai documenti:

L'utilizzo di scandir()invece di listdir()può aumentare in modo significativo le prestazioni del codice che richiede anche informazioni sul tipo di file o sugli attributi del file, poiché gli DirEntryoggetti espongono queste informazioni se il sistema operativo le fornisce durante la scansione di una directory. Tutti i DirEntrymetodi possono effettuare una chiamata di sistema, ma is_dir()e is_file()di solito richiedono solo una chiamata di sistema per i collegamenti simbolici; DirEntry.stat()richiede sempre una chiamata di sistema su Unix ma ne richiede solo una per i collegamenti simbolici su Windows.

È possibile accedere al nome dell'oggetto tramite il DirEntry.namequale è quindi equivalente all'output dios.listdir

Puoi anche fare quanto segue:

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

Ecco come l'ho risolto

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

...

C'è un problema quando si usa listdir. Os.path.isdir (identificatore) deve essere un percorso assoluto. Per scegliere le sottodirectory fai:

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

L'alternativa è passare alla directory per eseguire il test senza os.path.join ().

Puoi usare questo snippet

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

creare un elenco di esclusioni, utilizzare fnmatch per saltare la struttura della directory ed eseguire il processo

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

come per "include":

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

Perché non usare semplicemente un rangee os.walkcombinato con il zip? Non è la soluzione migliore, ma funzionerebbe anche.

Ad esempio in questo modo:

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

Funziona per me su Python 3.

Inoltre: A breakè anche più semplice. (Guarda la risposta di @Pieter)

Una leggera modifica alla risposta di Alex, ma utilizzando __next__():

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

con l' [2]essere il filein root, dirs, filedetto in altre risposte

modifiche alla cartella principale per ogni directory trovata da os.walk. Lo risolvo controllando se root == directory

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