[已解决]<python基础入门>统计代码量的程序问题

小菜鸟FLY

这是仿照小甲鱼的程序写的一个可以查找指定文件夹里写的代码量，实现的时候如果文件夹里面还有文件夹，就会出错：
Traceback (most recent call last):
  File "D:\mypython\python_py\统计代码量界面.py", line 27, in <module>
    cc=search_file(path)#此函数返回字典（名字：行数）
  File "D:\mypython\python_py\统计代码量界面.py", line 16, in search_file
    with open(each_file,'rb') as f:
FileNotFoundError: [Errno 2] No such file or directory:'wo.py'
我用的是迭代去查找文件夹，不知道怎么去修改

import easygui as g
import os

#递归搜索各个文件
def search_file(start_file):
    os.chdir(start_file)#将指定路径设置成当前工作空间
    file_name=os.listdir(os.curdir)
    line_numbers=0
    result_dict={}
    for each_file in file_name:
        if os.path.isdir(each_file):
            search_file(each_file)
        else:
            file_type=os.path.splitext(each_file)[1]
            if file_type in target:
                with open(each_file,'rb') as f:
                    for i in f:
                        line_numbers+=1
                    result_dict[each_file]=line_numbers
                    line_numbers=0
    return result_dict


#主程序
target=['.py','.c','.cpp','.java','.asm','.cc']#文件类型
path=g.diropenbox('请选择您的代码库：')
cc=search_file(path)#此函数返回字典（名字：行数）
aa=list(cc.values())

print(cc.items())

file_number=len(aa)
result=sum(aa)
o=float(result/1000)
ok=str(o)+'%'
msg='目标：十万行代码'
title='统计代码量'
fields=['代码篇数：','总共代码行数：','完成目标进度：']
values=[file_number,result,ok]
g.multenterbox(msg, title, fields,values)

最佳答案

月排行榜 / 总排行榜

青出于蓝

2021-7-28 11:46:29

小菜鸟FLY 发表于 2021-7-28 11:30
但是这样就把文件夹给忽略了，文件夹内还有文件就统计不到了

那找到文件夹后改变工作空间不就行吗

跳转到最佳答案楼层

青出于蓝

import easygui as g
import os

#递归搜索各个文件
def search_file(start_file):
    os.chdir(start_file)#将指定路径设置成当前工作空间
    file_name=os.listdir(os.curdir)
    line_numbers=0
    result_dict={}
    n=0
    for each_file in file_name:
        if os.path.isdir(each_file):
           file_name.remove(each_file)
        else:
            file_type=os.path.splitext(each_file)[1]
            
            if file_type in target:
                with open(each_file,'rb') as f:
                    for i in f:
                        line_numbers+=1
                    result_dict[each_file]=line_numbers
                    line_numbers=0
    return result_dict


#主程序

target=['.py','.c','.cpp','.java','.asm','.cc']#文件类型
path=g.diropenbox('请选择您的代码库：')
cc=search_file(path)#此函数返回字典（名字：行数）
aa=list(cc.values())

print(cc.items())

file_number=len(aa)
result=sum(aa)
o=float(result/1000)
ok=str(o)+'%'
msg='目标：十万行代码'
title='统计代码量'
fields=['代码篇数：','总共代码行数：','完成目标进度：']
values=[file_number,result,ok]
g.multenterbox(msg, title, fields,values)

青出于蓝

第12行改成

file_name.remove(each_file)

从文件名列表中直接把文件夹的名删掉就好
有问题欢迎追问～

小菜鸟FLY

青出于蓝 发表于 2021-7-28 08:06
第12行改成
从文件名列表中直接把文件夹的名删掉就好
有问题欢迎追问～

但是这样就把文件夹给忽略了，文件夹内还有文件就统计不到了

青出于蓝

小菜鸟FLY 发表于 2021-7-28 11:30
但是这样就把文件夹给忽略了，文件夹内还有文件就统计不到了

那找到文件夹后改变工作空间不就行吗

小菜鸟FLY

青出于蓝 发表于 2021-7-28 11:46
那找到文件夹后改变工作空间不就行吗

好吧