首页 > 代码库 > Python实现代码行数统计工具

Python实现代码行数统计工具

我们经常想要统计项目的代码行数,但是如果想统计功能比较完善可能就不是那么简单了, 今天我们来看一下如何用python来实现一个代码行统计工具。

思路:首先获取所有文件,然后统计每个文件中代码的行数,最后将行数相加.

实现的功能:

统计每个文件的行数;

统计总行数;

统计运行时间;

支持指定统计文件类型,排除不想统计的文件类型;

递归统计文件夹下包括子文件件下的文件的行数;

排除空行;

 

# coding=utf-8import osimport timebasedir = ‘/root/script‘filelists = []# 指定想要统计的文件类型whitelist = [‘php‘, ‘py‘]#遍历文件, 递归遍历文件夹中的所有def getFile(basedir):    global filelists    for parent,dirnames,filenames in os.walk(basedir):        #for dirname in dirnames:        #    getFile(os.path.join(parent,dirname)) #递归        for filename in filenames:            ext = filename.split(‘.‘)[-1]            #只统计指定的文件类型,略过一些log和cache文件            if ext in whitelist:                filelists.append(os.path.join(parent,filename))#统计一个文件的行数def countLine(fname):    count = 0    for file_line in open(fname).xreadlines():        if file_line != ‘‘ and file_line != ‘\n‘: #过滤掉空行            count += 1    print fname + ‘----‘ , count    return countif __name__ == ‘__main__‘ :    startTime = time.clock()    getFile(basedir)    totalline = 0    for filelist in filelists:        totalline = totalline + countLine(filelist)    print ‘total lines:‘,totalline    print ‘Done! Cost Time: %0.2f second‘ % (time.clock() - startTime)

 

结果:

[root@pythontab script]# python countCodeLine.py /root/script/test/gametest.php---- 16/root/script/smtp.php---- 284/root/script/gametest.php---- 16/root/script/countCodeLine.py---- 33/root/script/sendmail.php---- 17/root/script/test/gametest.php---- 16total lines: 382Done! Cost Time: 0.00 second[root@pythontab script]#

 

只会统计php和python文件,非常方便。

 

Python实现代码行数统计工具