python大型项目源代码行数_利用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: