Python代码行统计

用py简单的实现了一个统计代码行的小工具

#/usr/bin/python
import os

#count the line of a single file
def CountLine(path):
        tempfile = open(path)
        res = 0
        for lines in tempfile:
                res += 1
        print "%s %d" %(path, res) #output the file path and lines
		tempfile.close()
        return res

#count the total line of a folder, sub folder included
def TotalLine(path):
        total = 0
        for root, dirs, files in os.walk(path):
                for item in files:
                        ext = item.split('.')
                        ext = ext[-1]  #get the postfix of the file
                        if(ext in ["cpp", "c", "h", "java", "py", "xml", "properties", "php"]):
                                subpath = root + "/" + item
                                total += CountLine(subpath)
        return total

print "Input Path"
path = "D:\drug\src"
print TotalLine(path)

工具简单到，统计了源码中所有的行。包括空白行和注释行。

上面使用了os.walk来遍历目录。os.walk会直接列出一个目录下的所有文件，包括子目录的。

下面是用了os.path.listdir来实现。listdir仅仅会列出当前目录下的所有目录和文件，不会深入到子目录。

'''
Created on 2013-3-11

@author: naughty
'''
import os

#===============================================================================
# count the line of a single file
#===============================================================================
def CountLine(path):
    tempfile = open(path)
    res = 0
    for lines in tempfile:
        res += 1
    print "%s\t\t\t\t\t%d" % (path, res)
    tempfile.close()
    return res

#===============================================================================
# #count the total line of a folder, sub folder included
#===============================================================================
def TotalLine(path):
    total = 0
    for mfile in os.listdir(path):
        fullpath = path + "/" + mfile
        if os.path.isdir(fullpath):
            total += TotalLine(fullpath)
        elif os.path.isfile(fullpath):
            ext = mfile.split(".")
            ext = ext[-1] 
            if ext in ['c', 'cpp', 'java', 'py']:
                total += CountLine(fullpath)   
    return total

print "Input Path"
path = "D:/drug/src"
print TotalLine(path)