如何实现命令行切换到我的脚本？

Question

我是 python 的新手。我正在编写一个计算单词、行和字符的程序。当我尝试使用命令行开关时开始遇到问题：-w、-l、-c，直到那时一切正常。

我阅读了 Whosebug 上的帖子和 python 有关 argparse 的文档，但我现在不知道如何实现 argparse 库以及与之一起使用的代码。

当我运行python wc.py file.txt --l

我明白了

too many values to unpac

有人可以帮我解决这个问题吗？

from sys import argv
import os.path
import argparse

script, filename = argv


def word_count(filename):
    my_file = open(filename)
    counter = 0
    for x in my_file.read().split():
        counter += 1
    return counter
    my_file.close()

def line_count(filename):
    my_file = open(filename, 'r').read()
    return len(my_file.splitlines())
    my_file.close()

def character_count(filename):
    my_file = open(filename, 'r').read()
    return len(my_file)
    my_file.close()

parser = argparse.ArgumentParser()
parser.add_argument('--w', nargs='+', help='word help')
parser.add_argument('--l', nargs='+', help='line help')
parser.add_argument('--c', nargs='+', help='character help')
args = parser.parse_args()


if os.path.exists(filename):
    print word_count(filename), line_count(filename), character_count(filename)
else:
   print "There is no such file"

Answer 1

如果您使用 argparse 进行参数解析，您不应该尝试自己解析 argv：

script, filename = argv

如果 argv 少于两个元素，这将失败：

Traceback (most recent call last):
  File "wc.py", line 5, in <module>
    script, filename = argv
ValueError: need more than 1 value to unpack

或如果它有多于两个元素：

Traceback (most recent call last):
  File "wc.py", line 5, in <module>
    script, filename = argv
ValueError: too many values to unpack

您想使用 argparse 从参数列表中提取文件名：

parser.add_argument('filename')

您现有的命令行参数也可以进行一些修复。而不是这个：

parser.add_argument('--w', nargs='+', help='word help')

你想要：

parser.add_argument('-w', action='store_true', help='word help')

这为您提供了一个布尔选项，如果用户传递 -w，则 args.w 将为 True，否则为 None。这给你：

解析器 = argparse.ArgumentParser()

parser.add_argument('-w', action='store_true')
parser.add_argument('-c', action='store_true')
parser.add_argument('-l', action='store_true')
parser.add_argument('filename')
args = parser.parse_args()


if os.path.exists(args.filename):
    print word_count(args.filename), line_count(args.filename), character_count(args.filename)
else:
   print "There is no such file"

您可能还想为您的选项提供长等效值：

parser.add_argument('--words', '-w', action='store_true')
parser.add_argument('--characters', '-c', action='store_true')
parser.add_argument('--lines', '-l', action='store_true')

通过此更改，例如，用户可以使用 -w 或 --words。然后您将有 args.words、args.characters 和 args.lines（而不是 args.w、args.c、args.l）。

如何实现命令行切换到我的脚本？

How to implement command line switches to my script?

python

python-2.7

argparse