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

Question

Comment limiter os.walk pour ne renvoyer que les fichiers dans le répertoire que je lui donne ?

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_("ignorer")
    return outputList

Answer 1

Ne pas utiliser os.walk.

Exemple :

import os

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

Answer 2

Utilisez la fonction walklevel.

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 paramètre level qui indique la profondeur de la récursion.

Answer 3

Je pense que la solution est en réalité très simple.

utilise

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 tuples pour le répertoire actuel, puis à la prochaine itération le contenu du répertoire suivant.

Prenez le script original et ajoutez simplement un break.

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

Answer 4

La suggestion d'utiliser listdir est bonne. La réponse directe à votre question en Python 2 est root, dirs, files = os.walk(dir_name).next().

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

Answer 5

Vous pourriez 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 distinguer les fichiers des répertoires, appelez os.stat() sur chaque nom.