In Python, il modo più veloce per creare un elenco di file in una directory con una certa estensione

Question

In Python su un sistema GNU/Linux, qual è il modo più veloce per scansionare ricorsivamente una directory per tutti i file .MOV o .AVI e per memorizzarli in una lista?

Answer 1

Esempio per un elenco di file nella directory corrente. È possibile espandere questo per percorsi specifici.

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

Answer 2

È possibile utilizzare os.walk() per camminare recuresive e glob.glob() o fnmatch.filter() per la corrispondenza del file:

Scegli questa answer

Answer 3

userei os .walk per scansionare la directory, os.path.splitext per afferrare il suffisso e filtrarli da solo.

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)) 

Utilizzare uno matching_ext o matching_ext_ci con argomenti la cartella principale e una proroga o una tupla di estensioni:

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

Answer 6

Suggerisco l'uso di os.walk e un attentamente reading of its documentation.

Questo può essere un approccio uno di linea:

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

Dove in is_video di controllare le estensioni.