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

我们经常想要统计项目的代码行数，但是如果想统计功能比较完善可能就不是那么简单了， 今天我们来看一下如何用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]#