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

0.背景

前一段时间看Mybatis源码时，突然想看看Mybatis的源码到底有多少号（不算空行）。然后就想用Python写一个。开始的版本是参考网上，现在忘了地址了。。。大概思路是：
1.使用os.walk函数来遍历一个路径下的所有文件，将文件的绝对路径存放在一个list中

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

3.输出统计的总行数。

1.完善

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

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

2. 源代码

#! /usr/bin/python# encoding=utf-8import osimport optparseimport timefile_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 countdef 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