Question

我需要强制/检查移交给argparse的参数序列。我尝试了“尝试/除外”方法，但它似乎不起作用。

import argparse

epilog = '''usage example:'''

parser = argparse.ArgumentParser(description="add watchit hosts", epilog=epilog)
parser.add_argument('--add', nargs='*', default="-1", metavar="key 'value'")
args = vars(parser.parse_args())
try:
    args['add'][0] != 'name'
    args['add'][3] != 'desc'
    args['add'][5] != 'ip'
except:
    print "wrong argument sequence, read the help message!"

print args
print args['add'][1]
print args['add'][3]
print args['add'][5]

由于缺少名称，以下调用应引发错误。

root@lpgaixmgmtlx01:/root>python argparse_test.py --add desc 'aixbuildhost - buildhost - Test(Linz)' ip '172.17.14.37' nagiostmpl 'Enable without Notification' tsm 'AIXBUILDHOST' witnode 'lvgwatchit01t' loc 'loc_gru' devgrp 'hg_aix_lpar' persgrp 'cg_aix'

{'add': ['desc', 'aixbuildhost - buildhost - Test(Linz)', 'ip', '172.17.14.37', 'nagiostmpl', 'Enable without Notification', 'tsm', 'AIXBUILDHOST', 'witnode', 'lvgwatchit01t', 'loc', 'loc_gru', 'devgrp', 'hg_aix_lpar', 'persgrp', 'cg_aix']}
aixbuildhost - buildhost - Test(Linz)
172.17.14.37
Enable without Notification
...and so on

有什么聪明的方法可以强制参数序列吗？

Answer 1

try/except仅在try中的代码块返回错误时有效。对于您当前尝试应用的方法，最好使用if + sys.exit：

import sys
if args['add'][0] is not 'name' or args['add'][3] is not 'desc' or args['add'][5] is not 'ip':
    parser.error("Wrong argument sequence!")

示例：

import argparse
import sys

epilog = '''usage example:'''

parser = argparse.ArgumentParser(description="add watchit hosts", epilog=epilog)
parser.add_argument('--add', nargs='*', default="-1", metavar="key 'value'")
args = vars(parser.parse_args())

if args['add'][0] is not 'name' or args['add'][3] is not 'desc' or args['add'][5] is not 'ip':
    parser.error("Wrong argument sequence!")
    # Returns the usage, the error message, and exits (suggested by @hpaulj)

print args
print args['add'][1]
print args['add'][3]
print args['add'][5]

python argparse：强制参数序列

1 个答案: