[已解决]请问searchu_file函数下第一句是我自己加的，为啥加了后读取的代码为0??

杨东明

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):
    if 'D:\\python file' in start_dir or 'D:\\python project' in start_dir or start_dir == 'D:\\':
        #print (1)
        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)  # 递归调用后切记返回上一层目录
    else:
        pass


target = ['.c', '.cpp', '.py', '.cc', '.java', '.pas', '.asm']
file_list = {}
source_list = {}

g.msgbox("请打开您存放所有代码的文件夹......", "统计代码量")
path = g.diropenbox("请选择您的代码库：")

search_file(path)
show_result(path)

最佳答案

月排行榜 / 总排行榜

阿奇_o

2021-7-6 15:45:59

你的思路，搞复杂了，简单点：

先实现 针对一个指定目录内的所有.py文件，进行统计，然后再“泛化”。

ps: 最好不要 修改当前工作路径，(即尽量不用 os.chdir()，否则可能会搞乱或意外），
递归查看所有文件夹里的文件（不用自己实现），只需要用 os.walk() 即可。
另外，路径，尽量用 os.path.join() 。

---- 例如

import os
def count_lines(path=r'.\\数据结构', ext='.py'):
    cnt = {}
    for dirpath, dirnames, files in os.walk(path):
        # print(dirpath, files)
        for f in files:
            f_path = os.path.join(dirpath, f)
            if os.path.splitext(f_path)[1] == ext:
                with open(f_path, mode='r', encoding='utf-8') as cf:
                    cnt[f_path] = len(cf.readlines())
    return cnt

pylines = count_lines()
for i in pylines.items():
    print(*i)

print(f".py代码的总行数：{sum(pylines.values())}")

跳转到最佳答案楼层

阿奇_o

你的思路，搞复杂了，简单点：

先实现 针对一个指定目录内的所有.py文件，进行统计，然后再“泛化”。

ps: 最好不要 修改当前工作路径，(即尽量不用 os.chdir()，否则可能会搞乱或意外），
递归查看所有文件夹里的文件（不用自己实现），只需要用 os.walk() 即可。
另外，路径，尽量用 os.path.join() 。

---- 例如

import os
def count_lines(path=r'.\\数据结构', ext='.py'):
    cnt = {}
    for dirpath, dirnames, files in os.walk(path):
        # print(dirpath, files)
        for f in files:
            f_path = os.path.join(dirpath, f)
            if os.path.splitext(f_path)[1] == ext:
                with open(f_path, mode='r', encoding='utf-8') as cf:
                    cnt[f_path] = len(cf.readlines())
    return cnt

pylines = count_lines()
for i in pylines.items():
    print(*i)

print(f".py代码的总行数：{sum(pylines.values())}")

杨东明

题目是：
写一个程序统计你当前代码量的总和，并显示离十万行代码量还有多远？?-D+=mG
RN'o@:cu(O5=&?{t2q>!SK[|<;QPV
要求一：递归搜索各个文件夹
要求二：显示各个类型的源文件和源代码数量
要求三：显示总行数与百分比

因为我把python有关的东西都放在D盘，我想的就是单单计数python file 和python project两个文件夹

西瓜味的苹果

哇，8鱼币啊

青出于蓝

                               
登录/注册后可看大图

这里看程序写的挺好的，没有问题
建议LZ把path打印出来，对比一下路径是不是没有那些字眼
如还有问题欢迎追问

leafu

测试了一下代码，我也出现了你这个问题。
调试顺序：
1、打印path，确定path是否包含在条件内，建议用   r‘C:\Users\Administrator\Desktop\’，而不是用\
2、在各函数插入测试代码print（x），x根据位置改变
3、发现错误位置
主要原因就是调用calc_code(file_name)函数出现UnicodeDecodeError错误，原因是你要打开的文件里面有中文，所以建议在文件打开方式里面添加上encoding = "UTF-8"，指定编码方式。

杨东明

leafu 发表于 2021-7-6 16:47
测试了一下代码，我也出现了你这个问题。
调试顺序：
1、打印path，确定path是否包含在条件内，建议用    ...

感谢

杨东明

青出于蓝 发表于 2021-7-6 16:32
这里看程序写的挺好的，没有问题
建议LZ把path打印出来，对比一下路径是不是没有那些字眼
如 ...

感谢