简单的递归问题

第一浩男

一个查找给定路径下的视频格式的程序，死循环了，大神指教一下问题出在了哪里！

1. import os
   
2. 
   
3. def func(open_file):
   
4.     first_file=os.getcwd()
   
5. 
   
6.     os.chdir(open_file)
   
7.    
   
8.     ext_set=set()
   
9.    
   
10.     for each_file in os.listdir():
    
11.         if  os.path.isdir(each_file):
    
12.             func(each_file)
    
13.             os.chdir(os.pardir)
    
14. 
    
15.         else:
    
16.             ext=os.path.splitext(each_file)
    
17.             if ext[1] in ('avi','.rmvb','.mp4'):
    
18.                 video_list.append(os.getcwd()+each_file)
    
19.                 print(video_list)
    
20.                
    
21.                
    
22. 
    
23. 
    
24. open_file=input('请输入要查找的文件目录:')
    
25. video_list=[]
    
26. func(open_file)
    
27. txt=open_file+'videolist.txt'
    
28. txt=open(txt,'w')
    
29. txt.writelines(video_list)

复制代码

jackz007

    这一行有问题：

1.     for each_file in os.listdir():

复制代码

    应该改成这样：

1.     for each_file in os.listdir(os.getcwd()):

复制代码

    我把楼主的代码优化了一下，可以自动跨过没有访问权限的目录，使用不受限制，楼主可以测试一下。

1. import os
   
2. 
   
3. def func(open_file , type_list):
   
4.     try:
   
5.         for each_file in os . listdir(open_file):
   
6.             x = os . path . join(open_file , each_file)
   
7.             if os . path . isfile(x) :
   
8.                 ext = os . path . splitext(each_file)[1] . lower()
   
9.                 if ext in type_list :
   
10.                     video_list . append(x)
    
11.             elif os . path . isdir(x) :
    
12.                 func(x , type_list)
    
13.     except:
    
14.         pass
    
15. 
    
16. open_file = input('请输入要查找的文件目录:')
    
17. video_list = []
    
18. func(open_file , ('.avi' , '.rmvb' , '.mp4' , '.mkv' , '.flv' , '.wmv'))
    
19. txt = os . path . join(open_file , 'videolist.txt')
    
20. with open(txt , 'w') as f:
    
21.     for x in video_list :
    
22.         print(x)
    
23.         f . write(x + '\n')

复制代码

第一浩男

jackz007 发表于 2019-5-1 13:35
这一行有问题：

    应该改成这样：

哈哈哈  你这个还有许多语句我还没学  看不懂     但是我那个在你给出改正后还是循环个没完