python显示代码行数_python实现代码行数统计

第一步：我们应该定义一个static统计函数，要求函数能对指定路径，如果是目录能够递归访问，如果是文件就对文件进行行数计算。

import os

def static(path = '.'):

if os.path.isdir(path):

for  x in os.listdir(path):

childdir = os.path.join(path, x)

if os.path.isdir(childdir):

static(childdir)

elif os.path.isfile(childdir):

clacFilelines(childdir)

elif os.path.isfile(path):

clacFilelines(path)

return

第二步：实现文件行数计算函数功能，统计文本文件行数，过滤空白行(代码注释涉及到很多情况暂时不处理)，并将统计结果保存在以文件扩展名为key的字典result里。

def clacFilelines(filepath):

vpath = os.path.splitext(filepath)

ext = vpath[1].lower()

if not ext.strip():

return False

if not isTextFile(ext):

return False

lines = 0

with open(filepath, 'rU') as f:

filelist = f.readlines()

for line in filelist:

if not line.strip():

continue

lines += 1

if ext in result:

result[ext] += lines

else:

result[ext] = lines

print 'file:%s lines:%d' %(filepath, lines)

return True

第三步：实现文件过滤功能函数，因为文本分为文本文件和二进制文件，对于不符号要求的文件不计算在内。以下是一种较简单的实现方式：

def isTextFile(ext):

tExt = ('.c', '.cpp', '.h', '.py', '.htm', '.html', '.txt', '.lua', '.ini', '.hpp', '.lua', '.cfg')

if ext in tExt:

return True

return False

完成以上三个步骤，一个简单的代码统计工具实现了，完整代码如下：

#!/usr/bin/python

#-*- coding: utf-8 -*-

import os

result = {}

def static(path = '.'):

if os.path.isdir(path):

for  x in os.listdir(path):

childdir = os.path.join(path, x)

if os.path.isdir(childdir):

static(childdir)

elif os.path.isfile(childdir):

clacFilelines(childdir)

elif os.path.isfile(path):

clacFilelines(path)

return

def clacFilelines(filepath):

vpath = os.path.splitext(filepath)

ext = vpath[1].lower()

if not ext.strip():

return False

if not isTextFile(ext):

return False

lines = 0

with open(filepath, 'rU') as f:

filelist = f.readlines()

for line in filelist:

if not line.strip():

continue

lines += 1

if ext in result:

result[ext] += lines

else:

result[ext] = lines

print 'file:%s lines:%d' %(filepath, lines)

return True

def isTextFile(ext):

tExt = ('.c', '.cpp', '.h', '.py', '.htm', '.html', '.txt', '.lua', '.ini', '.hpp', '.lua', '.cfg')

if ext in tExt:

return True

return False

if __name__ == '__main__':

print '***********start code line static*************'

static()

print '****************static result*****************'

sum = 0

for k,v in result.iteritems():

sum += v

print '[%s]: %d lines' %(k, v)

print 'total: %d lines' %sum

print '******************static end******************'

运行结果：

***********start code line static*************

file:.\class.py lines:118

file:.\hello.py lines:483

file:.\metaclass.py lines:72

file:.\process.py lines:28

file:.\static_code_line.py lines:50

****************static result*****************

[.py]: 751 lines

total: 751 lines

******************static end******************

[Finished in 0.4s]

前后总共花了15min左右就完成了这个小程序，突然有点喜欢上python了。