行数统计

用python实现简单的行数统计功能(暂未排除注释行，仅计算除却空行以外的行数)

步骤

1. 遍历目录下文件名，获取后缀，指定要进行统计的文件类型
2. 对文件内容进行读取，排除空行内容，除空行外，读取到内容则计数加1

代码：

# coding=utf-8
import os
import time


basedir = os.path.abspath('test')
filelists = []
# 指定想要统计的文件类型
whitelist = ['txt', '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]
            print(ext)
            #只统计指定的文件类型，略过一些log和cache文件
            if ext in whitelist:
                filelists.append(os.path.join(parent,filename))


#统计一个文件的行数
def countLine(fname):
    count = 0
    for file_line in open(fname).readlines():
        if file_line != '' and file_line != '\n': #过滤掉空行
            count += 1
    print(fname + '----' , count)
    return count

if __name__ == '__main__' :
    startTime = time.clock()
    getFile(basedir)
    print(filelists)
    totalline = 0
    for filelist in filelists:
        totalline = totalline + countLine(filelist)
    print('total lines:',totalline)
    print('Done! Cost Time: %0.2f second' % (time.clock() - startTime))