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

杨东明

1. import easygui as g
   
2. import os
   
3. 
   
4. 
   
5. def show_result(start_dir):
   
6.     lines = 0
   
7.     total = 0
   
8.     text = ""
   
9. 
   
10.     for i in source_list:
    
11.         lines = source_list[i]
    
12.         total += lines
    
13.         text += "【%s】源文件 %d 个，源代码 %d 行\n" % (i, file_list[i], lines)
    
14.     title = '统计结果'
    
15.     msg = '您目前共累积编写了 %d 行代码，完成进度：%.2f %%\n离 10 万行代码还差 %d 行，请继续努力！' % (total, total / 1000, 100000 - total)
    
16.     g.textbox(msg, title, text)
    
17. 
    
18. 
    
19. def calc_code(file_name):
    
20.     lines = 0
    
21.     with open(file_name) as f:
    
22.         print('正在分析文件：%s ...' % file_name)
    
23.         try:
    
24.             for each_line in f:
    
25.                 lines += 1
    
26.         except UnicodeDecodeError:
    
27.             pass  # 不可避免会遇到格式不兼容的文件，这里忽略掉......
    
28.     return lines
    
29. 
    
30. 
    
31. def search_file(start_dir):
    
32.     if 'D:\\python file' in start_dir or 'D:\\python project' in start_dir or start_dir == 'D:\\':
    
33.         #print (1)
    
34.         os.chdir(start_dir)
    
35. 
    
36.         for each_file in os.listdir(os.curdir):
    
37.             ext = os.path.splitext(each_file)[1]
    
38.             if ext in target:
    
39.                 lines = calc_code(each_file)  # 统计行数
    
40.             # 还记得异常的用法吗？如果字典中不存，抛出 KeyError，则添加字典键
    
41.             # 统计文件数
    
42.                 try:
    
43.                     file_list[ext] += 1
    
44.                 except KeyError:
    
45.                     file_list[ext] = 1
    
46.             # 统计源代码行数
    
47.                 try:
    
48.                     source_list[ext] += lines
    
49.                 except KeyError:
    
50.                     source_list[ext] = lines
    
51. 
    
52.             if os.path.isdir(each_file):
    
53.                 search_file(each_file)  # 递归调用
    
54.                 os.chdir(os.pardir)  # 递归调用后切记返回上一层目录
    
55.     else:
    
56.         pass
    
57. 
    
58. 
    
59. target = ['.c', '.cpp', '.py', '.cc', '.java', '.pas', '.asm']
    
60. file_list = {}
    
61. source_list = {}
    
62. 
    
63. g.msgbox("请打开您存放所有代码的文件夹......", "统计代码量")
    
64. path = g.diropenbox("请选择您的代码库：")
    
65. 
    
66. search_file(path)
    
67. show_result(path)
    

复制代码

最佳答案

月排行榜 / 总排行榜

阿奇_o

2021-7-6 15:45:59

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

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

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

---- 例如

1. import os
   
2. def count_lines(path=r'.\\数据结构', ext='.py'):
   
3.     cnt = {}
   
4.     for dirpath, dirnames, files in os.walk(path):
   
5.         # print(dirpath, files)
   
6.         for f in files:
   
7.             f_path = os.path.join(dirpath, f)
   
8.             if os.path.splitext(f_path)[1] == ext:
   
9.                 with open(f_path, mode='r', encoding='utf-8') as cf:
   
10.                     cnt[f_path] = len(cf.readlines())
    
11.     return cnt
    
12. 
    
13. pylines = count_lines()
    
14. for i in pylines.items():
    
15.     print(*i)
    
16. 
    
17. print(f".py代码的总行数：{sum(pylines.values())}")
    

复制代码

跳转到最佳答案楼层

阿奇_o

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

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

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

---- 例如

1. import os
   
2. def count_lines(path=r'.\\数据结构', ext='.py'):
   
3.     cnt = {}
   
4.     for dirpath, dirnames, files in os.walk(path):
   
5.         # print(dirpath, files)
   
6.         for f in files:
   
7.             f_path = os.path.join(dirpath, f)
   
8.             if os.path.splitext(f_path)[1] == ext:
   
9.                 with open(f_path, mode='r', encoding='utf-8') as cf:
   
10.                     cnt[f_path] = len(cf.readlines())
    
11.     return cnt
    
12. 
    
13. pylines = count_lines()
    
14. for i in pylines.items():
    
15.     print(*i)
    
16. 
    
17. 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打印出来，对比一下路径是不是没有那些字眼
如 ...

感谢