我們經常想要統計專案的程式碼行數，但是如果想統計功能比較完善可能就不是那麼簡單了， 今天我們來看一下如何用python來實現一個程式碼行統計工具。
思路：首先獲取所有檔案，然後統計每個檔案中程式碼的行數，最後將行數相加.
實現的功能：
統計每個檔案的行數；
統計總行數；
統計執行時間；
支援指定統計檔案型別，排除不想統計的檔案型別；
遞迴統計資料夾下包括子檔案件下的檔案的行數；
排除空行；
# coding=utf-8
import os
import time
basedir = '/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 count
if __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---- 16 total lines: 382 Done! Cost Time: 0.00 second [root@pythontab script]#
只會統計php和python檔案，非常方便。
其實大家還可以在此基礎上進行改進，比如：排除註釋行等等。
如有問題歡迎留言提問！