Python3 使用GUI统计代码量的例子

# coding=utf-8
'''
选择一个路径
遍历路径下的每一个文件，统计代码量
字典存储 每一种类型文件的代码行数，eg：  *.py -> 行数
全局变量 总行数

需要注意的是，这里仅仅能打开utf-8编码的文件，其他类型的文件无法打开，会出现解码错误
解决方法：使用try-except语句，遇到解码错误就跳过，即 except UnicodeDecodeError:
'''
import easygui as g
import sys
import os

# 全局变量 总行数
total_line_num = 0
# 字典存储 每一种类型文件的代码行数，eg：  *.py -> 行数
code_file_dict = {}


def func1(file_path):
    if os.path.isdir(file_path):
        file_list = os.listdir(file_path)  # 列出当前路径下的全部内容
        for each in file_list:
            path_plus = file_path + os.sep + each
            if os.path.isdir(path_plus):
                if os.path.basename(path_plus) in [
                        'venv', '.idea']:  # 如果目录为venv或者.idea，则跳过，不统计
                    pass
                else:
                    func1(path_plus)
            elif os.path.isfile(path_plus):
                try:
                    with open(path_plus, 'r') as f:
                        # 每个文件的代码行数
                        line_num = 0
                        for eachline in f:
                            global total_line_num  # 声明全局变量
                            total_line_num += 1
                            line_num += 1
                        '''
                        将each分割出后缀名，存储在字典中
                        '''
                        (temp_path, temp_name) = os.path.basename(each).split('.')
                        temp = '.' + temp_name
                        global code_file_dict
                        if temp not in code_file_dict:
                            code_file_dict[temp] = line_num
                        else:
                            code_file_dict[temp] += line_num
                except UnicodeDecodeError:
                    pass
    else:
        g.msgbox('该路径只是一个文件', '提示')
        sys.exit(0)


if __name__ == '__main__':
    try:
        dir = g.diropenbox('请选择的你的代码库', '浏览文件夹', default='.')
        func1(dir)
        print(code_file_dict)
        g.textbox(
            '总行数为:{}\n已经完成了{}%\n离十万行代码还差{}行'.format(
                total_line_num,
                (total_line_num / 100000) * 100,
                100000 - total_line_num),
            title='统计结果',
            text=[
                '{a}类型的代码有{b}行\n'.format(a=k,b=v) for k,v in code_file_dict.items()],
            codebox=1)
    except TypeError as reason:
        g.msgbox('取消了统计代码行操作')