[已解决]《课后作业三十讲》第3小题对于自己所写答案的疑惑

我总得懂点什么

1. import os
   
2. 
   
3. def search(start_path):
   
4. 
   
5.     os.chdir(start_path)
   
6. 
   
7.     namelst = os.listdir(os.curdir)
   
8. 
   
9.     extension = ['.mp4','.rmvb','.avi']
   
10. 
    
11.     for each_file in namelst:
    
12. 
    
13.         if os.path.isfile(each_file):
    
14. 
    
15.             if os.path.splitext(each_file) in extension:
    
16. 
    
17.                 f = open('D:\\vedioList.txt','x')
    
18. 
    
19.                 final_path = start_path + each_file
    
20. 
    
21.                 f.writelines(final_path)
    
22. 
    
23.                 f.close()
    
24. 
    
25.         if os.path.isdir(each_file):
    
26. 
    
27.             search(each_file)
    
28. 
    
29.             os.chdir(os.pardir)
    
30. 
    
31. 
    
32. start_path = input('请输入待查找的初始目录：')
    
33. 
    
34. search(start_path)
    
35. 
    

复制代码

在运行过后没有报错也没有结果，这是为什么呢？

还有一个小问题：小甲鱼的片名能搜到嘛？

最佳答案

月排行榜 / 总排行榜

jackz007

2021-2-6 18:24:17

1. import os
   
2. 
   
3. def search(start_path):
   
4.     pwd = os . getcwd()
   
5.     os . chdir(start_path)
   
6.     cwd = os . getcwd()
   
7.     extension = ['.mp4','.mkv' ,'.flv' , '.rmvb' , '.rm' , '.wmv' , '.avi']
   
8.     try:
   
9.         for each_file in os.listdir(os . curdir):
   
10.             if os . path . isfile(each_file):
    
11.                 if os . path . splitext(each_file)[1] . lower() in extension:
    
12.                     f = open('D:\\vedioList.txt','a')
    
13.                     final_path = os . path . join(cwd , each_file)
    
14.                     f . write(final_path + '\n')
    
15.                     f . close()
    
16.             elif os . path . isdir(each_file):
    
17.                 search(each_file)
    
18.     except Exception as e:
    
19.         print(e)
    
20.     os . chdir(pwd)     
    
21. start_path = input('请输入待查找的初始目录：')
    
22. if os . path . exists('D:\\vedioList.txt') and  os . path . isfile('D:\\vedioList.txt'):
    
23.     os . remove('D:\\vedioList.txt')
    
24. search(start_path)

复制代码

跳转到最佳答案楼层

jackz007

1. import os
   
2. 
   
3. def search(start_path):
   
4.     pwd = os . getcwd()
   
5.     os . chdir(start_path)
   
6.     cwd = os . getcwd()
   
7.     extension = ['.mp4','.mkv' ,'.flv' , '.rmvb' , '.rm' , '.wmv' , '.avi']
   
8.     try:
   
9.         for each_file in os.listdir(os . curdir):
   
10.             if os . path . isfile(each_file):
    
11.                 if os . path . splitext(each_file)[1] . lower() in extension:
    
12.                     f = open('D:\\vedioList.txt','a')
    
13.                     final_path = os . path . join(cwd , each_file)
    
14.                     f . write(final_path + '\n')
    
15.                     f . close()
    
16.             elif os . path . isdir(each_file):
    
17.                 search(each_file)
    
18.     except Exception as e:
    
19.         print(e)
    
20.     os . chdir(pwd)     
    
21. start_path = input('请输入待查找的初始目录：')
    
22. if os . path . exists('D:\\vedioList.txt') and  os . path . isfile('D:\\vedioList.txt'):
    
23.     os . remove('D:\\vedioList.txt')
    
24. search(start_path)

复制代码