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

'''Author: liupengfeiFunction: count lines of code in a folder iterativelyShell-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 sysimport osimport codecsfrom _pyio import opentotalCount = 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, whichis 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();