os.walk sin buscar en los directorios siguientes

¿Cómo me limito os.walka devolver solo archivos en el directorio que proporciono?

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 walklevelfunción.

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

Funciona igual os.walk, pero puede pasarle un levelparámetro que indique qué tan profunda será la recursividad.

No use os.walk.

Ejemplo:

import os

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

Creo que la solución es realmente muy sencilla.

utilizar

break

para hacer solo la primera iteración del ciclo for, debe haber una forma más elegante.

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

La primera vez que llama a os.walk, devuelve tulipanes para el directorio actual, luego, en el siguiente ciclo, el contenido del siguiente directorio.

Tome el guión original y agregue un descanso .

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

La sugerencia de uso listdires buena. La respuesta directa a su pregunta en Python 2 es root, dirs, files = os.walk(dir_name).next().

La sintaxis equivalente de Python 3 es root, dirs, files = next(os.walk(dir_name))

Puede usar os.listdir()que devuelve una lista de nombres (tanto para archivos como para directorios) en un directorio determinado. Si necesita distinguir entre archivos y directorios, llame os.stat()a cada nombre.

Si tiene requisitos más complejos que solo el directorio superior (por ejemplo, ignore los directorios VCS, etc.), también puede modificar la lista de directorios para evitar que os.walk vuelva a recorrerlos.

es decir:

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: tenga cuidado de mutar la lista, en lugar de simplemente volver a enlazarla. Obviamente, os.walk no conoce la reencuadernación externa.

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

La misma idea con listdir, pero más breve:

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

Tenía ganas de tirar mis 2 peniques.

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

En Python 3, pude hacer esto:

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

Desde Python 3.5 puede usar en os.scandirlugar de os.listdir. En lugar de cadenas, obtienes un iterador de DirEntryobjetos a cambio. De los documentos:

Usar en scandir()lugar de listdir()puede aumentar significativamente el rendimiento del código que también necesita información sobre el tipo de archivo o los atributos del archivo, porque los DirEntryobjetos exponen esta información si el sistema operativo la proporciona al escanear un directorio. Todos los DirEntrymétodos pueden realizar una llamada al sistema, pero is_dir(), y is_file()por lo general sólo requiere una llamada al sistema para los enlaces simbólicos; DirEntry.stat()siempre requiere una llamada al sistema en Unix pero solo requiere una para enlaces simbólicos en Windows.

Puede acceder al nombre del objeto a través del DirEntry.namecual es equivalente a la salida deos.listdir

También puede hacer lo siguiente:

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

Así es como lo resolví

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

...

Hay un problema al usar listdir. Os.path.isdir (identificador) debe ser una ruta absoluta. Para elegir subdirectorios debe:

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

La alternativa es cambiar al directorio para realizar la prueba sin os.path.join ().

Puedes usar este fragmento

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

cree una lista de exclusiones, use fnmatch para omitir la estructura del directorio y realice el proceso

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

lo mismo que para 'incluye':

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

¿Por qué no simplemente usar una rangey os.walkcombinado con el zip? No es la mejor solución, pero también funcionaría.

Por ejemplo así:

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

Funciona para mí en Python 3.

Además: A también breakes más simple por cierto. (Mira la respuesta de @Pieter)

Un ligero cambio en la respuesta de Alex, pero usando __next__():

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

con el [2]ser el filede root, dirs, filemencionado en otras respuestas

La carpeta raíz cambia para cada directorio que encuentra os.walk. Solucioné eso comprobando si root == directorio

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