统计python代码量
今天闲着没事，花了一点时间写了一个程序来统计一个目录或多个目录下python文件的代码总行数、注释总行数、空行总行数、真正代码总行数。具体代码如下:python 代码
[*]class CalculateCode():
[*] def __init__(self):
[*] self.total_count = 0
[*] self.code_count = 0
[*] self.comment_count = 0
[*] self.blank_count = 0
[*]
[*] def calculate_file(self, filename):
[*] file = open(filename, 'r')
[*] lines = file.readlines()
[*] line_count = 0
[*] line_len = len(lines)
[*] self.total_count += line_len
[*] while line_count < line_len:
[*] line_without_space = lines.strip()
[*] if not line_without_space:
[*] self.blank_count += 1
[*] line_count += 1
[*] elif line_without_space.startswith('#'):
[*] self.comment_count += 1
[*] line_count += 1
[*] elif line_without_space.startswith("'''"):
[*] self.comment_count += 1
[*] line_count += 1
[*] if not line_without_space.endswith("'''"):
[*] while line_count < line_len:
[*] comment_line = lines.strip()
[*] self.comment_count += 1
[*] line_count += 1
[*] if comment_line.endswith("'''"):
[*] break
[*] else:
[*] self.code_count += 1
[*] line_count += 1
[*]
[*] def calculate_directory(self, directory):
[*] import os
[*] if os.path.isfile(directory):
[*] if os.path.splitext(directory) == '.py':
[*] self.calculate_file(directory)
[*] else:
[*] raise Exception('The file must be a python file')
[*] elif os.path.isdir(directory):
[*] for path in os.listdir(directory):
[*] if os.path.isdir(directory+'\\'+path):
[*] self.calculate_directory(directory+'\\'+path)
[*] elif os.path.splitext(path) == '.py':
[*] self.calculate_file(directory+'\\'+path)
[*] else:
[*] pass
[*] else:
[*] raise Exception('No this directory or file!')
[*]
[*] def calculate_directories(self, directories):
[*] for directory in directories:
[*] self.calculate_directory(directory)
[*]
[*] def show_result(self):
[*] print 'total_count: %d\ncode_count: %d\ncomment_count: %d\nblank_count: %d\n' \
[*] %(self.total_count,self.code_count,self.comment_count,self.blank_count)
[*]
[*]if __name__ == '__main__':
[*] import sys
[*] if len(sys.argv) > 1:
[*] path_list = sys.argv
[*] calculate_code = CalculateCode()
[*] calculate_code.calculate_directories(path_list)
[*] calculate_code.show_result()
[*] else:
[*] print 'Please use command like this:\n"python calculatecode.py directory1 directory2 ..."'
页:
[1]