2011-01-07 72 views
0

我想寫一個程序來獲得電影列表的電腦。我認爲有2個步驟:什麼是流行的視頻格式

  1. 獲取計算機中所有視頻文件的列表。
  2. 找出哪些是電影。

目前,我對如何完成第一步感興趣。我在哪裏可以找到流行的視頻文件擴展名列表?我怎樣才能搜索整個計算機(在Python中)這些文件?

回答

4

您將通過執行簡單的Google搜索從許多地方獲得視頻文件擴展名列表。維護一個擴展名列表並遍歷目錄路徑,看看文件是否以它們中的任何一個結尾。

list_of_extensions = ['flv','mp4','avi'] # Supplement it with what you find. 
list_of_files = [] 


import os 

def callback(arg, directory, files): 
    for file in files: 
     extn = file.split('.')[-1] 
     if extn in arg: 
       list_of_files.append(os.path.join(directory,file)) 

os.path.walk(".", callback, list_of_extensions) 
相關問題