Browse files and subfolders in Python

Question

I'd like to browse through the current folder and all its subfolders and get all the files with .htm|.html extensions. I have found out that it is possible to find out whether an object is a dir or file like this:

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 

and in the end, I would like to have all the files and their paths in an array. Is something like that possible?

Answer 1

You can use os.walk() to recursively iterate through a directory and all its subdirectories:

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

To build a list of these names, you can use a list comprehension:

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

I had a similar thing to work on, and this is how I did it.

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) 

Hope this helps.