Python实现代码行数统计工具

当前位置:

首页 > Python基础教程 >

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)

栏目列表