零基础入门学习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) #分离文件名与扩展名，返回元组。 的意思是指把扩展名赋值给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()
页:
[1]