练习:查找指定目录(包括子目录)下的视频(格式为.mp4,.rmvb,.avi),并将目录存放在一个文件中

时间:2021-01-26 12:48:31
import os


def search_file(start_dir,target):
    os.chdir(start_dir)  #切换目录
    for each_file in os.listdir(os.getcwd()):#当前目录的所有文件及文件夹

        if os.path.isfile(each_file): #如果是文件
            if os.path.splitext(each_file)[1] in target:  #判断文件后缀是否是查找对象
                resultlst.append(os.path.join(os.getcwd(),each_file)+"\n")  #如果是,先存到一个list中

        if os.path.isdir(each_file):  #如果是文件夹
             search_file(each_file,target) #进入文件夹搜索,递归搜索
             os.chdir(os.pardir) #返回上一层目录


start_dir="C:\\Program Files"  #指定目录
target=".mp4,.rmvb,.avi"       #查找文件类型
resultlst=[]                   #存放文件路径list
search_file(start_dir,target)


f=open("C:\\result.txt","w")
f.writelines(resultlst)       #将列表写入文件,如果没有换行,请在list.append()的时候加上换行符"\n"

f.close()