Utiliser Python os.walk pour identifier une liste de fichiers

Question

Duplicata possible :
En Python, le moyen le plus rapide de construire une liste de fichiers dans un répertoire avec une certaine extension

J'utilise actuellement os.walk pour parcourir récursivement un répertoire en identifiant les fichiers .MOV.

def fileList():
    matches = []
    for root, dirnames, filenames in os.walk(source):
        for filename in fnmatch.filter(filenames, '*.mov'):
            matches.append(os.path.join(root, filename))
    return matches

Comment puis-je étendre cela pour identifier plusieurs fichiers, par exemple, .mov, .MOV, .avi, .mpg ?

Gracias.

Answer 1

Essayez quelque chose comme ça :

def fileList(source):
    matches = []
    for root, dirnames, filenames in os.walk(source):
        for filename in filenames:
            if filename.endswith(('.mov', '.MOV', '.avi', '.mpg')):
                matches.append(os.path.join(root, filename))
    return matches

Answer 2

Une alternative utilisant os.path.splitext :

for root, dirnames, filenames in os.walk(source):
    filenames = [ f for f in filenames if os.path.splitext(f)[1] in ('.mov', '.MOV', '.avi', '.mpg') ]
    for filename in filenames:
        matches.append(os.path.join(root, filename))
return matches

Answer 3

pattern = re.compile('.*\.(mov|MOV|avi|mpg)$')

def fileList(source):
   matches = []
   for root, dirnames, filenames in os.walk(source):
       for filename in filter(lambda name:pattern.match(name),filenames):
           matches.append(os.path.join(root, filename))
   return matches