Python Argparse: последовательность аргументов силы - PullRequest
Новая
       6

Python Argparse: последовательность аргументов силы

0 голосов
/ 04 июля 2018

Мне нужно форсировать / проверять последовательность аргументов, переданную 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

Есть ли какое-нибудь умное решение для принудительной последовательности аргументов?

1 Ответ

0 голосов
/ 04 июля 2018

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]
...