《Python Case系列》一.Python实现统计项目的代码行数

0.背景

前一段时间看Mybatis源码时，突然想看看Mybatis的源码到底有多少号（不算空行）。然后就想用Python写一个。开始的版本是参考网上，现在忘了地址了。。。大概思路是：

1.使用os.walk函数来遍历一个路径下的所有文件，将文件的绝对路径存放在一个list中

2.遍历文件list，统计每一文件去除空行后的行数。

3.输出统计的总行数。

1.完善

第一版的要统计的项目路径时写死在程序中的，文件类型也是。后来我继续完善写成了第二版。在运行Python脚本时，可以将路径和文件类型作为参数，文件类型可以写多个。以下是运行截图：



-p之后跟的是路径

-t之后跟的文件的后缀名，多个文件后缀名用空格隔开

2. 源代码

#! /usr/bin/python
# encoding=utf-8

import os
import optparse
import time

file_list = []
type_list = []
def get_file(path):

if os.path.isdir(path):
for parent, dirNames, fileNames in os.walk(path,topdown=True):
for fileName in fileNames:
sufix = fileName.split('.')[-1]
if sufix in type_list:
file_list.append(os.path.join(parent, fileName))
elif os.path.isfile(path):
file_list.append(path)

def count_line(file):
count = 0
for file_line in open(file).xreadlines():
if file_line != '' and file_line != '\n':
count += 1

print file, count
return count

def main():
parser = optparse.OptionParser("-p project_path -t file_type1 file_type2...")
parser.add_option('-p', '--path', dest='path', type='string', help='project path')
parser.add_option('-t', '--type', dest='type', type='string', help='file type')
(options, args) = parser.parse_args()
path = options.path
file_type = options.type
args.append(file_type)
if (path == None) | (file_type == None):
print('[-] You must specify a project path and serveral file types.')
return

for fileType in args:
type_list.append(fileType)

start_time = time.clock()

total_count = 0
get_file(path)
for file in file_list:
total_count += count_line(file)

print('=== total line count:%s ===' % total_count)
print('time:%0.2f second.' % (time.clock() - start_time))

if __name__ == '__main__':
main()

更多Python case 请参考我的github：https://github.com/Smart-Ha/pycase