os.walk sans fouiller dans les répertoires ci-dessous

Comment limiter le os.walkretour des fichiers dans le répertoire que je fournis?

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

Utilisez la walklevelfonction.

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

Cela fonctionne exactement comme os.walk, mais vous pouvez lui passer un levelparamètre qui indique la profondeur de la récursivité.

N'utilisez pas os.walk.

Exemple:

import os

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

Je pense que la solution est en fait très simple.

utilisation

break

pour ne faire que la première itération de la boucle for, il doit y avoir une manière plus élégante.

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

La première fois que vous appelez os.walk, il renvoie des tulipes pour le répertoire courant, puis lors de la boucle suivante, le contenu du répertoire suivant.

Prenez le script original et ajoutez simplement une pause .

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 suggestion à utiliser listdirest bonne. La réponse directe à votre question dans Python 2 estroot, dirs, files = os.walk(dir_name).next() .

La syntaxe Python 3 équivalente est root, dirs, files = next(os.walk(dir_name))

Vous pouvez utiliser os.listdir()qui renvoie une liste de noms (pour les fichiers et les répertoires) dans un répertoire donné. Si vous avez besoin de faire la distinction entre les fichiers et les répertoires, appelez os.stat()chaque nom.

Si vous avez des exigences plus complexes que le seul répertoire supérieur (par exemple, ignorer les répertoires VCS, etc.), vous pouvez également modifier la liste des répertoires pour empêcher os.walk de les parcourir.

c'est à 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()

Remarque - veillez à modifier la liste, plutôt que de simplement la relier. De toute évidence, os.walk ne connaît pas la reliure externe.

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

La même idée avec listdir , mais en plus court:

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

J'avais l'impression de jeter mes 2 pence.

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, j'ai pu faire ceci:

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

Depuis Python 3.5, vous pouvez utiliser à la os.scandirplace de os.listdir. Au lieu de chaînes, vous obtenez un itérateur d' DirEntryobjets en retour. À partir de la documentation:

L'utilisation scandir()au lieu de listdir()peut augmenter considérablement les performances du code qui a également besoin d'informations sur le type de fichier ou les attributs de fichier, car les DirEntryobjets exposent ces informations si le système d'exploitation les fournit lors de l'analyse d'un répertoire. Toutes les DirEntryméthodes peuvent effectuer un appel système, mais is_dir()et is_file()nécessitent généralement un appel système pour les liens symboliques; DirEntry.stat()nécessite toujours un appel système sous Unix mais n'en nécessite qu'un pour les liens symboliques sous Windows.

Vous pouvez accéder au nom de l'objet via DirEntry.namequi équivaut alors à la sortie deos.listdir

Vous pouvez également effectuer les opérations suivantes:

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

Voilà comment je l'ai résolu

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

...

Il y a un problème lors de l'utilisation de listdir. Le os.path.isdir (identificateur) doit être un chemin absolu. Pour choisir les sous-répertoires que vous faites:

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

L'alternative est de passer au répertoire pour faire les tests sans os.path.join ().

Vous pouvez utiliser cet extrait

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

créer une liste d'exclusions, utiliser fnmatch pour ignorer la structure du répertoire et effectuer le processus

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

idem pour «comprend»:

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

Pourquoi ne pas simplement utiliser un rangeet os.walkcombiné avec le zip? Ce n'est pas la meilleure solution, mais cela fonctionnerait aussi.

Par exemple comme ceci:

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

Fonctionne pour moi sur python 3.

Aussi: A breakest plus simple trop btw. (Regardez la réponse de @Pieter)

Un léger changement dans la réponse d'Alex, mais en utilisant __next__() :

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

avec l' [2]étant fileen root, dirs, filementionné dans d' autres réponses

le dossier racine change pour chaque répertoire trouvé par os.walk. Je résolve cela en vérifiant si le répertoire racine ==

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