设为首页 - 加入收藏 ASP站长网(Aspzz.Cn)- 科技、建站、经验、云计算、5G、大数据,站长网!
热搜: 创业者 数据 手机
当前位置: 首页 > 运营中心 > 建站资源 > 优化 > 正文

Python学习教程:如何用Python统计代码行数

发布时间:2019-08-01 16:43 所属栏目:21 来源:千锋python学院
导读:Python学习教程:如何用python统计代码行数 改良后的代码可以对python和C系列的代码实行行数计算,包括代码、空行和注释行,用re抓取注释,传入一个目录自动对其下的文件进行读取计算。 流程 首先判断传入参数是否为文件夹,不是则打印出提示,否则继续(无

 Python学习教程:如何用python统计代码行数

改良后的代码可以对python和C系列的代码实行行数计算,包括代码、空行和注释行,用re抓取注释,传入一个目录自动对其下的文件进行读取计算。

Python学习教程:如何用Python统计代码行数

流程

首先判断传入参数是否为文件夹,不是则打印出提示,否则继续(无返回),获得目录后,yongos.listdir对路径下文件进行遍历,其中也包含文件夹,再次判断是否为文件夹,是的话则递归调用此函数,否则开始执行行数统计,这里用os.path.join将路径与文件名进行拼接,方便之后直接传给函数,逻辑很简单,无非是执行文件判断,判断是哪类文件,在调用对应的注释监测正则代码段进行抓取,抓取到则行数+1,空白行也是一样的原理,用strip(去除前后空格),然后行内内容为空则为空行,代码段即为总行数减去其他两类行数,最后在外层将所有文件对应的代码段累加即为total。

关键

函数内部是可以访问全局变量的,问题在于函数内部修改了变量,导致python认为它是一个局部变量。

所以,如果在函数内部访问并修改全局变量,应该使用关键字 global 来修饰变量。

  1. import os 
  2. import re 
  3. #定义规则抓取文件中的python注释 
  4. re_obj_py = re.compile('[(#)]') 
  5. #定义规则抓取文件中的C语言注释 
  6. re_obj_c = re.compile('[(//)(/*)(*)(*/)]') 
  7. #判断是否为python文件 
  8. def is_py_file(filename): 
  9. if os.path.splitext(filename)[1] == '.py': 
  10. return True 
  11. else: 
  12. return False 
  13. #判断是否为c文件 
  14. def is_c_file(filename): 
  15. if os.path.splitext(filename)[1] in ['.c', '.cc', '.h']: 
  16. return True 
  17. else: 
  18. return False 
  19. #定义几个全局变量用于计算所有文件总和(全部行数、代码行数、空行数、注释行数) 
  20. all_lines, code_lines, space_lines, comments_lines = 0, 0, 0, 0 
  21. #判断是否为文件夹,不是则输出提示 
  22. def count_codelines(dirpath): 
  23. if not os.path.isdir(dirpath): 
  24. print('input dir: %s is not legal!' % dirpath) 
  25. return 
  26. # 定义几个全局变量用于计算每个文件行数(全部行数、代码行数、空行数、注释行数) 
  27. global all_lines, code_lines, space_lines, comments_lines 
  28. #列出当前文件夹下的文件(包含目录) 
  29. all_files = os.listdir(dirpath) 
  30. for file in all_files: 
  31. #将文件(目录)名与路径拼接 
  32. file_name = os.path.join(dirpath, file) 
  33. if os.path.isdir(file_name): 
  34. count_codelines(file_name) 
  35. else: 
  36. temp_all_lines, temp_code_lines, temp_space_lines, temp_comments_lines = 0, 0, 0, 0 
  37. f = open(file_name) 
  38. for line in f: 
  39. temp_all_lines += 1 
  40. if line.strip() == '': 
  41. temp_space_lines += 1 
  42. continue 
  43. if is_py_file(file_name) and re_obj_py.match(line.strip()): 
  44. temp_comments_lines += 1 
  45. if is_c_file(file_name) and re_obj_c.match(line.strip()): 
  46. temp_comments_lines += 1 
  47. temp_code_lines = temp_all_lines - temp_space_lines - temp_comments_lines 
  48. print('%-15s : all_lines(%s)\t code_lines(%s)\t space_lines(%s)\t comments_lines(%s)' 
  49. % (file, temp_all_lines, temp_code_lines, temp_space_lines, temp_comments_lines)) 
  50. all_lines += temp_all_lines 
  51. code_lines += temp_code_lines 
  52. space_lines += temp_space_lines 
  53. comments_lines += temp_comments_lines 
  54. if __name__ == '__main__': 
  55. count_codelines('test') 
  56. print('\n**** TOTAL COUNT ****\nall_lines = %s\ncode_lines = %s\nspace_lines = %s\ncomments_lines = %s' % (all_lines, code_lines, space_lines, comments_lines)) 

本期的Python学习教程先跟大家分享这么多!

(编辑:ASP站长网)

    网友评论
    推荐文章
      热点阅读