用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)

部分结果如下：

两万行被我说成了两千行，哭晕在厕所