第30讲的第四题

Bruin

import os

def vedio_file(start_file):

    os.chdir(start_file)
    all_file = os.listdir(os.curdir)
    for each_file in all_file:
        if os.path.splitext(each_file)[1] in ['.mp4','.rmvb','.avi']:
            vedioList.append(os.getcwd() + os.sep + each_file +os.linesep)

        if os.path.isfile(each_file):
            vedio_file(each_file)
            os.chdir(os.pardir)

start_file = input('请输入待查找的初始目录：')
vedioList = []

vedio_file(start_file)

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

这是我自己写的，为啥不对啊

我运行小甲鱼的程序也是有问题。。。

白昼之殇

import os

def vedio_file(start_file):
    vedioList = []
    os.chdir(start_file)
    all_file = os.listdir(os.getcwd())
    for each_file in all_file:
        if os.path.splitext(each_file)[1] in ['.mp4','.rmvb','.avi']:
            vedioList.append(os.getcwd() + os.sep + each_file +os.linesep)

    return vedioList
start_file = input('请输入待查找的初始目录：')


vedioList = vedio_file(start_file)

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

代码帮你改过了，去除了递归文件夹的代码。
原函数几个问题：
1、原函数没有返回值。
2、os.curdir是动作不是路径
3、不要自己递归查找文件夹，可以用os.walk。