python实现目录递归遍历

#实现目录递归遍历，查找.log文件，并将结果存入文件

import os
res=[]
def findfile(dir):
    for i in os.listdir(dir):
        if os.path.isdir(dir+os.sep+i)==True:
            findfile(dir+os.sep+i)
        else:
            if i[-4:]=='.log':
                res.append(dir+os.sep+i)
findfile(os.path.abspath('.'))
#findfile('D:\Python\code')
#print len(res)

fp=open('result.log','w')
for i in res:
    fp.write(i+'\n')
fp.close()