第30讲的第四题

Bruin

1. import os
   
2. 
   
3. def vedio_file(start_file):
   
4. 
   
5.     os.chdir(start_file)
   
6.     all_file = os.listdir(os.curdir)
   
7.     for each_file in all_file:
   
8.         if os.path.splitext(each_file)[1] in ['.mp4','.rmvb','.avi']:
   
9.             vedioList.append(os.getcwd() + os.sep + each_file +os.linesep)
   
10. 
    
11.         if os.path.isfile(each_file):
    
12.             vedio_file(each_file)
    
13.             os.chdir(os.pardir)
    
14. 
    
15. start_file = input('请输入待查找的初始目录：')
    
16. vedioList = []
    
17. 
    
18. vedio_file(start_file)
    
19. 
    
20. f = open(os.getcwd() + os.sep + 'vedioList.txt','w')
    
21. f.writelines(vedioList)
    
22. 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。