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