Python显示所有指定（视频）格式文件

使用os模块找出所有后缀为.mp4/.avi/.rmvb/flv的视频格式文件，并将文件路径添加写入到指定的文本。

# 显示所有视频格式文件，MP4，avi，rmvb，flv
import os
import os.path

def search_file(start_dir,target):
    os.chdir(start_dir)                        # 改变当前工作目录为指定目录
    # os.listdir返回指定目录下的所有文件和目录名，os.curdir返回当前目录('.')
    for each_file in os.listdir(os.curdir):
        ext = os.path.splitext(each_file)[1]   # 分离文件名和拓展名
        if ext in target:
            # os.getcwd获得当前工作目录，os.sep取代操作系统特定的路径分隔符'/','\\'
            # 给出当前平台的行终止符。例如Windows使用'\r\n',linux使用'\n'而mac使用’\r‘
            vedio_list.append(os.getcwd()+os.sep+each_file+os.linesep)  #
        if os.path.isdir(each_file):           # 判断目标是不是目录，不是目录就返回false
            search_file(each_file,target)      #递归调用
            os.chdir(os.pardir)                #递归调用后返回上一层目录

start_dir = input('请输入带待查找的初始目录：').strip()
program_dir = os.getcwd()

target = ['.py','.avi','rmvb','.flv']
vedio_list = []

search_file(start_dir,target)

f = open(program_dir + os.sep + 'vedioList.txt','w')
f.writelines(vedio_list)
f.close()

实现效果：