os.walk sem procurar nos diretórios abaixo

Como faço os.walkpara limitar para retornar apenas arquivos no diretório que eu forneço?

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

Use a walklevelfunção.

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 da mesma forma os.walk, mas você pode passar um levelparâmetro que indica a profundidade da recursão.

Não use os.walk.

Exemplo:

import os

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

Acho que a solução é muito simples.

usar

break

para fazer apenas a primeira iteração do loop for, deve haver uma maneira mais elegante.

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

A primeira vez que você chama os.walk, ele retorna tulipas para o diretório atual e, no próximo loop, o conteúdo do próximo diretório.

Pegue o roteiro original e apenas adicione uma 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

A sugestão de uso listdiré boa. A resposta direta à sua pergunta no Python 2 é root, dirs, files = os.walk(dir_name).next().

A sintaxe Python 3 equivalente é root, dirs, files = next(os.walk(dir_name))

Você pode usar o os.listdir()qual retorna uma lista de nomes (para arquivos e diretórios) em um determinado diretório. Se você precisar distinguir entre arquivos e diretórios, chame os.stat()cada nome.

Se você tiver requisitos mais complexos do que apenas o diretório superior (por exemplo, ignore os diretórios VCS, etc.), você também pode modificar a lista de diretórios para evitar que os.walk volte a eles.

ie:

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

Observação - tome cuidado para alterar a lista, em vez de apenas religá-la. Obviamente, o os.walk não sabe sobre a religação externa.

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

A mesma ideia listdir, mas mais curta:

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

Tive vontade de jogar meus 2 centavos.

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

No Python 3, consegui fazer isso:

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, você pode usar em os.scandirvez de os.listdir. Em vez de strings, você obtém um iterador de DirEntryobjetos em retorno. Dos documentos:

Usar em scandir()vez de listdir()pode aumentar significativamente o desempenho do código que também precisa de informações de tipo de arquivo ou atributo de arquivo, porque os DirEntryobjetos expõem essas informações se o sistema operacional as fornecer durante a varredura de um diretório. Todos os DirEntrymétodos podem realizar uma chamada de sistema, mas is_dir()e is_file()normalmente requerem apenas uma chamada de sistema para links simbólicos; DirEntry.stat()sempre requer uma chamada de sistema no Unix, mas requer apenas uma para links simbólicos no Windows.

Você pode acessar o nome do objeto por meio do DirEntry.namequal é então equivalente à saída deos.listdir

Você também pode fazer o seguinte:

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

Foi assim que resolvi

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

...

Há um problema ao usar listdir. O os.path.isdir (identificador) deve ser um caminho absoluto. Para escolher subdiretórios, você:

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

A alternativa é mudar para o diretório para fazer o teste sem o os.path.join ().

Você pode usar este snippet

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

crie uma lista de exclusões, use fnmatch para pular a estrutura de diretório e fazer o 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
            ....

o mesmo que para 'inclui':

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

Por que não usar simplesmente um rangee os.walkcombinado com o zip? Não é a melhor solução, mas também funcionaria.

Por exemplo, assim:

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

Funciona para mim no python 3.

Além disso: A breaké mais simples, aliás. (Veja a resposta de @Pieter)

Uma ligeira mudança na resposta de Alex, mas usando __next__():

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

com a [2]ser o fileno root, dirs, filemencionado em outras respostas

A pasta raiz muda para cada diretório que os.walk encontrar. Eu resolvo isso verificando se root == diretório

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