Python - argparse要求或者

Question

運行腳本時，必須通過-g或-s。使用下面的代碼，它會向傳遞給它的參數拋出以下錯誤。

{~/nsnitro}-> ./sg-arg.py status -g test.server 
usage: sg-arg.py [-h] (-g SERVICEGROUP | -s SERVICENAME) {status} ... 
sg-arg.py: error: one of the arguments -g/--servicegroup -s/--servicename is required 

代碼：

parser = argparse.ArgumentParser() 
subparsers = parser.add_subparsers() 

check = subparsers.add_parser('status') 
check = parser.add_mutually_exclusive_group(required=True) 
check.add_argument('-g', '--servicegroup', action='store', help='servicegroup name', type=servicegroup_status) 
check.add_argument('-s', '--servicename', action='store', help='service name', type=servicegroup_status) 

args = parser.parse_args() 

Answer 1

您要添加的互斥組到錯誤的解析器。換句話說，對於您所撥打的電話，正確的呼叫將類似./sg-arg.py -g test.server status（請注意參數在之前的子分析程序聲明）。

要修復它，您需要將互斥組添加到子分析器並將參數添加到該組。這似乎工作：

import argparse 
parser = argparse.ArgumentParser() 
subparsers = parser.add_subparsers() 

check = subparsers.add_parser('status') 
check_mutex = check.add_mutually_exclusive_group(required=True) 
check_mutex.add_argument('-g', '--servicegroup', action='store', help='servicegroup name') 
check_mutex.add_argument('-s', '--servicename', action='store', help='service name') 

args = parser.parse_args() 

例如，

$ python foo.py status -g test.server 
$ python foo.py status -s test.server 
$ python foo.py status test.server 
usage: foo.py status [-h] (-g SERVICEGROUP | -s SERVICENAME) 
foo.py status: error: one of the arguments -g/--servicegroup -s/--servicename is required