代码行数统计

最近工作不忙,写了个脚本统计代码行数.记录于此方便以后查阅.python版本2.7

# -*- coding: cp936 -*-
import os

totalLineCounts = 0
fileCounts = 0

def GetLines(fullFilename):
    lineNum = 0
    endCommentLine = 0
    startCommentLine = 0
    totalBlankLineNum = 0
    totalCommentLineNum = 0
    efficientCodesLineNum = 0

    InCommentArea = 0  #防止重复统计.比如/* */块内包含有//开头的行
    
    for line in open(fullFilename).readlines():
        line = line.strip()   #去除首尾的空格
        lineNum += 1

        if InCommentArea:           
            if line[-2:] == '*/':
                InCommentArea = 0   #出注释区
                endCommentLine = lineNum
                #print ('注释区:%d行-%d行' % (startCommentLine,endCommentLine))
                totalCommentLineNum += (endCommentLine - startCommentLine + 1) 
        else:                
            if line == '':          #空行
                totalBlankLineNum += 1
            elif line == '//':      #以//开头的注释
                totalCommentLineNum += 1
            elif line[0:2] == '/*':
                InCommentArea = 1   #进入注释区
                startCommentLine = lineNum
            else:
                efficientCodesLineNum += 1       
    #print ('空白行数量:%d,注释行数量:%d,有效代码行数量:%d,总行数:%d' % (totalBlankLineNum,totalCommentLineNum,efficientCodesLineNum,lineNum))
    return [totalBlankLineNum,totalCommentLineNum,efficientCodesLineNum,lineNum]


exts = ['cpp','h','txt']
#key:ext value:[totalBlankLineNum,totalCommentLineNum,efficientCodesLineNum,lineNum]
statistic = {'cpp':[0,0,0,0],'h':[0,0,0,0],'txt':[0,0,0,0]}

codesDir = 'D:\SvnCodes\main_branch\code\billing'
#codesDir = 'D:TestSub1sub1.1sub1.1.1' 
if __name__ == '__main__':
    for curRoot,curSubDirs,curFiles in os.walk(codesDir):
        #print curRoot,curSubDirs,curFiles
        for file in curFiles:
            fileExt = file.split('.')[-1]
            if fileExt in exts:
                fullFilename = curRoot + os.path.sep + file
                r = GetLines(fullFilename)
                statistic[fileExt][0] += r[0]
                statistic[fileExt][1] += r[1]
                statistic[fileExt][2] += r[2]
                statistic[fileExt][3] += r[3]
        
    print '{ext:[blank,comment,efficient,total]}'
    print statistic