Python 基础实战 -- 统计代码量
1 import os 2 import easygui as g 3 4 def StatisticeCodeLine(dir_name): 5 file_dict = {".py":[0,0],".c":[0,0],".cpp":[0,0],".pas":[0,0],".asm":[0,0]} 6 all_child_dir = os.walk(dir_name) 7 8 for item in all_child_dir: 9 for file in item[2]: 10 file_name,file_extension = os.path.splitext(file) 11 if file_extension in file_dict: 12 file_dict[file_extension][0] += 1 13 14 #------------------------------------------------------------ 15 with open(os.path.join(item[0],file),"r",encoding='utf-8') as f: 16 for i in f: 17 file_dict[file_extension][1] += 1 18 #------------------------------------------------------------ 19 20 else: 21 pass 22 23 return file_dict 24 25 26 dir_name = g.diropenbox() #这里其实是一个获取需要递归的目录,在目录里找 27 file_dict = StatisticeCodeLine(dir_name) 28 total_line_count = 0 29 detail = "" 30 for key,value in file_dict.items(): 31 total_line_count += value[1] 32 detail += str(key) + "源文件" + str(value[0]) + "个,源代码" + str(value[1])+"行\n" 33 34 total_line_count = "您目前累积编写了{0}行代码,完成度{1}%\n离10万行代码还差{2}行,请继续努力!".format(total_line_count,total_line_count / 1000,100000-total_line_count) 35 36 g.textbox(total_line_count,"统计结果",text=detail) #其实你想的话这里也可以用print替代