[已解决]<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'
我用的是迭代去查找文件夹，不知道怎么去修改

1. import easygui as g
   
2. import os
   
3. 
   
4. #递归搜索各个文件
   
5. def search_file(start_file):
   
6.     os.chdir(start_file)#将指定路径设置成当前工作空间
   
7.     file_name=os.listdir(os.curdir)
   
8.     line_numbers=0
   
9.     result_dict={}
   
10.     for each_file in file_name:
    
11.         if os.path.isdir(each_file):
    
12.             search_file(each_file)
    
13.         else:
    
14.             file_type=os.path.splitext(each_file)[1]
    
15.             if file_type in target:
    
16.                 with open(each_file,'rb') as f:
    
17.                     for i in f:
    
18.                         line_numbers+=1
    
19.                     result_dict[each_file]=line_numbers
    
20.                     line_numbers=0
    
21.     return result_dict
    
22. 
    
23. 
    
24. #主程序
    
25. target=['.py','.c','.cpp','.java','.asm','.cc']#文件类型
    
26. path=g.diropenbox('请选择您的代码库：')
    
27. cc=search_file(path)#此函数返回字典（名字：行数）
    
28. aa=list(cc.values())
    
29. 
    
30. print(cc.items())
    
31. 
    
32. file_number=len(aa)
    
33. result=sum(aa)
    
34. o=float(result/1000)
    
35. ok=str(o)+'%'
    
36. msg='目标：十万行代码'
    
37. title='统计代码量'
    
38. fields=['代码篇数：','总共代码行数：','完成目标进度：']
    
39. values=[file_number,result,ok]
    
40. g.multenterbox(msg, title, fields,values)

复制代码

最佳答案

月排行榜 / 总排行榜

青出于蓝

2021-7-28 11:46:29

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

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

跳转到最佳答案楼层

青出于蓝

1. import easygui as g
   
2. import os
   
3. 
   
4. #递归搜索各个文件
   
5. def search_file(start_file):
   
6.     os.chdir(start_file)#将指定路径设置成当前工作空间
   
7.     file_name=os.listdir(os.curdir)
   
8.     line_numbers=0
   
9.     result_dict={}
   
10.     n=0
    
11.     for each_file in file_name:
    
12.         if os.path.isdir(each_file):
    
13.            file_name.remove(each_file)
    
14.         else:
    
15.             file_type=os.path.splitext(each_file)[1]
    
16.             
    
17.             if file_type in target:
    
18.                 with open(each_file,'rb') as f:
    
19.                     for i in f:
    
20.                         line_numbers+=1
    
21.                     result_dict[each_file]=line_numbers
    
22.                     line_numbers=0
    
23.     return result_dict
    
24. 
    
25. 
    
26. #主程序
    
27. 
    
28. target=['.py','.c','.cpp','.java','.asm','.cc']#文件类型
    
29. path=g.diropenbox('请选择您的代码库：')
    
30. cc=search_file(path)#此函数返回字典（名字：行数）
    
31. aa=list(cc.values())
    
32. 
    
33. print(cc.items())
    
34. 
    
35. file_number=len(aa)
    
36. result=sum(aa)
    
37. o=float(result/1000)
    
38. ok=str(o)+'%'
    
39. msg='目标：十万行代码'
    
40. title='统计代码量'
    
41. fields=['代码篇数：','总共代码行数：','完成目标进度：']
    
42. values=[file_number,result,ok]
    
43. g.multenterbox(msg, title, fields,values)
    

复制代码

青出于蓝

第12行改成

1. file_name.remove(each_file)

复制代码

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

小菜鸟FLY

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

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

青出于蓝

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

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

小菜鸟FLY

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

好吧