#!/usr/bin/env python
# -- coding: utf-8 --
# @Time : 2018/3/5 13:55
# @Author : likewind
# @mail : likewind1993@163.com
# @File : countLines.py
# @Software: sky-studio.cn
import os
'''
返回每個文件行數，其中行數不包括以“#”開頭的包含文件，宏定義等，
排除了c,cpp文件中的“//”, “/*...*/”等的注釋，
排除了python文件中import, from 等開頭的導入
'''

def get_lines(file_name):
 f = open(file_name)
 #flag用于處理c,cpp中“/*...*/”多行注釋
 flag = False
 count = 0
 while True:
  #讀取文件并去除開頭的空格，制表符
  line = f.readline()
  line = line.lstrip(' \t')
  if not line:
   break
  #如果該行有“#”， “import”等打頭的字符，忽略該行
  if flag == False:
   if line[0:1] == "#" or line[0:6] == "import" or line[0:4] == "from" or line == "\n" or line[0:2] == "http://":
    continue
  #如果該行存在“/*”并且不存在“*/”,表明多行注釋未在一行結束，flag=True
  if line.find("/*") != -1 :
   if line.find("*/") != -1:
    continue
   else:
    flag = True
    continue
  #如果flag=True，表明處于多行注釋中，判斷是否有“*/”結尾
  if flag == True :
   if line.find("*/") != -1:
    flag = False
    if line[-2:] != "*/":
     count = count+1
   continue
  #排除以上條件后，行數增加一
  count = count+1
 f.close()
 return count
'''
計算該文件目錄下所有符合條件的行數
'''
def count_lines(file_dir):
 #total_lines表示總行數，file_nums表示總文件數
 total_lines = 0
 file_nums = 0
 for root, dirs, files in os.walk(file_dir):
  for file in files:
   #不計算本文件的行數
   if file == "countLines.py":
    continue
   #只計算規范命名文件, 如[文件名.文件類型]
   file_type = file.split('.')
   if len(file_type) > 1 :
    #如果想計算其他類型的文件，可以在這里進行修改
    if file_type[1] not in ["py", "c", "cc", "cpp", "h"]:
     continue
   else:
    continue
   file_name = root +"\\" + file
   lines = get_lines(file_name)
   total_lines = total_lines + lines
   print file_name + " contains lines : " + repr(lines)
   file_nums = file_nums + 1
 #輸出結果
 print "------------------------------------"
 print "Total Files : " + repr(file_nums)
 print "Total lines : " + repr(total_lines)
 print "------------------------------------"

if __name__ == '__main__':
 cur_path = os.path.split(os.path.realpath(__file__))[0]
 count_lines(cur_path)
 raw_input("Press any key to exit...")