零基础入门学习python第30讲课后习题

炼己者

第三题
import os

def search_file(start_dir, target) :
    os.chdir(start_dir)  #改变工作目录
   
    for each_file in os.listdir(os.curdir) :     #把当前目录的文件名赋值给each_file
        ext = os.path.splitext(each_file)[1]       #分离文件名与扩展名，返回元组。 [1]的意思是指把扩展名赋值给ext
        if ext in target :        #判断假设ext在target里面
            vedio_list.append(os.getcwd() + os.sep + each_file + os.linesep)                                                                                                                                                # 就把(当前工作目录+操作系统特定的路径分隔符+目录的文件名+行中止符)添加到vedio_list列表中
        
        if os.path.isdir(each_file) :         #判断each_file是否存在且是一个目录，假设存在
            search_file(each_file, target)         # 递归调用
            os.chdir(os.pardir)           # 递归调用后切记返回上一层目录
            #查找该路径下(包括子文件夹)所有的视频格式文件
            
             #为什么递归后要返回上一层目录    这是铁棍阿童木 鱼油答的，醍醐灌顶啊
            #第一：一定要记住一点，递归就是先递进后回归，有去有回才能正确执行；
             #第二：每次进入子目录后，都会将其变更为工作目录，所以执行完搜索后你不退回到父级目录就不会遍历所有文件夹啊

start_dir = input('请输入待查找的初始目录：')
program_dir = os.getcwd()      #把当前工作目录赋值给program_dir

target = ['.mp4', '.avi', '.rmvb']
vedio_list = []

search_file(start_dir, target)#调用函数

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