En Python, la forma más rápida de crear una lista de archivos en un directorio con una cierta extensión

Question

En Python en un sistema GNU/Linux, ¿cuál es la forma más rápida de escanear recursivamente un directorio para todos los archivos .MOV o .AVI, y almacenarlos en una lista?

Answer 1

Ejemplo de una lista de archivos en el directorio actual. Puede expandir esto para rutas específicas.

import glob 
movlist = glob.glob('*.mov') 

Answer 2

Puede utilizar os.walk() para caminar y recuresive glob.glob() o fnmatch.filter() para la coincidencia de archivo:

Marque esta answer

Answer 3

que haría uso de OS .walk para escanear el directorio, os.path.splitext para tomar el sufijo y filtrarlo yo mismo.

suffixes = set(['.AVI', '.MOV']) 
for dirpath, dirnames, filenames in os.walk('.'): 
    for f in filenames: 
     if os.path.splitext(f)[1] in suffixes: 
      yield os.path.join(dirpath, f) 

Answer 4

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 

Answer 5

Python 2.x:

import os 

def generic_tree_matching(rootdirname, filterfun): 
    return [ 
     os.path.join(dirname, filename) 
     for dirname, dirnames, filenames in os.walk(rootdirname) 
     for filename in filenames 
     if filterfun(filename)] 

def matching_ext(rootdirname, extensions): 
    "Case sensitive extension matching" 
    return generic_tree_matching(
     rootdirname, 
     lambda fn: fn.endswith(extensions)) 

def matching_ext_ci(rootdirname, extensions): 
    "Case insensitive extension matching" 
    try: 
     extensions= extensions.lower() 
    except AttributeError: # assume it's a sequence of extensions 
     extensions= tuple(
      extension.lower() 
      for extension in extensions) 
    return generic_tree_matching(
     rootdirname, 
     lambda fn: fn.lower().endswith(extensions)) 

uso ya sea matching_ext o matching_ext_ci con argumentos la carpeta raíz y una extensión o una tupla de extensiones:

>>> matching_ext(".", (".mov", ".avi")) 

Answer 6

Sugiero el uso de os.walk y cuidadosamente reading of its documentation.

Esto puede ser un enfoque de un trazador de líneas:

[f for root,dirs,files in os.walk('/your/path') for f in files if is_video(f)] 

Donde en is_video comprobar sus extensiones.