nargs=* equivalent for options in Click
One way to approach what you are after is to inherit from click.Option, and customize the parser.
Custom Class:
import clickclass OptionEatAll(click.Option): def __init__(self, *args, **kwargs): self.save_other_options = kwargs.pop('save_other_options', True) nargs = kwargs.pop('nargs', -1) assert nargs == -1, 'nargs, if set, must be -1 not {}'.format(nargs) super(OptionEatAll, self).__init__(*args, **kwargs) self._previous_parser_process = None self._eat_all_parser = None def add_to_parser(self, parser, ctx): def parser_process(value, state): # method to hook to the parser.process done = False value = [value] if self.save_other_options: # grab everything up to the next option while state.rargs and not done: for prefix in self._eat_all_parser.prefixes: if state.rargs[0].startswith(prefix): done = True if not done: value.append(state.rargs.pop(0)) else: # grab everything remaining value += state.rargs state.rargs[:] = [] value = tuple(value) # call the actual process self._previous_parser_process(value, state) retval = super(OptionEatAll, self).add_to_parser(parser, ctx) for name in self.opts: our_parser = parser._long_opt.get(name) or parser._short_opt.get(name) if our_parser: self._eat_all_parser = our_parser self._previous_parser_process = our_parser.process our_parser.process = parser_process break return retval
Using Custom Class:
To use the custom class, pass the cls
parameter to @click.option()
decorator like:
@click.option("--an_option", cls=OptionEatAll)
or if it is desired that the option will eat the entire rest of the command line, not respecting other options:
@click.option("--an_option", cls=OptionEatAll, save_other_options=False)
How does this work?
This works because click is a well designed OO framework. The @click.option()
decorator usually instantiates aclick.Option
object but allows this behavior to be over ridden with the cls parameter. So it is a relativelyeasy matter to inherit from click.Option
in our own class and over ride the desired methods.
In this case we over ride click.Option.add_to_parser()
and the monkey patch the parser so that we caneat more than one token if desired.
Test Code:
@click.command()@click.option('-g', 'greedy', cls=OptionEatAll, save_other_options=False)@click.option('--polite', cls=OptionEatAll)@click.option('--other')def foo(polite, greedy, other): click.echo('greedy: {}'.format(greedy)) click.echo('polite: {}'.format(polite)) click.echo('other: {}'.format(other))if __name__ == "__main__": commands = ( '-g a b --polite x', '-g a --polite x y --other o', '--polite x y --other o', '--polite x -g a b c --other o', '--polite x --other o -g a b c', '-g a b c', '-g a', '-g', 'extra', '--help', ) import sys, time time.sleep(1) print('Click Version: {}'.format(click.__version__)) print('Python Version: {}'.format(sys.version)) for cmd in commands: try: time.sleep(0.1) print('-----------') print('> ' + cmd) time.sleep(0.1) foo(cmd.split()) except BaseException as exc: if str(exc) != '0' and \ not isinstance(exc, (click.ClickException, SystemExit)): raise
Test Results:
Click Version: 6.7Python Version: 3.6.3 (v3.6.3:2c5fed8, Oct 3 2017, 18:11:49) [MSC v.1900 64 bit (AMD64)]-----------> -g a b --polite xgreedy: ('a', 'b', '--polite', 'x')polite: Noneother: None-----------> -g a --polite x y --other ogreedy: ('a', '--polite', 'x', 'y', '--other', 'o')polite: Noneother: None-----------> --polite x y --other ogreedy: Nonepolite: ('x', 'y')other: o-----------> --polite x -g a b c --other ogreedy: ('a', 'b', 'c', '--other', 'o')polite: ('x',)other: None-----------> --polite x --other o -g a b cgreedy: ('a', 'b', 'c')polite: ('x',)other: o-----------> -g a b cgreedy: ('a', 'b', 'c')polite: Noneother: None-----------> -g agreedy: ('a',)polite: Noneother: None-----------> -gError: -g option requires an argument-----------> extraUsage: test.py [OPTIONS]Error: Got unexpected extra argument (extra)-----------> --helpUsage: test.py [OPTIONS]Options: -g TEXT --polite TEXT --other TEXT --help Show this message and exit.
You can use this trick.
import click@click.command()@click.option('--users', nargs=0, required=True)@click.argument('users', nargs=-1)@click.option('--bar')def fancy_command(users, bar): users_str = ', '.join(users) print('Users: {}. Bar: {}'.format(users_str, bar))if __name__ == '__main__': fancy_command()
Add fake option
with a needed name and none arguments nargs=0
, then add 'argument' with the unlimited args nargs=-1
.
$ python foo --users alice bob charlie --bar bazUsers: alice, bob, charlie. Bar: baz
But be careful with the further options:
$ python foo --users alice bob charlie --bar baz fazUsers: alice, bob, charlie, faz. Bar: baz
I ran into the same issue. Instead of implementing a single command line option with n number of arguments, I decided to use multiple of the same command line option and just letting Click make a tuple out of the arguments under the hood. I ultimately figured if Click didn't support it, that decision was probably made for a good reason.
https://click.palletsprojects.com/en/7.x/options/#multiple-options
here is an example of what I am saying:
instead of passing a single string argument a splitting on a delimiter:
commit -m foo:bar:baz
I opted to use this:
commit -m foo -m bar -m baz
here is the source code:
@click.command()@click.option('--message', '-m', multiple=True)def commit(message): click.echo('\n'.join(message))
This is more to type, but I do think it makes the CLI more user friendly and robust.