nargs=* equivalent for options in Click nargs=* equivalent for options in Click python python

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.