python代码行数统计工具_Python实现代码统计工具

本文实例为大家分享了Python实现代码统计工具的具体代码&＃xff0c;供大家参考&＃xff0c;具体内容如下

思路&＃xff1a;首先获取所有文件&＃xff0c;然后统计每个文件中代码的行数&＃xff0c;最后将行数相加.

实现的功能&＃xff1a;

统计每个文件的行数&＃xff1b;

统计总行数&＃xff1b;

支持指定统计文件类型&＃xff0c;排除不想统计的文件类型&＃xff1b;

排除空行&＃xff1b;

排除注释行

import os

import sys

import os.path

#for i in sys.argv:

# print (i)

# 判断单个文件的代码行数

def count_file_lines(file_path):

line_count &＃61; 0

flag&＃61;True

try:

fp &＃61; open(file_path,"r",encoding&＃61;"utf-8")

encoding_type&＃61;"utf-8"

fp.close()

except:

encoding_type&＃61;"gbk"

with open(file_path,"r",encoding&＃61;encoding_type) as fp:

for line in fp:

#print (line_count)

if line.strip()&＃61;&＃61;"":

continue

else:

if line.strip().endswith("&＃39;&＃39;&＃39;") and flag &＃61;&＃61; False:

flag&＃61;True

continue

if line.strip().endswith(&＃39;"""&＃39;) and flag &＃61;&＃61; False:

flag&＃61;True

continue

if flag &＃61;&＃61; False:

continue

if line.strip().startswith("#encoding") or line.strip().startswith("#-*-"):

line_count &＃43;&＃61; 1

#elif line.strip().startswith(&＃39;"""&＃39;) and line.strip().endswith(&＃39;"""&＃39;) and line.strip()!&＃61;&＃39;"""&＃39;:

#continue

#elif line.strip().startswith("&＃39;&＃39;&＃39;") and line.strip().endswith("&＃39;&＃39;&＃39;") and line.strip()!&＃61;"&＃39;&＃39;&＃39;":

#continue

elif line.strip().startswith(&＃39;#&＃39;):

continue

elif line.strip().startswith("&＃39;&＃39;&＃39;") and flag &＃61;&＃61; True:

flag &＃61; False

continue

elif line.strip().startswith(&＃39;"""&＃39;) and flag &＃61;&＃61; True:

flag &＃61; False

continue

else:

line_count &＃43;&＃61; 1

return line_count

def count_code_lines(path,file_types&＃61;[]):

# 判断路径是否存在

if not os.path.exists(path):

print("您输入的目录或文件路径不存在")

return 0

line_count&＃61;0 #代码行总数

file_lines_dict&＃61;{} #每个文件代码行数

# 判断是否为文件

if os.path.isfile(path):

file_type &＃61; os.path.splitext(path)[1][1:] #取到文件后缀名

# 判断文件类型是否满足条件

if len(file_types)&＃61;&＃61;0:

file_types&＃61;["py","cpp","c","java","ruby","ini","go","html","css","js","txt","vbs","php","asp","sh"]

if file_type in file_types:

line_count &＃61; count_file_lines(path)

return line_count

else:

file_path &＃61; []

for root, dirs, files in os.walk(path):

for file in files:

file_path.append(os.path.join(root,file))

for f in file_path:

file_type &＃61; os.path.splitext(f)[1][1:]

if len(file_types)&＃61;&＃61;0:

file_types&＃61;

["py","cpp","c","java","ruby","ini","go","html","css","js","txt","vbs","php","asp","sh"]

if file_type not in file_types:

continue

line_num &＃61; count_file_lines(f)

line_count &＃43;&＃61; line_num

file_lines_dict[f] &＃61; line_num

return line_count,file_lines_dict

if __name__&＃61;&＃61;"__main__":

print (sys.argv)

if len(sys.argv) <2:

print ("请输入待统计行数的代码绝对路径&＃xff01;")

sys.exit()

count_path &＃61; sys.argv[1]

file_types &＃61; []

if len(sys.argv) >2:

for i in sys.argv[2:]:

file_types.append(i)

#print(count_path,file_types)

print(count_code_lines(count_path,file_types))

#print(count_file_lines("b.py"))

以上就是本文的全部内容&＃xff0c;希望对大家的学习有所帮助&＃xff0c;也希望大家多多支持脚本之家。