python实现查找文件夹中文件

# -*- coding: utf-8 -*-
import os

#查找路径
PATH='D:\\lcc'
#查找文件名后缀，如果为''则查找所有文件
SUFFIX = '.c'       #查找.c文件
#查找的字符串
STRING = '06'
#存放文件名的链表
files = []
result = []

def getFiles(path):

    #如果路径不是以\\结尾则补上
    if path[-1:] != '\\':
        path += '\\'

    dirFiles = os.listdir(path)
    for file in dirFiles:
        filePath = path + file
        if os.path.isdir(filePath):
            getFiles(filePath + '\\')
        else:
            if SUFFIX == '' or file[-2:] == SUFFIX:
                files.append(filePath)

def findStr(files, str):
    for file in files:
        fp = open(file, 'r')
        lines = fp.readlines()
        cnt = 0
        for line in lines:
            cnt+=1
            if line.find(str) > -1:
                #print file, ':', cnt , line
                tmp = (file, cnt, line)
                result.append(tmp)
        fp.close()

if __name__ == '__main__':
    getFiles(PATH)
    #print 'files: ',files
    findStr(files, STRING)
    print result