文件系统

小白码农超

import os
def search_file(path, target):
    os.chdir(path)
    for each_file in os.listdir(os.curdir):
        ext = os.path.splitext(each_file)[1]
        if ext in target:
            video_list.append(os.getcwd() + os.sep + each_file + os.linesep)
        if os.path.isdir(each_file):
            search_file(each_file, target)
            os.chdir(os.pardir)
path = input('请输入需要检索的路径：')
program_dir = os.getcwd()
video_list = []
target = ['avi', 'rmvb', 'jpg', 'png']
search_file(path, target)
f = open(program_dir + os.sep + 'vediolist1.txt', 'w')
f.writelines(video_list)
f.close()

我找了好多遍真的找不到bug,怎么建立的txtl里没东西。求大佬指点！！！

jackz007

    试试这个代码：

1. import os
   
2. def search_file(path , target):
   
3.     try:                                                   # 错误捕获
   
4.         for each_file in os . listdir(path):               # 风险语句，当路径 path 无权限访问时，执行该语句会发生错误
   
5.             x = os . path . join(path , each_file)         # 如果风险语句执行顺利，就从这里继续，否则就跳去 except: 块继续执行
   
6.             if os . path . isfile(x):
   
7.                 ext = os . path . splitext(each_file)[1]
   
8.                 if ext . lower() in target:
   
9.                     video_list . append(x)
   
10.             elif os . path . isdir(x):
    
11.                 search_file(x , target)
    
12.     except:                                                # 错误捕获，如果风险语句出现错误就会立即跳到这里继续执行
    
13.         pass                                               # 什么都不做
    
14. 
    
15. video_list = []
    
16. target = ('.avi', '.rmvb', '.jpg', '.png')
    
17. path = input('请输入需要检索的路径：')
    
18. search_file(path , target)
    
19. with open('vediolist1 . txt' , 'wt') as f:
    
20.     for x in video_list:
    
21.         f . write(x + '\n')
    
22.         print(x)
    

复制代码