【每日一坑 4】 查找文件

fl0w

HAO 发表于 2013-12-19 21:33

这样不能找出嵌套的子文件夹中的以.txt结尾的文件

HAO

1. # -*- coding: utf-8 -*-
   
2. 
   
3. import os
   
4. from sys import argv
   
5. 
   
6. filename, path = argv
   
7. 
   
8. txt_files = []
   
9. 
   
10. if os.path.isdir(path):
    
11.     for dir_path, dir_names, file_names in os.walk(path):
    
12.         for file_name in file_names:
    
13.             if file_name.endswith('.txt'):
    
14.                 txt_files.append(file_name)
    
15. 
    
16. 
    
17. print txt_files

复制代码

fl0w

findThem.py

1. #! /usr/bin/env python
   
2. # coding:utf-8
   
3. 
   
4. import os
   
5. import sys
   
6. 
   
7. if __name__ == "__main__":
   
8.     if len(sys.argv) != 1 and os.path.exists(sys.argv[1]):
   
9.         themAll = []
   
10.         for dirpath, dirnames, filenames in os.walk(sys.argv[1]):
    
11.             for filename in filenames:              
    
12.                 if os.path.splitext(filename)[1] == '.txt':
    
13.                     themAll.append(os.path.join(dirpath, filename))
    
14.         print themAll
    
15.     else:
    
16.         print 'Please input like this:./findThem.py /home/fl0w'

复制代码

我回来啦

1. # coding:utf-8
   
2. import os
   
3. import fnmatch
   
4. 
   
5. for root, dirs, files in os.walk('.'):
   
6.     for file in files:
   
7.         if fnmatch.fnmatch(file, "*.txt"):
   
8.             print file
   

复制代码