我的代码问题在哪里呢——30课后题第三题求助

selene

先上课后题的题目，如图：



之后我的代码如下：编译器给我反馈的错误在16、8、5行，但是我觉得，我没错啊，然后，对比了一下小甲鱼的答案，觉得，虽然没有弄成列表，但是意思是一致的啊，请大神们帮我看看我错在了哪里，求指点。。。

1. import os
   
2. 
   
3. def find_file(search_dir):
   
4.     os.chdir(search_dir)
   
5.     all_file = os.listdir(search_dir)
   
6.     for each_file in all_file:
   
7.         if os.path.isdir(each_file):
   
8.             find_file(each_file)
   
9.             os.chdir(os.pardir)  # 递归调用后切记返回上一层目录
   
10.         elif (os.path.splitext(each_file)[1] == '.mp4') \
    
11.                 or(os.path.splitext(each_file)[1] == '.rmvb') \
    
12.                 or(os.path.splitext(each_file)[1] == '.avi'):
    
13.             f.write(os.getcwd() + os.sep + each_file + os.linesep)
    
14. search_dir = input('请输入需要搜索的路径：')
    
15. f = open('vediolist.txt', 'w')
    
16. find_file(search_dir)
    
17. f.close()

复制代码

小甲鱼的标准答案如下：

selene

好了 自己调试出来了 我的代码第五行， 应该是 all_file = os.listdir(os.curdir)
因为严格意义上来说，这才是表示当前路径下的所有文件，大家以后要当心

南城顾她

看不到你发的啥答案

selene

南城顾她 发表于 2018-4-4 16:21
看不到你发的啥答案

刚刚重新编辑了一下帖子，现在你看看，应该有了

ba21

请上代码。难不成还要别人帮你打一遍

selene

ba21 发表于 2018-4-4 16:31
请上代码。难不成还要别人帮你打一遍

上了上了

ba21

1. import os
   
2. 
   
3. ext = ['.mp4', '.rmvb', '.avi']
   
4. search_file = []
   
5. def find_file(search_dir):   
   
6.     os.chdir(search_dir)
   
7.     for each_file in os.listdir():
   
8.         if os.path.isdir(each_file):
   
9.             find_file(each_file)
   
10.             os.chdir(os.pardir)  # 递归调用后切记返回上一层目录
    
11.         elif os.path.splitext(each_file)[1] in ext:
    
12.             search_file.append(os.getcwd() + os.sep + each_file + os.linesep)
    
13.             
    
14. search_dir = input('请输入需要搜索的路径：')
    
15. 
    
16. f = open('d:/vediolist.txt', 'w')
    
17. find_file(search_dir)
    
18. f.writelines(search_file)
    
19. f.close()
    

复制代码

selene

ba21 发表于 2018-4-4 16:55

谢谢你的回复，但我原来的那个代码错在哪里呢。。。您只是在我的代码的基础上，结合了标准答案，给出一个正确的答案，但是我的那个，依然不知道为什么错。。。

ba21

selene 发表于 2018-4-4 18:00
好了 自己调试出来了 我的代码第五行， 应该是 all_file = os.listdir(os.curdir)
因为严格意义上来说，这 ...

代码都上了，真的不会对比吗？
'.mp4', '.rmvb', '.avi'  os.path.splitext(each_file)[1] == 'mp4'
os.listdir() all_file = os.listdir(search_dir)