目标:统计指定目录下任意编程语言的代码行数。

代码:
# coding=utf-8import osimport time# 设定根目录basedir = './'     #这种写法示意:与本文件在同一目录下的所有文件及文件夹中的多层文件filelists = []# 指定想要统计的文件类型# whitelist = ['cpp', 'h']whitelist = ['java', 'xml', 'yml', 'jsp', 'html']#遍历文件, 递归遍历文件夹中的所有def getFile(basedir):    global filelists    for parent,dirnames,filenames in os.walk(basedir):        for filename in filenames:            ext = filename.split('.')[-1]            #只统计指定的文件类型,略过一些log和cache文件            if ext in whitelist:                filelists.append(os.path.join(parent,filename))#统计一个的行数def countLine(fname):    count = 0    # 把文件做二进制对待,read.    for file_line in open(fname, 'rb').readlines():        if file_line != '' and file_line != '\n': #过滤掉空行            count += 1    print (fname + '----' , count)    return countif __name__ == '__main__' :    startTime = time.clock()    getFile(basedir)    totalline = 0    for filelist in filelists:        totalline = totalline + countLine(filelist)    print ('total lines:',totalline)    print ('Done! Cost Time: %0.2f second' % (time.clock() - startTime))

 

运行后果:

c代码测试

Java代码测试

 


参考:实验楼