Python使用argparse命令行选项计算文件中单词个数

跟上一篇文章需求一样：Python计算文件中单词个数，只不过这次使用的是命令行选项与参数解析包argparse来做的。

如果想了解argparse，可以看一下这一篇文章：argparse – Command line option and argument parsing.

功能描述：

在linux命令行运行Python程序WordCount

1.如果只输入单个文件无命令则顺序输出此文件中单词个数、文本行数、字节数。

2.如果输入多个文件无命令则顺序输出这两个文件中单词个数、文本行数、字节数，并且计算每一列的总和total。

3.如1,2所示那样，只不过如果输入带有参数命令的则根据参数命令分别打印相应的文本，命令-w打印单词个数，-l打印文本行数，-c打印文本字节数。

代码如下：

程序名称为ArgsWordCount的缩写，awc.py。

import argparseimport sysbcount = 0  # 字节数wcount = 0  # 单词数lcount = 0  # 行数btotal = 0  # 字节总数wtotal = 0  # 单词总数ltotal = 0  # 行总数# 创建解释器对象parser = argparse.ArgumentParser()# 添加命令parser.add_argument('-l', action='store_true', dest='lcount', default=False, help='File line count')parser.add_argument('-w', action='store_true', dest='wcount', default=False, help='File word count')parser.add_argument('-b', action='store_true', dest='bcount', default=False, help='File word count')parser.add_argument('i',action='store_true')parser.add_argument("filepaths", nargs='*')args = parser.parse_args(sys.argv[1:])args = parser.parse_args()# 文件名IMG = args.filepaths# 读取文件for i in IMG:    if ".txt" in i:        file = open(i, 'r')        text = file.read()        file.close()    # 计算字节数        bcount = len(text)    # 计算单词数        wcount = len(text.split())    # 计算行数        lcount = len(text.split("\n"))    # 字节数总和        btotal = btotal + bcount    # 单词个数总和        wtotal = wtotal + wcount    # 行数总和        ltotal = ltotal + lcount    # 1 flags    if args.bcount and not args.lcount and not args.wcount:            print(bcount,i)    if not args.bcount and args.lcount and not args.wcount:            print(lcount,i)    if not args.bcount and not args.lcount and args.wcount:            print(wcount,i)     # 2 flags    if args.bcount and args.lcount and not args.wcount:            print(bcount,lcount,i )    if args.bcount and args.wcount and not args.lcount:            print(bcount,wcount,i )    if args.lcount and args.wcount and not args.bcount:            print(lcount,wcount,i )    # 3 flags    if args.lcount and args.wcount and args.bcount:            print(lcount,wcount,bcount,i )    # 1 total    if args.bcount and not args.lcount and not args.wcount and  btotal != bcount:            print(btotal, "total")    if args.lcount and not args.bcount and not args.wcount and  ltotal != lcount:            print(ltotal, "total")    if args.wcount and not args.lcount and not args.bcount and  wtotal != wcount:            print(wtotal, "total")    # 2 total    if args.bcount and args.lcount and not args.wcount and btotal != bcount and ltotal != lcount:            print(btotal,ltotal, "total")    if args.bcount and args.wcount and not args.lcount and btotal != bcount and wtotal != wcount:            print(btotal,wtotal, "total")    if args.lcount and args.wcount and not args.bcount and ltotal != lcount and wtotal != wcount:            print(ltotal,wtotal, "total")    # 3 total    if args.lcount and args.wcount and args.bcount and ltotal != lcount and wtotal != wcount and btotal != bcount:            print(ltotal,wtotal,bcount, "total")

运行结果：