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

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

用法如下：

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

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

源代码如下：

import os,sys

def 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_linenum
with 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+1
source_file.close()
return linenum

def flatten(lst):
for x in lst:
if isinstance(x,list):
for x in flatten(x):
yield x
else:
yield x

def 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_list

line_count = 0
for file in process_list:
line_count = line_count + count_line(file)
print file,"line_number",line_count

#Generate Source Files in One
line_count = 1
target_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_filename

def 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_list
process_list=[]
#processing the file_list, retrieve all target files to process_list
for file in file_list:
fullfile = cur_dir+"\\"+file
print fullfile
if os.path.isfile(fullfile):
print "is file"
found_flag = False
for filter in file_filter:
if file.rfind(filter) != -1:
print "match!",file
process_list.append(fullfile)
found_flag = True
if found_flag==False:
print "pass this file",fullfile
elif os.path.isdir(fullfile):
print "is dir,recursive"
dir_extra_list = get_process_files(fullfile,file_filter)
print dir_extra_list
if len(dir_extra_list)!=0:
process_list.append(dir_extra_list)
else:
print "not defined"
return process_list

if __name__ == '__main__':
print sys.argv
root_dir = sys.argv[1]
print "INFO: Root_Dir:",root_dir
file_filter = [".cpp",".h"]
count_files(root_dir,file_filter)

也许以后还用得着。

REF:

http://rosettacode.org/wiki/Flatten_a_list