Trong Python, cách nhanh nhất để xây dựng danh sách các tệp trong một thư mục với một số mở rộng nhất định

Question

Trong Python trên hệ thống GNU/Linux, cách nhanh nhất để đệ quy quét thư mục cho tất cả các tệp .MOV hoặc .AVI và lưu trữ chúng trong một danh sách?

Answer 1

Ví dụ về danh sách tệp trong thư mục hiện tại. Bạn có thể mở rộng điều này cho các đường dẫn cụ thể.

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

Answer 2

Bạn có thể sử dụng os.walk() để đi bộ recuresive và glob.glob() hoặc fnmatch.filter() cho phù hợp với file:

Kiểm tra này answer

Answer 3

Tôi muốn sử dụng os .walk để quét thư mục, os.path.splitext để lấy hậu tố và tự lọc chúng.

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

Sử dụng một trong hai matching_ext hoặc matching_ext_ci với đối số thư mục gốc và một phần mở rộng hoặc một tuple các phần mở rộng:

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

Answer 6

Tôi đề nghị việc sử dụng os.walk và cẩn thận reading of its documentation.

Đây có thể là một cách tiếp cận một liner:

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

đâu trong is_video bạn kiểm tra phần mở rộng của bạn.