import os

#根據文件擴展名判斷文件類型
def endWith(s,*endstring):
 array = map(s.endswith,endstring)
 if True in array:
  return True
 else:
  return False

#將全部已搜索到的關鍵字列表中的內容保存到result.log文件中
def writeResultLog(allExistsKeywords):
 #行分隔符
 ls = os.linesep
 #結果日志文件名
 logfilename = "result.log" #相對路徑,文件在.py文件所在的目錄中
 try:
  fobj = open(logfilename,'w')
 except IOError,e:
  print "*** file open error:",e
 else:
  fobj.writelines(['%s%s' % (keyword,ls) for keyword in allExistsKeywords])
  fobj.close() 


#搜索指定關鍵字是否在指定的文件中存在
def searchFilesContent(dirname):
 #從searchkeywords.txt文件中初始化待搜索關鍵字列表
 filename = "searchkeywords.txt" #相對路徑,文件在.py文件所在的目錄中
 #待搜索關鍵字列表
 allSearchKeywords=[]
 #遍歷文件當前行已搜索到的關鍵字列表
 existsKeywordsThisLine=[]
 #全部已搜索到的關鍵字列表
 allExistsKeywords=[]

 try:
  fobj = open(filename,'r');
 except IOError,e:
  print "*** file open error:",e
 else:
  for eachLine in fobj:
   allSearchKeywords.append(eachLine.strip('\n')); #使用strip函數(shù)去除每行的換行符
  fobj.close();

 #從excludekeywords.txt文件中初始化要排除的搜索關鍵字列表
 filename = "excludekeywords.txt" #相對路徑,文件在.py文件所在的目錄中
 #要排除的搜索關鍵字列表
 allExcludedKeywords=[]
 try:
  fobj = open(filename,'r');
 except IOError,e:
  print "*** file open error:",e
 else:
  for eachLine in fobj:
   allExcludedKeywords.append(eachLine.strip('\n')); #使用strip函數(shù)去除每行的換行符
  fobj.close();

 #從全部已搜索到的關鍵字列表排除掉不用搜索的關鍵字
 for excluedkw in allExcludedKeywords:
  if(excluedkw in allSearchKeywords):
   allSearchKeywords.remove(excluedkw);


 #遍歷打開所有要在其中搜索內容的文件，若待搜索關鍵字列表為空，則不再繼續(xù)遍歷
 for root,dirs,files in os.walk(dirname):
  for file in files:
   if endWith(file,'.java','.xml','.properties'): #只在擴展名為.java/.xml/.properties文件中搜索
    #打開文件
    filename = root + os.sep + file #絕對路徑
    filename = filename.replace("\\","\\\\") #將路徑中的單反斜杠替換為雙反斜杠，因為單反斜杠可能會導致將路徑中的內容進行轉義了，replace函數(shù)中"\\"表示單反斜杠，"\\\\"表示雙反斜杠
    try:
     fobj = open(filename,'r');
    except IOError,e:
     print "*** file open error:",e
    else:
     #遍歷文件的每一行
     for fileLine in fobj:
      #判斷當前行是否包含所有搜索關鍵字
      for keyword in allSearchKeywords:
       #若包含，并添加到該行已搜索到的關鍵字列表中
       if keyword.upper() in fileLine.upper(): #將搜索關鍵字和該行文本內容都轉換為大寫后再進行匹配
        existsKeywordsThisLine.append(keyword)

      #將這些搜索到的關鍵字添加到全部已搜索到的關鍵字列表中，并包含文件名信息
      for keyword in existsKeywordsThisLine:
       allExistsKeywords.append(keyword+"\t"+filename.replace("\\\\","\\"))

      #將這些搜索到的關鍵字從待搜索關鍵字列表中移除（后續(xù)將不再搜索該關鍵字）
      for keyword in existsKeywordsThisLine:
       allSearchKeywords.remove(keyword)

      #清空該行已搜索到的關鍵字列表內容
      existsKeywordsThisLine = []

      #若所有的關鍵字都搜索到了，則記錄日志文件，并結束搜索工作
      if len(allSearchKeywords)==0:
       fobj.close();
       writeResultLog(allExistsKeywords)
       print "DONE!",
       return
     fobj.close();

 #全部文件遍歷結束
 writeResultLog(allExistsKeywords)
 print "DONE!",

#僅當本python模塊直接執(zhí)行時，才執(zhí)行如下語句，若被別的python模塊引入，則不執(zhí)行
if __name__ == '__main__':
 searchFilesContent(r"G:\ccsSmartPipe\SmartPipe\src\java")