|
|
发表于 2018-1-15 10:51:56
|
显示全部楼层
遇到同样的问题,以下是按答案重新敲入的代码:
import easygui as g
import os
def show_result(start_dir):
lines = 0
total = 0
text = ''
for i in source_list:
lines = source_list[i]
total += lines
text += '【%s】源文件 %d 个, 源代码 %d 行\n' % (i, file_list[i], lines)
title = '统计结果'
msg = '您目前共累积编写了 %d 行代码,完成进度: %.2f %%\n离 10 万行代码还差 %d 行,请继续努力!' % (total, total/1000, 100000-total)
g.textbox(msg, title, text)
def calc_code(file_name):
lines = 0
with open(file_name) as f:
print('正在分析文件: %s ...' % file_name)
try:
for each_line in f:
lines += 1
except UnicodeDecodeError:
pass #不可避免遇到格式不兼容的文件,这里忽略掉
return lines
def search_file(start_dir):
os.chdir(start_dir)
for each_file in os.listdir(os.curdir):
ext = os.path.splitext(each_file)[1]
if ext in target:
lines = calc_code(each_file) #统计行数
#如字典中不存在, 抛出KeyError,则添加字典键
#统计文件数
try:
file_list[ext] += 1
except KeyError:
file_list[ext] = 1
#统计源代码行数
try:
source_list[ext] += lines
except KeyError:
source_list[ext] = lines
if os.path.isdir(each_file):
search_file(each_file) #递归调用
os.chdir(os.pardir) #递归调用后切记返回上一层目录
target = ['.c', '.cpp', '.py', '.cc', '.java', '.pas', '.asm']
file_list = {}
source_list = {}
g.msgbox('请打开您存放所有代码的文件夹......', '统计代码量')
path = g.diropenbox('请选择您的代码库:')
search_file(path)
show_result(path) |
|