D.H.Q的烂笔头

Just code for fun~

python统计特定目录下指定类型文件总行数

闲时无聊,想看看项目后台代码现有的总行数,也趁好想拿python练练手:^)。

程序很简单,主要是先用“os.walk(dir)”获取特定目录下所有文件,再判断文件类型,最后用“enumerate”统计文件行数,以此循环累加出总行数。不多说,上代码:

import os

#要统计的指定目录
dir = "E:/test"
#要统计的文件类型
file_types = ('.erl', '.hrl')

def linecount(filepath):
    count = -1
    for count, line in enumerate(open(filepath, 'rU')):
        pass
    count += 1
    return count

def scan_dir(dir):
    num = 0
    for roots, dirs, files in os.walk(dir):
        for file in files:
            name, type = os.path.splitext(file)
            if type in file_types:
                filepath = os.path.join(roots, file)
                num += linecount(filepath)
    print num

if __name__ == '__main__':
    scan_dir(dir)

最后统计出有37多万行的 erlang 代码,相对于某国白富美90天写代码40多万行,我们几个后台屌丝一年多时间还没写够40万行代码,情何以堪啊:(

comments powered by Disqus
返回顶部