[已解决]第30讲第三题

zhongyuanadler

我自己写的是，但是感觉不好运作，我自己的思路是只要一个参数就是路径，然后每次遍历时写入videolist文件，写完再关闭，但是保存位置也不确定在哪，之后再递归。 但是总感觉自己摸不透自己的问题在哪？ 麻烦大家看看了。谢谢了！

1. import os
   
2. ext = ['.mp4','.rmvb','.avi']
   
3. def search_video(file_path):
   
4.          allfiles = os.listdir(file_path)
   
5.          for eachfile in allfiles:
   
6.              if os.path.splitext(eachfile)[1] in ext:
   
7.                  f = open('videolist.txt','w')
   
8.                  f.write(str(os.path.dirname(eachfile) + '\\' +str(os.path.basename(eachfile))))
   
9.                  f.close()
   
10.              if os.path.isfile(eachfile):
    
11.                  search_video(eachfile)
    
12.                  os.chdir(os.pardir)
    
13. 
    
14. filepath = input('路径：')
    
15. search_video(filepath)
    

复制代码

答案是

1. import os
   
2. 
   
3. def search_file(start_dir, target) :
   
4.     os.chdir(start_dir)
   
5.    
   
6.     for each_file in os.listdir(os.curdir) :
   
7.         ext = os.path.splitext(each_file)[1]
   
8.         if ext in target :
   
9.             vedio_list.append(os.getcwd() + os.sep + each_file + os.linesep) # 使用os.sep是程序更标准
   
10.         if os.path.isdir(each_file) :
    
11.             search_file(each_file, target) # 递归调用
    
12.             os.chdir(os.pardir) # 递归调用后切记返回上一层目录
    
13. 
    
14. start_dir = input('请输入待查找的初始目录：')
    
15. program_dir = os.getcwd()
    
16. 
    
17. target = ['.mp4', '.avi', '.rmvb']
    
18. vedio_list = []
    
19. 
    
20. search_file(start_dir, target)
    
21. 
    
22. f = open(program_dir + os.sep + 'vedioList.txt', 'w')
    
23. f.writelines(vedio_list)
    
24. f.close()
    

复制代码

我可以理解答案是把采集到的文件名存在列表里最后再写入txt，但是txt文件保存位置是哪？

最佳答案

月排行榜 / 总排行榜

txxcat

2020-4-14 17:37:25

先说几个明显得错误：

1. import os
   
2. ext = ['.mp4','.rmvb','.avi']
   
3. def search_video(file_path):
   
4.          allfiles = os.listdir(file_path)
   
5.          for eachfile in allfiles:
   
6.              if os.path.splitext(eachfile)[1] in ext:
   
7.                  f = open('videolist.txt','w')            #用w打开文件都会覆盖掉老文件，而且，没有绝对路径名，在每个符合条件的目录下都生成一个文件，乱！
   
8.                  f.write(str(os.path.dirname(eachfile) + '\\' +str(os.path.basename(eachfile))))
   
9.                  f.close()
   
10.              if os.path.isfile(eachfile):                #不该是os.path.isdir ???
    
11.                  search_video(eachfile)
    
12.                  os.chdir(os.pardir)                    #之前又没有chdir，现在chdir，混乱了
    
13. 
    
14. filepath = input('路径：')
    
15. search_video(filepath)

复制代码

再修正一下代码：

1. import os
   
2. ext = ['.mp4','.rmvb','.avi']
   
3. def search_video(file_path):
   
4.     allfiles = os.listdir('.')                    #已在目标目录，涉及到递归调用，不能使用目录名
   
5.     for eachfile in allfiles:
   
6.         if os.path.splitext(eachfile)[1] in ext:
   
7.             f = open(cdir+'\\'+'videolist.txt','a')                                               #要使用绝对路径名
   
8.             f.write(os.getcwd() + '\\' +str(os.path.basename(eachfile))+'\n')      #os.getcwd()才能获取完整的路径     
   
9.             f.close()
   
10.         if os.path.isdir(eachfile):
    
11.             os.chdir(eachfile)               #转到新的目标目录
    
12.             search_video(eachfile)        #递归调用
    
13.             os.chdir(os.pardir)              #回到上一级目录
    
14. 
    
15. filepath = input('路径：')
    
16. cdir=os.getcwd()           #记录工作目录
    
17. os.chdir(filepath)           #去到目标目录
    
18. search_video(filepath)
    
19. os.chdir(cdir)                #回到工作目录
    

复制代码

代码是可以跑了，不过还有些不合理的地方，比如你每发现一个视频文件，就打开一次写入再关闭，动作频繁，应该程序开始就打开，然后发现写入，程序结束时关闭才合理：

1. def search_video(file_path):
   
2.     allfiles = os.listdir('.')
   
3.     for eachfile in allfiles:
   
4.         if os.path.splitext(eachfile)[1] in ext:
   
5.             f.write(os.getcwd() + '\\' +str(os.path.basename(eachfile))+'\n')
   
6.         if os.path.isdir(eachfile):
   
7.             os.chdir(eachfile)
   
8.             search_video(eachfile)
   
9.             os.chdir(os.pardir)
   
10. 
    
11. import os
    
12. ext = ['.mp4','.rmvb','.avi']
    
13. f = open('videolist.txt','w')     #这回可以用w，保证每次写入的是新数据
    
14. filepath = input('路径：')
    
15. cdir=os.getcwd()
    
16. os.chdir(filepath)
    
17. search_video(filepath)
    
18. os.chdir(cdir)
    
19. f.close()

复制代码

跳转到最佳答案楼层

老八秘制

默认和代码一个目录

