argparse: combine current option value with positional argument

P

Peter Otten

I'd like to capture the current value of an option --prefix=<whatever> along
with a positional value as it is seen by argparse.

Example:
python script.py -p1 alpha beta -p2 gamma -p3

should result in a list

[(1, "alpha"), (1, "beta"), (2, "gamma")]

Here's a working script that uses --name=<some-value> instead of of just
<some-value>:

$ cat tmp.py
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-p", "--prefix")
parser.add_argument("-n", "--name")

class Namespace(object):
def __init__(self):
self.pairs = []
self.prefix = None
def set_name(self, value):
if value is not None:
self.pairs.append((self.prefix, value))
name = property(None, set_name)

ns = Namespace()
parser.parse_args(namespace=ns)
print ns.pairs
$ python tmp.py -p1 -nalpha -nbeta -p2 -ngamma
[('1', 'alpha'), ('1', 'beta'), ('2', 'gamma')]

However, modifying the --name option to a positional with

parser.add_argument("name", nargs="*")

results in an error:

$ python tmp2.py -p1 alpha beta -p2 gamma
usage: tmp2.py [-h] [-p PREFIX] [name [name ...]]
tmp2.py: error: unrecognized arguments: gamma

Am I missing a simple way to avoid that?

Peter

PS: I've not yet "used the source" ;)
 

Ask a Question

Want to reply to this thread or ask your own question?

You'll need to choose a username for the site, which only take a couple of moments. After that, you can post your question and our members will help you out.

Ask a Question

Members online

No members online now.

Forum statistics

Threads
473,769
Messages
2,569,580
Members
45,055
Latest member
SlimSparkKetoACVReview

Latest Threads

Top