统计代码行数

from os import path
from sys import argv
from os import listdir

totalLine=0

def countFile(fileName):
    fd=open(fileName)
    lines=fd.readlines()
    return len(lines)

def isSource(fileName):
    sourceList=('py', 'c', 'cpp', 'h')
    tmp=fileName.split('.')
    tailName=tmp[-1]
    if tailName in sourceList:
        return True
    else:
        return False

def countDir(cur_path):
    file_list=listdir(cur_path)
    global totalLine
    for file_iter in file_list:
        file_name=cur_path+'/'+file_iter
        if path.isdir(file_name) :
            countDir(file_name)
        elif(isSource(file_name)):
            totalLine+=countFile(file_name)
        else:
            pass

def main():
    cur_path=argv[1]
    countDir(cur_path)
    print 'The project %s has %d lines' % (cur_path, totalLine)

main()