本文介绍代码统计工具控制台界面部分。
控制台界面就是用命令行形式执行程序。主要内容是输入参数和结果输出。Python有一个命令行解析模块getopt。它能解析两种选项格式：短格式，长格式。
短格式："-"号后面要紧跟一个选项字母。如果还有此选项的附加参数，可以用空格分开，也可以不分开。长度任意，可以用引号。
-v [正确]
-omyfile.py [正确]
-o myfile.py [正确]
-o "myfile.py" [正确]
长格式："--"号后面要跟一个单词。如果还有些选项的附加参数，后面要紧跟"="，再加上参数。"="号前后不能有空格。
--output=myfile.py [正确]
-- output=myfile.py [不正确]
--output= myfile.py [不正确]
--output = myfile.py [不正确]
getopt模块中函数介绍
opts, args=getopt(args, shortopts, longopts=[])
参数args：需要解析的全部字符串
参数shortopt：短格式选项，如'hf:'，h表示-h，f:表示-f string，冒号表示后面跟有参数。
参数longopt：长格式选项，如['help', 'output=']，help表示--help，output=表示output=string，等号表示选项需要赋值。
返回值opts：选项和参数对。
返回值args：没有匹配上的选项的参数。
注：-f filename.py 中‘-f’是选项，‘filename’是参数。
如果匹配不符就会抛出异常，如格式是'hf:'，则（假设程序名为program）
program -s [不适合的选项]
program -f [此选项需要跟参数]
都会抛出异常。
此程序制裁采用了短格式，它需要调用counter.py的函数。
下面是CodeCounter.py中的代码：
# -*- coding: cp936 -*-
'''
统计文件或目录的代码和注释
usage: CodeCounter [-hfdlmto] [string|n]
-h 显示帮助信息
-f string 统计string文件
-d string 统计string目录
-l n 统计n层子目录,-1表示所有子目录
-m string 要统计文件的后缀名,格式如:.c,.h,.cpp
-t string 代码类型,c表示C/C++语言,py表示Python
-o n 输出格式,0表示简易输出,1表示全部输出
'''
import sys
import getopt
# 统计工具的工作部分,见couter.py
from counter import CodeCounter
def usage():
print __doc__
def errmsg():
print "参数不对,请参考帮助信息. \nCodeCounter -h显示帮助信息."
def print_result(result, out):
print "全部\t代码\t注释\t空行\t文件名"
total = [0,0,0,0]
for ele in result:
total[0] += ele[1][4]
total[1] += ele[1][1]+ele[1][3]
total[2] += ele[1][2]+ele[1][3]
total[3] += ele[1][0]
if out == 1:
for ele in result:
print "%d\t%d\t%d\t%d\t%s" %(ele[1][4], ele[1][1]+ele[1][3],
ele[1][2]+ele[1][3], ele[1][0], ele[0])
print "%d\t%d\t%d\t%d\t总计" % (total[0], total[1], total[2], total[3])
def main(argv):
# 解析参数
try:
opts, args = getopt.getopt(argv[1:], 'hf:d:m:l:t:o:')
except getopt.GetoptError:
errmsg()
return
(counter, out, result) = (CodeCounter(), 0, [])
# 设置参数
for opt, arg in opts:
if opt in ['-h']:
usage()
return
elif opt in ['-f']:
counter.AddCodeFiles('f', arg)
elif opt in ['-d']:
counter.AddCodeFiles('d', arg)
elif opt in ['-m']:
counter.SetModes(arg)
elif opt in ['-l']:
counter.SetLevel(int(arg))
elif opt in ['-t']:
counter.SetCodeType(arg)
elif opt in ['-o']:
out = int(arg)
# 统计和输出结果
counter.Count(result)
print_result(result, out)
if __name__ == '__main__':
try:
main(sys.argv)
except:
errmsg()运行如下：
D:\Work\stat>CodeCounter.py -h
统计文件或目录的代码和注释
usage: CodeCounter [-hfdlmto] [string|n]
-h 显示帮助信息
-f string 统计string文件
-d string 统计string目录
-l n 统计n层子目录,-1表示所有子目录
-m string 要统计文件的后缀名,格式如:.c,.h,.cpp
-t string 代码类型,c表示C/C++语言,py表示Python
-o n 输出格式,0表示简易输出,1表示全部输出
D:\Work\stat>CodeCounter.py -d . -o 1
全部 代码 注释 空行 文件名
46 40 15 3 .\stat.
46 40 15 3 总计
待续