论.技术.Python.(旧)零学Python第035讲:代码量与十万行差距

小蜂队

1.写一个程序统计你当前代码量的总和，并显示离十万行代码量还有多远？
要求一：递归搜索各个文件夹
要求二：显示各个类型的源文件和源代码数量
要求三：显示总行数与百分比

# 代码来自小火柴棒 and 小甲鱼 and 加入我对此代码的理解
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)  # 需要显示的文本内容 and 格式化操作
        title = '统计结果'
        msg1 = '您目前共写了 %d 行代码，完成进度：%.2f %%\n离10万行代码还差 %d 行，请继续努力！' % (total, total/1000, 100000 - total)
        msg2 = '您目前共写了 %d 行代码，完成进度：%.2f %%\n超额完成 %d 行，真棒！' % (total,  total/1000, total - 100000)
        if total <= 100000:
                g.textbox(msg1, title, text)
        else:
                g.textbox(msg2, title, text)

def calc_code(file_name):
        lines = 0
        with open(file_name) as f:           # 使用 with 打开文件不用担心文件的关闭与否
                print('正在分析文件：%s ...' % file_name)
                try:                         # 使用 try 语句来检测代码块是否有问题
                        for each_line in f:
                                lines += 1
                except UnicodeDecodeError:  # 有问题则执行 except 语句，若 try 语句没有问题则跳过 except 语句，UnicodeDecodeError 表示:Unicode解码时的错误
                        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]   # os.path.splitext(path) 分离文件名与扩展名
                if ext in target:
                        lines = calc_code(each_file)    # calc_code(each_file) 返回的是 lines, 即seach_file(start_dir) 中 lines 对cale_code(file_name) lines 的引用
                        try:
                                file_list[ext] += 1
                        except KeyError:                # KeyError 关键字错误
                                file_list[ext] = 1

                        try:
                                source_list[ext] += lines
                        except KeyError:
                                source_list[ext] = lines
                if os.path.isdir(each_file):            # 递归调用， os.chdir(os.pardir) 表示返回上一级目录
                        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)