python实现统计代码行数的方法

本文实例讲述了python实现统计代码行数的方法。分享给大家供大家参考。具体实现方法如下：

    '''
    Author: liupengfei
    Function: count lines of code in a folder iteratively
    Shell-format: cmd [dir]
    Attention: default file encode is utf8 and default file type is java-source-file. But users can customize this script by just modifing global variables.
    '''
    import sys
    import os
    import codecs
    from _pyio import open
    totalCount = 0;
    fileType = '.java'
    descLineBegin = '//'
    descBlockBegin = r'/**'
    descBlockEnd = r'*/'
    fileEncode = 'utf-8'
    def main():
      DIR = os.getcwd()
      if len(sys.argv) >= 2:
        DIR = sys.argv[1]
      if os.path.exists(DIR) and os.path.isdir(DIR):
        print('target directory is %s' % DIR)
        countDir(DIR)
        print('total code line is %d' % totalCount)
      else:
        print('target should be a directory!')
    def isFileType(file):
      return len(fileType) + file.find(fileType) == len(file)
    def countDir(DIR):
      for file in os.listdir(DIR):
        absPath = DIR + os.path.sep + file;
        if os.path.exists(absPath):
          if os.path.isdir(absPath):
            countDir(absPath)
          elif isFileType(absPath):
            try:
              countFile(absPath)
            except UnicodeDecodeError:
              print(
                '''encode of %s is different, which
    is not supported in this version!'''
                )
    def countFile(file):
      global totalCount
      localCount = 0
      isInBlockNow = False
      f = codecs.open(file, 'r', fileEncode);
      for line in f:
        if (not isInBlockNow) and line.find(descLineBegin) == 0:
          pass;
        elif (not isInBlockNow) and line.find(descBlockBegin) >= 0:
          if line.find(descBlockBegin) > 0:
            localCount += 1
          isInBlockNow = True;
        elif isInBlockNow and line.find(descBlockEnd) >= 0:
          if line.find(descBlockEnd) + len(descBlockEnd) < len(line):
            localCount += 1
          isInBlockNow = False;
        elif (not isInBlockNow) and len(line.replace('\\s+', '')) > 0:
          localCount += 1
      f.close()
      totalCount += localCount
      print('%s : %d' % (file, localCount))
    if __name__ == '__main__':
      main();

希望本文所述对大家的Python程序设计有所帮助。