【每日一坑 6】 查找文件内容

fl0w

findThemAll.py

1. #! /usr/bin/env python
   
2. # coding:utf-8
   
3. 
   
4. import sys
   
5. import os
   
6. import fnmatch
   
7. import re
   
8. 
   
9. def filterFilename(path, text):
   
10.     fileList = []
    
11.     for dirpath, dirnames, filenames in os.walk(path):
    
12.         for filename in filenames:
    
13.             if fnmatch.fnmatch(filename, text):
    
14.                 fileList.append(os.path.join(dirpath, filename))
    
15.     return fileList
    
16. 
    
17. if __name__ == "__main__":
    
18.     if len(sys.argv) < 3:
    
19.         print 'Please input like this:./findThemAll.py your_dir your_match_string'
    
20.     elif os.path.exists(sys.argv[1]):
    
21.         fileList = filterFilename(sys.argv[1],'*.txt')
    
22.         themAll = []
    
23.         for file in fileList:
    
24.             with open(file) as f:
    
25.                 content = f.read()
    
26.                 if len(re.findall(sys.argv[2], content)) > 0:
    
27.                     themAll.append(file)
    
28.         print 'The files under %s contain "%s" are:' % (sys.argv[1], sys.argv[2])
    
29.         print themAll
    
30.     else:
    
31.         print 'Are you sure the path is exist?'
    

复制代码