zhongyuanadler

这道题自己做的有点晕 隔了两天没复习os模块的指令 就有点不知所措了

wuqramy

在程序代码的同一个目录下，如果指定位置保存可以在文件名前加上文件路径

wuqramy

zhongyuanadler 发表于 2020-4-14 14:17
这道题自己做的有点晕 隔了两天没复习os模块的指令 就有点不知所措了

那你就收藏一下吧 --> https://fishc.com.cn/thread-45512-1-1.html

zhongyuanadler

wuqramy 发表于 2020-4-14 14:20
那你就收藏一下吧 --> https://fishc.com.cn/thread-45512-1-1.html

嗯 谢谢 我是开着这个扩展阅读 一边看一边做的

zhongyuanadler

老八秘制 发表于 2020-4-14 14:16
默认和代码一个目录

恩恩 谢谢，有时间看看我的代码的问题是什么吗？

wuqramy

zhongyuanadler 发表于 2020-4-14 14:22
恩恩 谢谢，有时间看看我的代码的问题是什么吗？

你发现了什么问题呢

zhongyuanadler

wuqramy 发表于 2020-4-14 14:28
你发现了什么问题呢

跑了一下代码，输入路径什么反应都没有，而且文件都没创建

永恒的蓝色梦想

zhongyuanadler 发表于 2020-4-14 14:22
恩恩 谢谢，有时间看看我的代码的问题是什么吗？

你的代码缩进有问题

zhongyuanadler

永恒的蓝色梦想 发表于 2020-4-14 14:49
你的代码缩进有问题

改成这样了：

1. import os
   
2. ext = ['.mp4','.rmvb','.avi']
   
3. def search_video(file_path):
   
4.     allfiles = os.listdir(file_path)
   
5.     for eachfile in allfiles:
   
6.         if os.path.splitext(eachfile)[1] in ext:
   
7.           f = open('videolist.txt','w')
   
8.           f.write(str(os.path.dirname(eachfile) + '\\' +str(os.path.basename(eachfile))))
   
9.           f.close()
   
10.         if os.path.isfile(eachfile):
    
11.           search_video(eachfile)
    
12.           os.chdir(os.pardir)
    
13. 
    
14. filepath = input('路径：')
    
15. search_video(filepath)
    

复制代码

不过感觉没什么区别啊

txxcat

先说几个明显得错误：

1. import os
   
2. ext = ['.mp4','.rmvb','.avi']
   
3. def search_video(file_path):
   
4.          allfiles = os.listdir(file_path)
   
5.          for eachfile in allfiles:
   
6.              if os.path.splitext(eachfile)[1] in ext:
   
7.                  f = open('videolist.txt','w')            #用w打开文件都会覆盖掉老文件，而且，没有绝对路径名，在每个符合条件的目录下都生成一个文件，乱！
   
8.                  f.write(str(os.path.dirname(eachfile) + '\\' +str(os.path.basename(eachfile))))
   
9.                  f.close()
   
10.              if os.path.isfile(eachfile):                #不该是os.path.isdir ???
    
11.                  search_video(eachfile)
    
12.                  os.chdir(os.pardir)                    #之前又没有chdir，现在chdir，混乱了
    
13. 
    
14. filepath = input('路径：')
    
15. search_video(filepath)

复制代码

再修正一下代码：

1. import os
   
2. ext = ['.mp4','.rmvb','.avi']
   
3. def search_video(file_path):
   
4.     allfiles = os.listdir('.')                    #已在目标目录，涉及到递归调用，不能使用目录名
   
5.     for eachfile in allfiles:
   
6.         if os.path.splitext(eachfile)[1] in ext:
   
7.             f = open(cdir+'\\'+'videolist.txt','a')                                               #要使用绝对路径名
   
8.             f.write(os.getcwd() + '\\' +str(os.path.basename(eachfile))+'\n')      #os.getcwd()才能获取完整的路径     
   
9.             f.close()
   
10.         if os.path.isdir(eachfile):
    
11.             os.chdir(eachfile)               #转到新的目标目录
    
12.             search_video(eachfile)        #递归调用
    
13.             os.chdir(os.pardir)              #回到上一级目录
    
14. 
    
15. filepath = input('路径：')
    
16. cdir=os.getcwd()           #记录工作目录
    
17. os.chdir(filepath)           #去到目标目录
    
18. search_video(filepath)
    
19. os.chdir(cdir)                #回到工作目录
    

复制代码

代码是可以跑了，不过还有些不合理的地方，比如你每发现一个视频文件，就打开一次写入再关闭，动作频繁，应该程序开始就打开，然后发现写入，程序结束时关闭才合理：

1. def search_video(file_path):
   
2.     allfiles = os.listdir('.')
   
3.     for eachfile in allfiles:
   
4.         if os.path.splitext(eachfile)[1] in ext:
   
5.             f.write(os.getcwd() + '\\' +str(os.path.basename(eachfile))+'\n')
   
6.         if os.path.isdir(eachfile):
   
7.             os.chdir(eachfile)
   
8.             search_video(eachfile)
   
9.             os.chdir(os.pardir)
   
10. 
    
11. import os
    
12. ext = ['.mp4','.rmvb','.avi']
    
13. f = open('videolist.txt','w')     #这回可以用w，保证每次写入的是新数据
    
14. filepath = input('路径：')
    
15. cdir=os.getcwd()
    
16. os.chdir(filepath)
    
17. search_video(filepath)
    
18. os.chdir(cdir)
    
19. f.close()

复制代码

zhongyuanadler

txxcat 发表于 2020-4-14 17:37
先说几个明显得错误：

感谢