用python3統計代碼行數

今天接到一個電話面試，對方問我在一個項目中維護了多少行代碼。

我懵逼了，從來沒有統計過啊，怎么還會有這種需求？

當時一臉茫然的想了想，回答了一個，呃...差不多兩千多行吧...感覺很心虛

掛完電話之后大概看了一下最近的一個項目，光其中某一個頂層文件就一千多行了好吧，感覺自己回答的好low

但是又不能自己挨個去統計每個文件中到底有多少行代碼啊，還要去掉注釋和空格，一個文件一個文件的去計算，這顯然不是程序員的風格

所以又不務正業了一下，拿python寫了個小程序，用來統計某個目錄下一個有多少.v文件，有多少行代碼。

當然，子目錄也是可以統計的。

用法：把以下代碼復制到新建的py文件中，將該文件放到你想統計的工程的根目錄，運行一下，統計結果會生成到該PY文件的同一級目錄，文件名為"code_line_count.txt"

代碼如下：

# -*- coding: utf-8 -*-
import sys
import os
import codecs

exts = ['.v','.vhd']
def read_line_count(fname):
    count = 0
    with open('code_line_count.txt','a') as f:
        f.write('fname:%s\n' % fname)
    with open(fname,'r',encoding='utf8') as f:
        for file_line in f.readlines():
            file_line = file_line.strip() 
            if not len(file_line) or file_line.startswith('//'):
                continue   
            count += 1
    with open('code_line_count.txt','a') as f:
        f.write('line count::%s\n' % count)
    return count

if __name__ == '__main__':
    with open('code_line_count.txt','w') as f:
        f.write('\n')
    count = 0
    fcount = 0
    for root,dirs,files in os.walk(os.getcwd()):
        for f in files:
            # Check the sub directorys
            print(f)
            fname = (root + '\\'+ f).lower()
            if os.path.splitext(f)[1]:
                ext = f[f.rindex('.'):]
                try:
                    if(exts.index(ext) >= 0):
                        fcount += 1
                        c = read_line_count(fname)
                        count += c
                        with open('code_line_count.txt','a') as f:
                            f.write('total count:%s\n' % count)
                except:
                    pass

    with open('code_line_count.txt','a') as f:
        f.write('\n')
        f.write('--------------------------------------\n')
        f.write('total file count:%d\n' % fcount)
        f.write('total line count:%d\n' % count)

 

部分結果如下：

 兩萬行被我說成了兩千行，哭暈在廁所