python简单习题系列6

写一个脚本统计代码的行数。

由于手工统计代码行数费时费力，不如来一个简单的python 脚本来的快速便捷。顺便练习一下刚刚学习的python语言。

写的只是简单的功能，还是有很多地方需要改进和加强对错误的处理之类。

#codeing=UTF-8
# 根据输入的目录，统计该目录下代码行数

import os
from os import listdir
from os.path import isdir, isfile


def CountAFileLines(fileName):
    file = open(fileName, 'r')
    allLines = file.readlines();
    i = 0
    for li in allLines:
        if li != '':
            i = i+1        
    file.close()
    return i

def CodeCount(in_path):
    # First, check the input path if is valid directory
    try:
        assert(isdir(in_path) == True)
    except AssertionError:
        in_path = None
        print ('The input path is not a valid directory!\n')
        print ('Please input another right path:')

    # list all files or dirs in input directory
    # if is a dir, call back
    # else if is a cpp/h/cs file, count the liens
    line_number = 0
    if(in_path):
        for one in listdir(in_path):
            one = "/".join([in_path, one])
            curNum = 0
            
            if (isdir(one)):                
                curNum = CodeCount(one)
            elif (isfile(one)):
                (filename, extention) = os.path.splitext(one)
                if(extention == '.h' or extention == '.cpp' or extention == '.cs'):
                    #print all calculate files
                    print (one)
                    curNum = CountAFileLines(one)
            line_number = line_number + curNum
    return line_number



inputPathStr = input('Please input the code path you want count: \n')
print ('\n')
print ('List all calculate files: ')
line_number = CodeCount(inputPathStr)
print ('\nAll codes line number is: ', line_number)