统计指定路径代码量

 1 import easygui as g
 2 import os
 3
 4 def show_result(path):
 5     text=‘‘
 6     total=0
 7     for each_ext in ext_dict:
 8         total += ext_dict[each_ext]
 9         text += ‘【%s】源文件有【%d】个，代码【%d】行\n‘% (each_ext,ext_list[each_ext],ext_dict[each_ext])
10
11         msg = ‘您目前累计编写了【%d】行代码，完成进度：【%.2f%%】，离10万行代码还差【%d】，请继续努力！‘ % (total,total/1000,100000-total)
12         title =‘统计结果‘
13     g.textbox(msg,title,text)
14
15
16 def calc_line(file_name):
17     lines = 0
18     with open(file_name) as f:
19         try:
20             for each_line in f:
21                 lines+=1
22         except UnicodeDecodeError:
23             pass #遇到不兼容情况则继续
24     return lines #返回计算的行数
25
26
27 def search_file(start_dir):
28     os.chdir(start_dir)
29
30     for each_file in os.listdir(os.curdir):#遍历当前目录下的每一个文件
31
32         if os.path.isfile(each_file):#如果是一个文件
33             ext = os.path.splitext(each_file)[1]#取其扩展名
34             if ext in target:# 如果扩展名在目标中
35                 lines = calc_line(each_file)#打开这个文件并计算行数
36
37                 try:
38                     ext_dict[ext] += lines #扩展名的字典 如果扩展名不存在则新建一个并且赋值lines
39                 except KeyError:
40                     ext_dict[ext] = lines
41                 try:
42                     ext_list[ext] += 1 #记录源文件有多少个，区别于记录多少行的字典
43                 except KeyError:
44                     ext_list[ext] = 1
45
46         if os.path.isdir(each_file):#如果是目录，则创造递归
47             search_file(each_file)
48             os.chdir(os.pardir)#递归完成返回上一层目录
49
50
51 def choice_path():
52     msg = ‘请选择您的代码库‘
53     title = ‘浏览您的文件夹‘
54
55     path = g.diropenbox(msg,title)
56     return path
57
58
59
60 target=[‘.py‘,‘.c‘]
61
62 ext_dict={}
63 ext_list={}
64 path = choice_path()
65 search_file(path)
66 show_result(path)