39

如何在 Click 中创建互斥选项组？我想要么接受标志“--all”，要么选择带有“--color red”之类的参数的选项。

4

3 回答 3

42

我最近遇到了同样的用例；这就是我想出的。对于每个选项，您可以给出一个冲突选项列表。

from click import command, option, Option, UsageError


class MutuallyExclusiveOption(Option):
    def __init__(self, *args, **kwargs):
        self.mutually_exclusive = set(kwargs.pop('mutually_exclusive', []))
        help = kwargs.get('help', '')
        if self.mutually_exclusive:
            ex_str = ', '.join(self.mutually_exclusive)
            kwargs['help'] = help + (
                ' NOTE: This argument is mutually exclusive with '
                ' arguments: [' + ex_str + '].'
            )
        super(MutuallyExclusiveOption, self).__init__(*args, **kwargs)

    def handle_parse_result(self, ctx, opts, args):
        if self.mutually_exclusive.intersection(opts) and self.name in opts:
            raise UsageError(
                "Illegal usage: `{}` is mutually exclusive with "
                "arguments `{}`.".format(
                    self.name,
                    ', '.join(self.mutually_exclusive)
                )
            )

        return super(MutuallyExclusiveOption, self).handle_parse_result(
            ctx,
            opts,
            args
        )

然后使用常规option装饰器但传递cls参数：

@command(help="Run the command.")
@option('--jar-file', cls=MutuallyExclusiveOption,
        help="The jar file the topology lives in.",
        mutually_exclusive=["other_arg"])
@option('--other-arg',
        cls=MutuallyExclusiveOption,
        help="The jar file the topology lives in.",
        mutually_exclusive=["jar_file"])
def cli(jar_file, other_arg):
    print "Running cli."
    print "jar-file: {}".format(jar_file)
    print "other-arg: {}".format(other_arg)

if __name__ == '__main__':
    cli()

这是一个要点 ，其中包含上面的代码并显示了运行它的输出。

如果这对您不起作用，那么在单击 github 页面上还有一些（已关闭）问题提到了这一点，其中有一些您可以使用的想法。

于 2016-05-27T20:07:45.090 回答
14

您可以使用以下软件包： https ://github.com/espdev/click-option-group

import click
from click_option_group import optgroup, RequiredMutuallyExclusiveOptionGroup

@click.command()
@optgroup.group('Grouped options', cls=RequiredMutuallyExclusiveOptionGroup,
                help='Group description')
@optgroup.option('--all', 'all_', is_flag=True, default=False)
@optgroup.option('--color')
def cli(all_, color):
    print(all_, color)

if __name__ == '__main__':
    cli()

应用帮助：

$ app.py --help
Usage: app.py [OPTIONS]

Options:
  Grouped options: [mutually_exclusive, required]
                                  Group description
    --all
    --color TEXT
  --help                          Show this message and exit.
于 2019-12-05T16:28:48.853 回答
7

您可以使用Cloup，这是一个将选项组和约束添加到 Click 的包。在 Cloup 中，您有两种选择来解决这个问题。

免责声明：我是包的作者。

选项 1：@option_group

当您使用 定义选项组@option_group时，每个组中的选项显示在单独的帮助部分中（如在 argparse 中）。您可以将约束（如mutually_exclusive）应用于选项组，如下所示：

from cloup import command, option, option_group
from cloup.constraints import mutually_exclusive

@command()
@option_group(
    'Color options',
    option('--all', 'all_colors', is_flag=True),
    option('--color'),
    constraint=mutually_exclusive
)
def cmd(**kwargs):
    print(kwargs)

帮助将是：

Usage: cmd [OPTIONS]

Color options [mutually exclusive]:
  --all       
  --color TEXT

Other options:
  --help        Show this message and exit.

选项 2：应用约束而不定义选项组

如果您不希望选项组显示在命令帮助中，您可以使用@constraint并通过其（目标）名称指定受约束的选项：

from cloup import command, option
from cloup.constraints import constraint, mutually_exclusive

@command()
@option('--all', 'all_colors', is_flag=True)
@option('--color')
@constraint(mutually_exclusive, ['all_colors', 'color'])
def cmd(**kwargs):
    print(kwargs)

以这种方式定义的约束可以记录在命令帮助中！默认情况下禁用此功能，但可以轻松启用传递show_constraints=True给@command. 结果：

Usage: cmd [OPTIONS]

Options:
  --all       
  --color TEXT
  --help        Show this message and exit.

Constraints:
  {--all, --color}  mutually exclusive

更新：现在可以使用约束作为装饰器，而不是使用@contraint：

@command()
@mutually_exclusive(
    option('--all', 'all_colors', is_flag=True),
    option('--color'),
)
def cmd(**kwargs):
    print(kwargs)

错误信息

在这两种情况下，如果你运行cmd --all --color red，你会得到：

Usage: cmd [OPTIONS]
Try 'cmd --help' for help.

Error: the following parameters are mutually exclusive:
  --all 
  --color

其他约束

Cloup 定义了应满足 99.9% 需求的约束。它甚至支持条件约束！

例如，如果用户必须提供您的互斥选项之一，请在上面的示例中替换mutually_exclusive为RequireExactly(1)。

您可以在此处找到所有可用的约束。

于 2021-02-10T19:39:34.640 回答