Parcourir les fichiers et les sous-dossiers en Python

Question

J'aimerais parcourir le dossier actuel et tous ses sous-dossiers et obtenir tous les fichiers portant l'extension .htm|.html. J'ai découvert qu'il est possible de savoir si un objet est un dir ou un fichier comme ceci :

import os

dirList = os.listdir("./") # current directory
for dir in dirList:
  if os.path.isdir(dir) == True:
    # I don't know how to get into this dir and do the same thing here
  else:
    # I got file and i can regexp if it is .htm|html

et à la fin, je voudrais avoir tous les fichiers et leurs chemins dans un tableau. Est-ce que cela est possible ?

Answer 1

Vous pouvez utiliser os.walk() pour itérer récursivement à travers un répertoire et tous ses sous-répertoires :

for root, dirs, files in os.walk(path):
    for name in files:
        if name.endswith((".html", ".htm")):
            # whatever

Pour construire une liste de ces noms, vous pouvez utiliser une compréhension de liste :

htmlfiles = [os.path.join(root, name)
             for root, dirs, files in os.walk(path)
             for name in files
             if name.endswith((".html", ".htm"))]

Answer 2

J'avais une chose similaire à faire, et voici comment j'ai procédé.

import os

rootdir = os.getcwd()

for subdir, dirs, files in os.walk(rootdir):
    for file in files:
        #print os.path.join(subdir, file)
        filepath = subdir + os.sep + file

        if filepath.endswith(".html"):
            print (filepath)

J'espère que cela vous aidera.

Answer 3

En python 3, vous pouvez utiliser os.scandir() :

for i in os.scandir(path):
    if i.is_file():
        print('File: ' + i.path)
    elif i.is_dir():
        print('Folder: ' + i.path)

Answer 4

Utilisez newDirName = os.path.abspath(dir) pour créer un nom de chemin d'accès complet pour le sous-répertoire et ensuite lister son contenu comme vous l'avez fait avec le parent (i.e. newDirList = os.listDir(newDirName) )

Vous pouvez créer une méthode distincte de votre extrait de code et l'appeler de manière récursive à travers la structure des sous-répertoires. Le premier paramètre est le chemin d'accès au répertoire. Il sera modifié pour chaque sous-répertoire.

Cette réponse est basée sur la documentation de la version 3.1.1 de la bibliothèque Python. Il existe un bon exemple de ce modèle en action à la page 228 de la référence de la bibliothèque Python 3.1.1 (Chapitre 10 - Accès aux fichiers et aux répertoires). Bonne chance !

Answer 5

Version légèrement modifiée de la solution de Sven Marnach

import os

folder_location = 'C:\SomeFolderName'
file_list = create_file_list(folder_location)

def create_file_list(path):
    return_list = []

for filenames in os.walk(path):
    for file_list in filenames:
        for file_name in file_list:
            if file_name.endswith((".txt")):
                return_list.append(file_name)

return return_list