python 文档搜索_python 文本搜索

# coding: UTF-8

import os,sys,codecs

from stat import *

root = 'D:/hjhtemp/cdt8' #path

search = 'buildConsoleLines' #keyword

search_file_type = '.java' #file type

def worktree(top,callback):

for f in os.listdir(top):

pathname = os.path.join(top,f)

mode = os.stat(pathname)[ST_MODE]

if S_ISDIR(mode):

worktree(pathname,callback)

elif S_ISREG(mode):

if f.endswith(search_file_type):

callback(pathname)

else:

print('Skipping %s' % pathname)

def visitFile(filepath):

file = codecs.open(filepath,'r','utf-8','replace')

find = 0

lineNum=0

for line in file:

lineNum=lineNum+1

if search in line:

print(line)

find=1

break

if find == 1:

print(filepath)

if __name__ == '__main__':

if len(sys.argv) == 2:

search = sys.argv[1]

worktree(root,visitFile)