CODEWITHSUNDEEP
pythonpython-3.xargparse
tsorn
tsorn

Reputation: 3625

Argparse optional argument with different default if specified without a value

Consider three different runs of a program:

python3 prog.py
python3 prog.py --x
python3 prog.py --x 2

Is it possible to use argparse such that, for example, in the first case, x==None, in the second case, x==1, and in the third, x==2?

Upvotes: 2

Views: 833

Answers (2)

wim
wim

Reputation: 363596

You may want to consider instead the count action, which is commonly used to specify verbosity levels. The usage is different, for example:

python3 prog.py
python3 prog.py -v
python3 prog.py -vv

Upvotes: 2

hpaulj
hpaulj

Reputation: 231738

nargs'?' with a const parameter handles this 3-way input nicely..

In [2]: parser = argparse.ArgumentParser()
In [3]: parser.add_argument('-x','--x', nargs='?', type=int, const=1)
...
In [4]: parser.parse_args([])
Out[4]: Namespace(x=None)
In [5]: parser.parse_args(['-x'])
Out[5]: Namespace(x=1)
In [6]: parser.parse_args(['-x','2'])
Out[6]: Namespace(x=2)

I could have also given it a default parameter.

how to add multiple argument options in python using argparse?

Upvotes: 4

Related Questions