У меня есть необязательный аргумент, который поддерживает список самих аргументов.
Я имею в виду, он должен поддерживать:
но не:
Есть ли способ заставить это внутри argparse? Теперь я использую nargs = "*", а затем проверяю длину списка.
Изменить: В соответствии с запросом мне нужно определить диапазон допустимого количества аргументов. Я имею в виду, говоря (в примере) 2 или 3 аргумента правильно, но не 1 или 4 или ничего, что не внутри диапазона 2..3
Вы можете сделать это с помощью пользовательского действия:
import argparse
def required_length(nmin,nmax):
class RequiredLength(argparse.Action):
def __call__(self, parser, args, values, option_string=None):
if not nmin<=len(values)<=nmax:
msg='argument "{f}" requires between {nmin} and {nmax} arguments'.format(
f=self.dest,nmin=nmin,nmax=nmax)
raise argparse.ArgumentTypeError(msg)
setattr(args, self.dest, values)
return RequiredLength
parser=argparse.ArgumentParser(prog='PROG')
parser.add_argument('-f', nargs='+', action=required_length(2,3))
args=parser.parse_args('-f 1 2 3'.split())
print(args.f)
# ['1', '2', '3']
try:
args=parser.parse_args('-f 1 2 3 4'.split())
print(args)
except argparse.ArgumentTypeError as err:
print(err)
# argument "f" requires between 2 and 3 arguments