OR在Python的命令行中使用带有两个变量的argparse函数

Question

我有一个程序接受一个文件作为输入,有些工作与文件的内容,并将其推送到服务器.我想添加一个可选的命令行开关来指定程序的"干运行",从而执行所有文件处理,但跳过写入服务器的操作.我使用argparse引入命令行参数,但我没有看到在参数之间执行"OR"函数的方法.这是我或多或少要寻找的......

允许的选项:

1)prog.py输入文件服务器名称

2)prog.py输入文件-d

3)prog.py inputfile -d servername

Dissallowed:

1)prog.py输入文件

我想确保命令行上的服务器名称"OR"是干运行标志.并且,如果两者都在那里......那也没关系.(因此是OR而不是异或).如果我使用required = true互斥,我可以获得异或; 但是,我似乎无法弄清楚如何做到这两个可以存在的"或".更复杂的是,服务器名称是位置参数,而干运行标志是可选参数,可以在命令行的任何位置.有没有人知道如何解决这个问题？

Answer 1

这就是我要做的事情:

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-d', '--dry-run', action='store_true')
parser.add_argument('input_file', type=argparse.FileType('r'))
parser.add_argument('servername', nargs='?')

args = parser.parse_args()
if args.servername is None and not args.dry_run:
    parser.error("Option 'servername' is required when not in dry-run mode.")

print args

例子:

$ ./prog.py inputfile servername
Namespace(dry_run=False, input_file=<open file 'inputfile', mode 'r' at 0x283440>, servername='servername')
$ ./prog.py inputfile -d
Namespace(dry_run=True, input_file=<open file 'inputfile', mode 'r' at 0x2cf440>, servername=None)
$ ./prog.py -d inputfile servername
Namespace(dry_run=True, input_file=<open file 'inputfile', mode 'r' at 0x1f4440>, servername='servername')
$ ./prog.py inputfile
usage: prog.py [-h] [-d] input_file [servername]
prog.py: error: Option 'servername' is required when not in dry-run mode.

您也可以使用自定义操作执行此操作,它具有相同的效果:

class ServernameAction(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        if values is None and not namespace.dry_run:
            parser.error("Option 'servername' is required when not in dry-run mode.")
        setattr(namespace, self.dest, values)

...
parser.add_argument('servername', nargs='?', action=ServernameAction)
...