在目錄中遞歸搜索具有特定擴展名的文件

Question

由於某種原因，這會返回一個空列表，我不知道爲什麼。

import os, fnmatch 

vidext = ['.avi', '.mkv', '.wmv', '.mp4', '.mpg', '.mpeg', '.mov', '.m4v'] 

def findExt(folder): 
    matches = [] 
    for root, dirnames, filenames in os.walk(folder): 
     for extension in vidext: 
      for filename in fnmatch.filter(filenames, extension): 
       matches.append(os.path.join(root, filename)) 
    return matches 

print(findExt("D:\TVRip")) 

Answer 1

你需要一個通配符添加到每個擴展fnmatch.filter()匹配：

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

但沒有必要在這裏使用fnmatch可言。只需使用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)) 

或表示爲一個列表理解：

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