Python 检索目录下所有文件中包含指定字符串的文件

http://blog.csdn.net/authorzhh/article/details/8933758

#!/usr/bin/env python
# 1.py
# use UTF-8
# Python 3.3.0

# 检索目录下所有文件中包含指定字符串的文件
import os
import re

# 枚举dirPath目录下的所有文件
def ListFiles(dirPath):
#begin
fileList = []
for root, dirs, files in os.walk(dirPath):# 注意os.walk的功能
#begin
for fileObj in files:
#begin
fileList.append(os.path.join(root, fileObj))
#end
#end
return fileList
#end

def FindString(filePath, regex):
#begin
fileObj = open(filePath, 'r')
for eachLine in fileObj:
#begin
if re.search(regex, eachLine, re.I):
#begin
print(fileObj)#打印文件对象
break
#end
#end
#end


def main():
#begin
fileDir = "F:" + os.sep + "aaa"# 查找F:\aaa 目录下
regex = 'include'# 包含include 的文件
fileList = ListFiles(fileDir)
print(fileList)
for fileObj in fileList:
#begin
FindString(fileObj, regex)
#end
os.system("pause")
#end

if __name__ == '__main__':
#begin
main()
#end

# 输出:
# <_io.TextIOWrapper name='F:\\aaa\\1.cpp' mode='r' encoding='cp936'>
# <_io.TextIOWrapper name='F:\\aaa\\2.h' mode='r' encoding='cp936'>
# <_io.TextIOWrapper name='F:\\aaa\\3.c' mode='r' encoding='cp936'>
# <_io.TextIOWrapper name='F:\\aaa\\4.h' mode='r' encoding='cp936'>
# <_io.TextIOWrapper name='F:\\aaa\\5.cpp' mode='r' encoding='cp936'>
# <_io.TextIOWrapper name='F:\\aaa\\6.h' mode='r' encoding='cp936'>
# <_io.TextIOWrapper name='F:\\aaa\\我.cpp' mode='r' encoding='cp936'>

# 遇到的问题有
# 如果做枚举到的文件不是文本文件, 例如rar文件, 就会有异常产生