7

我正在尝试指定一个采用标准输入的可选参数。这将主要用于我程序中的管道数据，所以someprog that outputs | python my_prog.

我遵循了 argparse 文档，并在 Stackoverflow 上阅读了很多关于此的问题/答案，但似乎没有一个对我有用。

这是我最初拥有的：

parser = argparse.ArgumentParser(description='Upgrade Instance.')
parser.add_argument('--app', '-a', dest='app', action='store', required=True)
parser.add_argument('--version', '-v', dest='version', action='store', default='', required=False)
parser.add_argument('--config', '-c', dest='config', action='store', default = '', required=False)
args = parser.parse_args()

现在我想要做的是允许用户version使用管道传递，而不是传递它。

我添加parser.add_argument('infile', nargs='?', type=argparse.FileType('r'), default=sys.stdin)到顶部，但这使它成为一个位置参数。这怎么可能？我认为nargs=?使它成为可选的。

我需要它是一个可选参数。所以我把它改成：

parser.add_argument('--infile', nargs='?', type=argparse.FileType('r'), default=sys.stdin)

这使它成为一个可选参数，但如果没有传递管道，程序会挂起等待 stdin 作为默认值。在我的程序中删除default=sys.stdin和管道的东西我得到：

close failed in file object destructor:
sys.excepthook is missing
lost sys.stderr

运行时。当我打印args时，我得到：Namespace(app='app', config='', g=False, hosts='03.app', infile=None, version='')。

看来我正在做的事情很简单，很常见，很多人都问过。但它似乎不适合我。

关于如何让它工作的任何建议？

4

3 回答 3

5

这样做......没有指定参数。如果你将管道输入传递给它运行的程序，你不这样做，它仍然会运行。raw_input()也会起作用。

import sys

if not sys.stdin.isatty():
    stdin = sys.stdin.readlines()
    print stdin
    sys.stdin = open('/dev/tty')
else:
    print "No stdin"

test_raw = raw_input()
print test_raw

演示 -

rypeck$ echo "stdin input" | python test_argparse.py -a test 
['stdin input\n']
raw_input working!
raw_input working!
rypeck$ python test_argparse.py -a test
No stdin
raw_input working!
raw_input working!
于 2013-11-14T04:35:28.030 回答
4

我自己也在探讨这个问题，发现这里的选项有一个小的改进——希望它能帮助未来的旅行者。如果您希望在提供标志时仅从标准输入读取，这不是 100% 清楚，但似乎这可能是您的目标；我的回答是基于这个假设。我在 3.4 工作，以防万一成为问题......

我可以声明一个可选参数，例如：

parser.add_argument("-v", "--version", nargs='?', const=sys.stdin, action=StreamType)

我没有指定默认值，因为它仅在选项完全不存在时使用，而 const 仅在标志存在但没有参数时使用。StreamType是我写的一个很小的argparse.Action子类，它只是试图从标准输入流中读取一行，如果这不起作用就保存原始值：

class StreamType(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        try:
            setattr(namespace, self.dest, values.readline().strip())
        except AttributeError:
            setattr(namespace, self.dest, values)

这应该产生类似的东西：

$ blah --version v1.1.1
Namespace(version='v1.1.1')

$ echo "v1.0.3" | blah --version
Namespace(version='v1.0.3')
于 2015-04-19T19:59:21.957 回答
3

你做什么args.infile？由于您获得了命名空间，argparse因此不是挂起或给出错误的部分。 

p = argparse.ArgumentParser()
p.add_argument('--infile', type=argparse.FileType('r'),default='-')
# p.add_argument('infile', nargs='?', type=argparse.FileType('r'),default='-') # positional version
args = p.parse_args()
print(args)
print args.infile.read()
-----------
$ cat myprog.py | python myprog.py --infile -
$ cat myprog.py | python myprog.py
$ python myprog.py myprog.py  # with the positional version
$ python myprog.py - < myprog.py  # with the positional version

很好地呼应了代码。第二个调用也适用于“可选位置”。

不幸的是，术语optional/positional和optional/required.

如果位置参数（是的，“位置”的另一种用法）具有前缀字符，例如-，或者--它被称为optional. 默认情况下，它的required参数是False，但您可以将其设置为True。但是如果参数是'infile'（没有前缀），它是positional，即使?是是optional（不是必需的）。

顺便说一句，默认action是'store'，所以你不需要指定。你也不需要指定required，除非它是True。

使用 a FileType，一种方便的指定方式stdin是-。

除非你真的想要一个，否则不要使用'?'with--infileNone

于 2013-11-14T02:37:23.997 回答