代码行数统计工具

为了便于统计自己代码的行数，用python实现了该小工具，可以支持对单个文件和一个目录下的多个文件进行统计，并输出统计的代码行数，空行数和注释行数。对不同语言的代码，只需要修改其注释符就可以正确的统计行数信息。

import os, re, sys

class CodeCount:
    """
    this py is used to calculate the total number of source,
    comment and blank lines.
    """
    def __init__(self, path, commentsign='#'):
        self.path = path
        self.all = 0
        self.allComment = 0
        self.allBlank = 0
        self.allSource = 0
        self.commentsign = commentsign

    def Calculate(self):
        if not os.path.isdir(self.path):
            if os.path.exists(self.path):
                self.CountLines(self.path)
            else:
                print '%s is not a valid path or file name!' % path
                return
        for root, dirs, files in os.walk(self.path):
            for file in files:
                file = os.path.join(root, file)
                self.CountLines(file)

    def CountLines(self, filename):
        f = file(filename)
        lines = f.readlines()
        rBlank = re.compile(r'^s*$')
        rComment = re.compile('^s*%sw*' % self.commentsign)
        nBlank = 0
        nComment = 0
        nSource = 0
        for line in lines:
            if rBlank.match(line) or not line:
                nBlank += 1
            elif rComment.match(line):
                nComment += 1
            else:
                nSource += 1

        print filename;
        print '           Total      :',nBlank+nComment+nSource
        print '           Comment    :',nComment
        print '           Blank      :',nBlank
        print '           Source     :',nSource
        self.all += (nBlank+nComment+nSource)
        self.allBlank += nBlank
        self.allComment += nComment
        self.allSource += nSource
        

    def GetCommentCount(self):
        return self.allComment

    def GetBlankCount(self):
        return self.allBlank

    def GetSourceCount(self):
        return self.allSource

    def GetAllCount(self):
        return self.allBlank+self.allComment+self.allSource

def main( ):
    if len(sys.argv) == 2:
        return
    #path = raw_input('input a path or filename:')
    path = sys.argv[1]
    cCut = CodeCount(path)
    cCut.Calculate()
    print path, ':'
    print '           Total      :',cCut.GetAllCount()
    print '           Comment    :',cCut.GetCommentCount()
    print '           Blank      :',cCut.GetBlankCount()
    print '           Source     :',cCut.GetSourceCount()

if __name__ == '__main__':
    main()