教你一招利用Python来统计你的代码量！成就感爆棚！

2019-04-22 本文已影响0人 14e61d025165

前面我们一直在介绍图形界面，这次换个口味，在命令行实现代码行数统计程序。本程序共 135 行，其中 18 行空行、110 行有效代码、7行注释，大小为 7.71 KB。

<tt-image data-tteditor-tag="tteditorTag" contenteditable="false" class="syl1555917391651 ql-align-center" data-render-status="finished" data-syl-blot="image" style="box-sizing: border-box; cursor: text; text-align: left; color: rgb(34, 34, 34); font-family: "PingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", "Helvetica Neue", Arial, sans-serif; font-size: 16px; font-style: normal; font-variant-ligatures: normal; font-variant-caps: normal; font-weight: 400; letter-spacing: normal; orphans: 2; text-indent: 0px; text-transform: none; white-space: pre-wrap; widows: 2; word-spacing: 0px; -webkit-text-stroke-width: 0px; background-color: rgb(255, 255, 255); text-decoration-style: initial; text-decoration-color: initial; display: block;">

image

主要思路

通过 argparse 设置四个参数 -f, -p, -s, -r，其中 -f 表示单个 py 文件，-p 表示文件夹地址。-s 接排序的关键字，也就是根据哪个属性显示信息。默认情况下，信息按 total 关键字降序排序，我们可以通过加上 -r 参数来得到升序序列。

对于单个 Py 文件，我们首先以只读方式读取文件，按行读取。每读取一行，total ＋ 1。如果该行长度为 0，blank ＋ 1。如果以‘#’开头，note + 1。否则的话，code + 1。其中，如果遇到单引号、双引号或者三引号，程序要单独计算这些注释的行数，主循环直接跳过相应次数的迭代。最后，通过 format 字符串打印字符。其中，代码要以英文命名，否则会显示混乱。

解析参数

我们使用标准库中的 argparse 来设置命令行参数。首先新建一个 parser 对象，并设置相应的提示信息。通过 group 对象添加的参数互斥，也就是说 -p -f 参数不能同时使用。parse_args() 用来获取对象参数。

<pre spellcheck="false" style="box-sizing: border-box; margin: 5px 0px; padding: 5px 10px; border: 0px; font-style: normal; font-variant-ligatures: normal; font-variant-caps: normal; font-variant-numeric: inherit; font-variant-east-asian: inherit; font-weight: 400; font-stretch: inherit; font-size: 16px; line-height: inherit; font-family: inherit; vertical-align: baseline; cursor: text; counter-reset: list-1 0 list-2 0 list-3 0 list-4 0 list-5 0 list-6 0 list-7 0 list-8 0 list-9 0; background-color: rgb(240, 240, 240); border-radius: 3px; white-space: pre-wrap; color: rgb(34, 34, 34); letter-spacing: normal; orphans: 2; text-align: left; text-indent: 0px; text-transform: none; widows: 2; word-spacing: 0px; -webkit-text-stroke-width: 0px; text-decoration-style: initial; text-decoration-color: initial;">import argparse
def parse_args():
parser = argparse.ArgumentParser(usage='命令行统计代码函数', description='统计文件夹或者单个文件信息')
group = parser.add_mutually_exclusive_group()
group.add_argument('-p', '--path', help='输入你的文件地址', default='')
group.add_argument('-f', '--file', help='输入你的文件名', default='')
parser.add_argument('-s', '--sort', help='排序', default='total')
parser.add_argument('-r', '--reverse', help='逆序：不需要参数', action='store_false', default=True)
args = parser.parse_args()
return args
</pre>

基本判断

简单的判断，有三个分支。长度为 0，blank + 1；以‘#’开头，note + 1。如果既不是空行，也不是注释，那么代码的行数 + 1。

注释判断

对于以引号开头的行，我们就要小心了。

<pre spellcheck="false" style="box-sizing: border-box; margin: 5px 0px; padding: 5px 10px; border: 0px; font-style: normal; font-variant-ligatures: normal; font-variant-caps: normal; font-variant-numeric: inherit; font-variant-east-asian: inherit; font-weight: 400; font-stretch: inherit; font-size: 16px; line-height: inherit; font-family: inherit; vertical-align: baseline; cursor: text; counter-reset: list-1 0 list-2 0 list-3 0 list-4 0 list-5 0 list-6 0 list-7 0 list-8 0 list-9 0; background-color: rgb(240, 240, 240); border-radius: 3px; white-space: pre-wrap; color: rgb(34, 34, 34); letter-spacing: normal; orphans: 2; text-align: left; text-indent: 0px; text-transform: none; widows: 2; word-spacing: 0px; -webkit-text-stroke-width: 0px; text-decoration-style: initial; text-decoration-color: initial;">elif line.startswith('"' or '"""' or "'" or "'''"):
if line.startswith('"""') or line.startswith("'''"):
temp_note = process_note(lines[num:], line[:3])
else:
temp_note = process_note(lines[num:], line[0])
note += temp_note
</pre>

我们需要 process_note 函数计算这些引号所占的行数。从引号开始的那一行计数，一直都引号结束的地方停止计数，最后返回注释的行数。

<pre spellcheck="false" style="box-sizing: border-box; margin: 5px 0px; padding: 5px 10px; border: 0px; font-style: normal; font-variant-ligatures: normal; font-variant-caps: normal; font-variant-numeric: inherit; font-variant-east-asian: inherit; font-weight: 400; font-stretch: inherit; font-size: 16px; line-height: inherit; font-family: inherit; vertical-align: baseline; cursor: text; counter-reset: list-1 0 list-2 0 list-3 0 list-4 0 list-5 0 list-6 0 list-7 0 list-8 0 list-9 0; background-color: rgb(240, 240, 240); border-radius: 3px; white-space: pre-wrap; color: rgb(34, 34, 34); letter-spacing: normal; orphans: 2; text-align: left; text-indent: 0px; text-transform: none; widows: 2; word-spacing: 0px; -webkit-text-stroke-width: 0px; text-decoration-style: initial; text-decoration-color: initial;"># 总行数空行数代码行注释行文件大小
def process_note(lines, symbol):
note = 0
for line in lines:
note += 1
line = line.strip()
if line.endswith(symbol):
break
return note
</pre>

与之配合的代码如下。对于引号所占的行，我们可以直接跳过，继续执行 for 循环。

可以添加QQ群1004391443，有飞机大战、颜值打分器、打砖块小游戏、红包提醒神器、小姐姐表白神器等具体的实训项目，有清晰源码，有相应的文件

教你一招利用Python来统计你的代码量！成就感爆棚！

猜你喜欢

热点阅读