利用Python统计源代码行数以及对源代码排版

近段时间写的一个软件需要做软件著作权登记，需要统计源代码行数以及提交部分源代码，刚好利用Python写了这么一个小工具。

用法如下：

在windows命令行窗口，使用python python_count_line.py source_code_root_directory即可。

其中，source_code_root_directory为需要统计的源代码的根目录，源代码目录中如果有其它子目录，也可以递归遍历。

源代码如下：

import os,sysdef count_line(filename):""" count the line number of file"""with open(filename) as file:lines = file.readlines()file.close()return len(lines)def read_lines_and_write_to_target(source_filename,target_file,begin_linenum):linenum = begin_linenumwith open(source_filename) as source_file:lines = source_file.readlines()for i in range(len(lines)):target_file.write("%4d"%linenum+"  ")target_file.write(lines[i])linenum = linenum+1source_file.close()return linenumdef flatten(lst):for x in lst:if isinstance(x,list):for x in flatten(x):yield xelse:yield xdef count_files(root_dir,file_filter):process_list = get_process_files(root_dir,file_filter)process_list = list(flatten(process_list))print "Flatten Process List", print process_listline_count = 0for file in process_list:line_count = line_count + count_line(file)print file,"line_number",line_count#Generate Source Files in Oneline_count = 1target_filename=root_dir+"\\"+"result.txt"with open(target_filename,"w") as target_file:for file in process_list:target_file.write("///////"+file+"///////\n")line_count = read_lines_and_write_to_target(file,target_file,line_count)target_file.close()print "Result is generated in target file:",target_filenamedef get_process_files(root_dir,file_filter):""" process all files in directory that matches file_filter """cur_dir = os.path.abspath(root_dir)file_list = os.listdir(cur_dir)print file_listprocess_list=[]#processing the file_list, retrieve all target files to process_listfor file in file_list:fullfile = cur_dir+"\\"+fileprint fullfileif os.path.isfile(fullfile):print "is file"found_flag = Falsefor filter in file_filter:if file.rfind(filter) != -1:print "match!",fileprocess_list.append(fullfile)found_flag = Trueif found_flag==False:print "pass this file",fullfileelif os.path.isdir(fullfile):print "is dir,recursive"dir_extra_list = get_process_files(fullfile,file_filter)print dir_extra_listif len(dir_extra_list)!=0:process_list.append(dir_extra_list)else:print "not defined"return process_listif __name__ == '__main__':print sys.argvroot_dir = sys.argv[1]print "INFO: Root_Dir:",root_dirfile_filter = [".cpp",".h"]count_files(root_dir,file_filter)

也许以后还用得着。

REF:

http://rosettacode.org/wiki/Flatten_a_list