Recursively searching for files with specific extensions in a directory

Question

For some reason this returns me an empty list, and I have no idea why.

import os, fnmatch

vidext = [\'.avi\', \'.mkv\', \'.wmv\', \'.mp4\', \'.mpg\', \'.mpeg\',         


        

Answer 1

You'd need to add a wildcard to each extension for fnmatch.filter() to match:

fnmatch.filter(filenames, '*' + extension)

but there is no need to use fnmatch here at all. Just use str.endswith():

for root, dirnames, filenames in os.walk(folder):
    for filename in filenames:
        if filename.endswith(extensions):
            matches.append(os.path.join(root, filename))

or expressed as a list comprehension:

return [os.path.join(r, fn)
        for r, ds, fs in os.walk(folder) 
        for fn in fs if fn.endswith(extensions)]