import os

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

#將全部已搜索到的關(guān)鍵字列表中的內(nèi)容保存到result.log文件中
def writeResultLog(allExistsKeywords):
 #行分隔符
 ls = os.linesep
 #結(jié)果日志文件名
 logfilename = "result.log" #相對(duì)路徑,文件在.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() 


#搜索指定關(guān)鍵字是否在指定的文件中存在
def searchFilesContent(dirname):
 #從searchkeywords.txt文件中初始化待搜索關(guān)鍵字列表
 filename = "searchkeywords.txt" #相對(duì)路徑,文件在.py文件所在的目錄中
 #待搜索關(guān)鍵字列表
 allSearchKeywords=[]
 #遍歷文件當(dāng)前行已搜索到的關(guān)鍵字列表
 existsKeywordsThisLine=[]
 #全部已搜索到的關(guān)鍵字列表
 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文件中初始化要排除的搜索關(guān)鍵字列表
 filename = "excludekeywords.txt" #相對(duì)路徑,文件在.py文件所在的目錄中
 #要排除的搜索關(guān)鍵字列表
 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();

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


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

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

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

      #清空該行已搜索到的關(guān)鍵字列表內(nèi)容
      existsKeywordsThisLine = []

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

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

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