qin_yin 发表于 2020-10-3 19:35:31

30课后答案

import os

def search_file(start_dir, target) :
    os.chdir(start_dir)
   
    for each_file in os.listdir(os.curdir) :
      ext = os.path.splitext(each_file)
      if ext in target :
            vedio_list.append(os.getcwd() + os.sep + each_file + os.linesep) # 使用os.sep是程序更标准
      if os.path.isdir(each_file) :
            search_file(each_file, target) #这里函数是怎么调用成功的,在我看来each_file难道不是一个文件名吗,并没有进入具体路径，因此它也无法正确的修改当前路径，但事实不是这样，这是为什么呢
            os.chdir(os.pardir) #

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

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()

疾风怪盗 发表于 2020-10-3 19:38:49

os.chdir(start_dir)
这句话的作用就是更改当前目录，这样不需要路径，就文件名，就可以查找同级目录下的文件
函数里增加一句os.getcwd()，可以查看当前路径
页: [1]
查看完整版本: 30课后答案