用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)