#coding：utf-8
import os,re
 
#代碼所在目錄
FILE_PATH = './'
 
def analyze_code(codefilesource):
  '''
  打開一個(gè)py文件，統(tǒng)計(jì)其中的代碼行數(shù)，包括空行和注釋
  返回含該文件總行數(shù)，注釋行數(shù)，空行數(shù)的列表
  :param codefilesource:
  :return:
  '''
  total_line = 0
  comment_line = 0
  blank_line = 0
  with open(codefilesource,encoding='gb18030',errors='ignore') as f:
    lines = f.readlines()
    total_line = len(lines)
    line_index = 0
    #遍歷每一行
    while line_index < total_line:
      line = lines[line_index]
      #檢查是否為注釋
      if line.startswith("#"):
        comment_line += 1
      elif re.match("\s*'''",line) is not None:
        comment_line += 1
        while re.match(".*'''$",line) is None:
          line = lines[line_index]
          comment_line += 1
          line_index += 1
      #檢查是否為空行
      elif line =='\n':
        blank_line += 1
      line_index += 1
  print("在%s中:"%codefilesource)
  print("代碼行數(shù)：",total_line)
  print("注釋行數(shù):",comment_line,"占%0.2f%%"%(comment_line*100/total_line))
  print("空行數(shù):", blank_line, "占%0.2f%%"%(blank_line * 100 / total_line))
  return [total_line,comment_line,blank_line]
def run(FILE_PATH):
  os.chdir(FILE_PATH)
  #遍歷py文件
  total_lines = 0
  total_comment_lines = 0
  total_blank_lines = 0
  for i in os.listdir(os.getcwd()):
    if os.path.splitext(i)[1] == '.py':
      line = analyze_code(i)
      total_lines,total_comment_lines,total_blank_lines=total_lines+line[0],total_comment_lines+line[1],total_blank_lines+line[2]
  print("總代碼行數(shù):",total_lines)
  print("總注釋行數(shù):",total_comment_lines,"占%0.2f%%"%(total_comment_lines*100/total_lines))
  print("總空行數(shù):", total_blank_lines, "占%0.2f%%"% (total_blank_lines * 100 / total_lines))
 
if __name__ == '__main__':
  run(FILE_PATH)