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

creek

1. #!/usr/bin/env python
   
2. #coding:UTF-8
   
3. 
   
4. import os
   
5. import re
   
6. def VisitDir(path):
   
7.   for root,dirs,files in os.walk(path):
   
8.       for f in files:
   
9.           if os.path.splitext(f)[1]=='.py':
   
10.               content=open(os.path.join(root,f),'r').read()
    
11.               if re.search('import os',content):
    
12.                 print f
    
13.          
    
14. if __name__=="__main__":
    
15.     path="."
    
16.     VisitDir(path)
    

复制代码

crossin先生

scz_000 发表于 2013-12-27 14:27
这题是不是可以拓展成为比较实用的搜索相关文本功能？不同的格式直接用open是不是默认用记事本打开，里面的 ...

open只是在python打开，把内容读入内存，不会调用任何其他程序打开。但这个过程是很慢的。所以虽然可以实现文本搜索，也还是不能用来全盘搜索。

scz_000

这题是不是可以拓展成为比较实用的搜索相关文本功能？不同的格式直接用open是不是默认用记事本打开，里面的那些编码有什么价值啊？小的文件夹还行，尝试的搜整个盘系统就会变得非常慢，用时似乎还要很久。

1. # coding=gbk
   
2. import os,fnmatch
   
3. def find_txt(root,patterns='*'):#定义主函数，接受目录与所要匹配的拓展名为参数
   
4.     list_of_file=[]#列表储存匹配文件路径
   
5.     for dirname,subdir,filenames in os.walk(root):#使用os。walk（遍历包括子目录下所有文件），生成三组列表，第一组为当前目录地址，第二组为当下前目录下的文件夹名（没有则返回一个空列表），第三组为当前目录的所有文件
   
6.         for i in filenames:#遍历当前目录下的所有文件，每个文件名加上当前文件夹路径，构成完整的文件路径
   
7.             file_object='\\'.join([dirname,i])#构造完整目录
   
8.             try:
   
9.                 f=open(file_object,'r')#打开文件
   
10.                 file_content= f.read()#读取文件
    
11.                 if key_words in file_content:#判断关键词是否在文件内容内
    
12.                     list_of_file.append(file_object)
    
13.             finally:
    
14.                 f.close()
    
15.     return list_of_file
    
16. if __name__=='__main__':
    
17.     root=raw_input('所要搜索目录:')
    
18.     key_words=raw_input('内容关键词:')
    
19.     match_files=find_txt(root)
    
20.     if match_files:
    
21.         print "以下是包含关键词%s的文件"%key_words
    
22.         for i in match_files:
    
23.             print i
    
24.     else:
    
25.         print"没有找到相关文件"

复制代码

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?'
    

复制代码