我想写一个程序来获取计算机中的电影列表。我认为这有两个步骤:
目前,我对如何完成第一步感兴趣。我在哪里可以找到热门视频文件扩展名列表?如何在整个计算机(在python中)搜索这些文件?
答案 0 :(得分:4)
通过简单的谷歌搜索,您将从许多地方获得视频文件扩展名列表。维护一个扩展列表并遍历目录路径,看看文件是否以任何一个结束。
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)