python实现代码统计程序


Posted in Python onSeptember 19, 2019

本文实例为大家分享了python实现代码统计程序的具体代码,供大家参考,具体内容如下

# encoding="utf-8"

"""
统计代码行数
"""

import sys
import os

def count_file_line(path):
 """统计文件的有效行数"""
 countLine = 0
 # 设置一个标志位,当遇到以"""或者'''开头或者结尾的时候,置为False
 flag = True

 # 使用utf-8格式的编码方式读取文件,如果读取失败,将使用gbk编码方式读取文件
 try:
 fp = open(path, "r", encoding="utf-8")
 encoding_type = "utf-8"
 fp.close()
 except:
 encoding_type = "gbk"

 with open(path, "r", encoding=encoding_type) as fp:
 for line in fp:
  # 空行不统计
  if line.strip():
  line = line.strip()
  # 注意下面的这两个elif必须要前面,这样子当('"""')结束之后及时将flag置为True
  if line.endswith('"""') and flag == False:
   flag = True
   continue
  if line.endswith("'''") and flag == False:
   flag = True
   continue
  if flag == False:
   continue
  if line.startswith("#!") or line.startswith("#-*-") or line.startswith("# encoding"):
   countLine += 1
  # 如果以“#”号开头的,不统计
  elif line.startswith("#"):
   continue
  # 如果同时以("'''")或者('"""')开头或者结尾(比如:"""aaa"""),那么不统计
  elif line.startswith('"""') and line.endswith('"""') and line != '"""':
   continue
  elif line.startswith("'''") and line.endswith("'''") and line != "'''":
   continue
  # 如果以("'''")或者('"""')开头或者结尾(比如:aaa"""或者"""bbb),那么不统计
  # 注意下面的这两个elif必须要放后面
  elif line.startswith('"""') and flag == True:
   flag = False
   continue
  elif line.startswith("'''") and flag == True:
   flag = False
   continue
  else:
   countLine += 1
 return countLine

def count_codes(path,file_types=[]):
 """统计所有文件代码行"""
 # 判断path是目录还是文件,如果是目录的话,遍历目录下所有的文件
 if not os.path.exists(path):
 print("您输入的路径不存在!")
 return 0
 countTotalLine = 0
 file_paths = {}
 if os.path.isdir(path):
 for root,dirs,files in os.walk(path):
  for name in files:
  if not file_types:
   file_types = ["txt","py"]
   # print(file_types)
  if os.path.splitext(name)[1][1:] in file_types:
   file_path = os.path.normpath(os.path.join(root,name))
   # print(file_path)
   file_lines = count_file_line(file_path)
   countTotalLine += file_lines
   file_paths[file_path] = file_lines
 else:
 if not file_types:
  file_types = ["txt","py"]
 if os.path.splitext(path)[1][1:] in file_types:
  countTotalLine = count_file_line(path)
  file_paths[path] = count_file_line(path)

 return countTotalLine,file_paths


if __name__ == "__main__":
 # 打印出命令行输入的参数
 # print(sys.argv)
 if len(sys.argv) < 2:
 print("请输入路径!")
 sys.exit()
 path = sys.argv[1]
 # print(path)
 file_types = sys.argv[2:]
 # print(file_types)
 print(count_codes(path,file_types))

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中lambda的用法及其与def的区别解析
Jul 28 Python
Python中实现三目运算的方法
Jun 21 Python
python实现装饰器、描述符
Feb 28 Python
Python批量提取PDF文件中文本的脚本
Mar 14 Python
Python常见MongoDB数据库操作实例总结
Jul 24 Python
python学生管理系统开发
Jan 30 Python
Django实现CAS+OAuth2的方法示例
Oct 30 Python
pytorch之添加BN的实现
Jan 06 Python
python中的django是做什么的
Jul 31 Python
Django crontab定时任务模块操作方法解析
Sep 10 Python
解决numpy和torch数据类型转化的问题
May 23 Python
python tkinter图形界面代码统计工具(更新)
Sep 18 #Python
python3获取url文件大小示例代码
Sep 18 #Python
弄懂这56个Python使用技巧(轻松掌握Python高效开发)
Sep 18 #Python
python3使用GUI统计代码量
Sep 18 #Python
django中的图片验证码功能
Sep 18 #Python
python tkinter图形界面代码统计工具
Sep 18 #Python
Python自动生成代码 使用tkinter图形化操作并生成代码框架
Sep 18 #Python
You might like
十天学会php之第十天
2006/10/09 PHP
PHP 字符串分割和比较
2009/10/06 PHP
ThinkPHP的URL重写问题
2014/06/22 PHP
解决phpcms更换javascript的幻灯片代码调用图片问题
2014/12/26 PHP
php实现屏蔽掉黑帽SEO的搜索关键字
2015/04/15 PHP
PHP+Apache+Mysql环境搭建教程
2016/08/01 PHP
ThinkPHP5.1框架数据库链接和增删改查操作示例
2019/08/03 PHP
JavaScipt基本教程之JavaScript语言的基础
2008/01/16 Javascript
javascript TextArea动态显示剩余字符
2008/10/22 Javascript
基于jQuery捕获超链接事件进行局部刷新代码
2012/05/10 Javascript
javascript for循环从入门到偏门(效率优化+奇特用法)
2012/08/01 Javascript
jquery中focus()函数实现当对象获得焦点后自动把光标移到内容最后
2013/09/29 Javascript
jQuery的选择器中的通配符[id^='code']或[name^='code']及jquery选择器总结
2015/12/24 Javascript
javascript中错误使用var造成undefined
2016/03/31 Javascript
微信小程序 视图容器组件的详解及实例代码
2017/01/19 Javascript
webpack打包单页面如何引用的js
2017/06/07 Javascript
Django+Vue跨域环境配置详解
2018/07/06 Javascript
JavaScript设计模式之构造器模式(生成器模式)定义与用法实例分析
2018/07/26 Javascript
Javascript通过控制类名更改样式
2019/05/24 Javascript
Layui实现数据表格中鼠标悬浮图片放大效果,离开时恢复原图的方法
2019/09/11 Javascript
vue点击页面空白处实现保存功能
2019/11/06 Javascript
React冒泡和阻止冒泡的应用详解
2020/08/18 Javascript
浅析Python多线程下的变量问题
2015/04/28 Python
python实现文件路径和url相互转换的方法
2015/07/06 Python
Python判断列表是否已排序的各种方法及其性能分析
2016/06/20 Python
Python控制键盘鼠标pynput的详细用法
2019/01/28 Python
在python tkinter中Canvas实现进度条显示的方法
2019/06/14 Python
pycharm访问mysql数据库的方法步骤
2019/06/18 Python
MNIST数据集转化为二维图片的实现示例
2020/01/10 Python
Python中join()函数多种操作代码实例
2020/01/13 Python
python openssl模块安装及用法
2020/12/06 Python
实习生的自我鉴定范文欣赏
2013/11/20 职场文书
庆祝教师节标语
2014/10/09 职场文书
2015年度绩效考核工作总结
2015/05/27 职场文书
卡特教练观后感
2015/06/08 职场文书
golang生成vcf通讯录格式文件详情
2022/03/25 Golang