python argparser for multiple arguments for partial choices

Question

I create a argparser like this:

  parser = argparse.ArgumentParser(description='someDesc')
  parser.add_argument(-a,required=true,choices=[x,y,z])
  parser.add_argument( ... )

However, only for choice "x" and not for choices "y,z", I want to have an additional REQUIRED argument. For eg.

python test -a x       // not fine...needs additional MANDATORY argument b
python test -a y       // fine...will run
python test -a z       // fine...will run  
python test -a x -b "ccc"       // fine...will run

How can I accomplish that with ArgumentParser? I know its possible with bash optparser

hpaulj · Accepted Answer · 2013-08-01 02:37:31Z

5

To elaborate on the subparser approach:

sp = parser.add_subparsers(dest='a')
x = sp.add_parser('x')
y=sp.add_parser('y')
z=sp.add_parser('z')
x.add_argument('-b', required=True)

this differs from your specification in that no -a is required.
dest='a' argument ensures that there is an 'a' attribute in the namespace.
normally an optional like '-b' is not required. Subparser 'x' could also take a required positional.

If you must use a -a optional, a two step parsing might work

p1 = argparse.ArgumentParser()
p1.add_argument('-a',choices=['x','y','z'])
p2 = argparse.ArgumentParser()
p2.add_argument('-b',required=True)
ns, rest = p1.parse_known_args()
if ns.a == 'x':
    p2.parse_args(rest, ns)

A third approach is to do your own test after the fact. You could still use the argparse error mechanism

parser.error('-b required with -a x')

edited Aug 1, 2013 at 2:37

answered Jul 31, 2013 at 22:22

hpaulj

233k14 gold badges260 silver badges392 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Udy · Accepted Answer · 2013-07-29 06:32:20Z

2

ArgumentParser supports the creation of such sub-commands with the add_subparsers() method. The add_subparsers() method is normally called with no arguments and returns a special action object. This object has a single method, add_parser(), which takes a command name and any ArgumentParser constructor arguments, and returns an ArgumentParser object that can be modified as usual.

ArgumentParser.add_subparsers()

answered Jul 29, 2013 at 6:32

Udy

2,5424 gold badges26 silver badges34 bronze badges

Collectives™ on Stack Overflow

python argparser for multiple arguments for partial choices

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related