タイトルがほぼすべてを物語っています。 nargs が 1 より大きい場合、解析される個々の引数に制限 (choice/type など) を設定する方法はありますか?
これはサンプルコードです:
parser = argparse.ArgumentParser()
parser.add_argument('-c', '--credits', nargs=2,
help='number of credits required for a subject')
-c 引数には、科目と必要な単位数を指定する必要があります。科目は定義済みの科目リストに限定し、必要な単位数は浮動小数点数にする必要があります。
おそらくサブパーサーを使用してこれを行うこともできますが、現状ではこれはすでにサブコマンドの一部であるため、これ以上複雑にしたくありません。
ベストアンサー1
それを検証するにはカスタムアクション:
import argparse
import collections
class ValidateCredits(argparse.Action):
def __call__(self, parser, args, values, option_string=None):
# print '{n} {v} {o}'.format(n=args, v=values, o=option_string)
valid_subjects = ('foo', 'bar')
subject, credits = values
if subject not in valid_subjects:
raise ValueError('invalid subject {s!r}'.format(s=subject))
credits = float(credits)
Credits = collections.namedtuple('Credits', 'subject required')
setattr(args, self.dest, Credits(subject, credits))
parser = argparse.ArgumentParser()
parser.add_argument('-c', '--credits', nargs=2, action=ValidateCredits,
help='subject followed by number of credits required',
metavar=('SUBJECT', 'CREDITS')
)
args = parser.parse_args()
print(args)
print(args.credits.subject)
print(args.credits.required)
例えば、
% test.py -c foo 2
Namespace(credits=Credits(subject='foo', required=2.0))
foo
2.0
% test.py -c baz 2
ValueError: invalid subject 'baz'
% test.py -c foo bar
ValueError: could not convert string to float: bar