Python argparse: Existe-t-il un moyen de spécifier une plage dans nargs?

Question

J'ai un argument optionnel qui supporte une liste d'arguments elle-même.

Je veux dire, il devrait soutenir:

• -f 1 2
• -f 1 2 3

mais pas:

• -f 1
• -f 1 2 3 4

Existe-t-il un moyen de forcer cela dans argparse? Maintenant j'utilise nargs = "*", puis je vérifie la longueur de la liste.

Modifier: Comme demandé, ce dont j'avais besoin est de pouvoir définir une plage de nombre acceptable d'arguments. Je veux dire, en disant (dans l'exemple) 2 ou 3 args est droite, mais pas 1 ou 4 ou tout ce qui est pas dans la gamme 2..3

Answer 1

Vous pouvez faire cela avec un custom action:

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