countlines.py

#!/usr/bin/env python

'countlines.py -- count the lines of files end by ".*" in a folder.'

import os

def countLines(filename):
    count = 0
    f = open(filename, 'r')
    for eachline in f:
        count += 1
    f.close()
    return count

def getFilePathsDict(path):
    pyDict = {}
    filesList = os.listdir(path)
    for each in filesList:
        filepath = os.path.join(path, each)
        if os.path.isfile(filepath) and (each.endswith('.h') or each.endswith('.cpp') or each.endswith('.cu')):  # change this line to choose the file kinds you want to count
            pyDict[filepath] = 0
        elif os.path.isdir(filepath):
            pyDict = dict(pyDict, **getFilePathsDict(filepath))   # recursive when it is a folder
    return pyDict

def getTotalNumber(pyDict):
    return sum(pyDict.values())

def showEachFileNumber(pyDict):
    print 'File Name\t\tLine Number'
    print '-' * 40
    for key in sorted(pyDict):
        print '%s\t%s' % (os.path.basename(key).ljust(20), pyDict[key])

def main():
    path = raw_input('Please enter the folder path: ').strip()
    pyDict = getFilePathsDict(path)
    for key in pyDict:
        pyDict[key] = countLines(key)
    sum = getTotalNumber(pyDict)
    print 'The total line number is %d.' % sum
    print 'Details:'
    showEachFileNumber(pyDict)

if __name__ == '__main__':
    main()