Python平台：写的源代码统计工具

'''

Created on 2013-1-8

@author: Oneday

'''

import os

import re

class pyCounter():

def __init__(self):

self.__isInDCommand = False

self.__isInSCommand = False

def __is_pyfile(self, filename):

return filename.endswith('.py')

def __is_multi_commandline(self, line):

line = line.strip()

command_result = re.findall("'''|\"\"\"", line)

if 0==len(command_result):

return self.__isInDCommand or self.__isInSCommand

for result in command_result:

if '"""' == result:

self.__isInDCommand = not self.__isInDCommand

else:

self.__isInSCommand = not self.__isInSCommand

return True

def __is_single_commandline(self, line):

if re.search(r'#', line):

return True

else:

return False

def __is_emptyline(self, line):

if re.match(r'^\s*$', line):

return True

else:

return False

def getlinenum(self, rootdir):

line_num = source_num = command_num = empty_num = 0

for parent, _, filenames in os.walk(rootdir):

for filename in filenames:

if self.__is_pyfile(filename):

with open(parent + os.sep + filename, 'r') as fp:

for line in fp.readlines():

if self.__is_multi_commandline(line):

command_num += 1

elif self.__is_single_commandline(line):

command_num += 1

elif self.__is_emptyline(line):

empty_num += 1

else:

source_num += 1

line_num += 1

return line_num, source_num, command_num, empty_num

def main():

while True:

rootdir = input('please input source folder, type "q" to quit->').strip()

if rootdir == 'q':

break

if os.path.isdir(rootdir):

line_num, source_num, command_num, empty_num = pyCounter().getlinenum(rootdir)

print('line_num->%d'%line_num)

print('source_num->%d'%source_num)

print('command_num->%d'%command_num)

print('empty_num->%d'%empty_num)

else:

raise OSError('need folder,but passed file')

if __name__ == '__main__':

main()