Используйте несколько вариантов вместе или не используйте их вообще

Я хочу использовать несколько вариантов вместе или не использовать вообще, как следует из названия, но мои методы кажутся относительно уродливыми, и мне было интересно, есть ли более чистый способ реализовать это. Кроме того, я просмотрел это о том, как это может быть сделано в argparse, но я хотел бы реализовать это в click, если это возможно (я стараюсь избегать использования nargs=[...]).

Пока это то, что у меня есть:

@click.group(invoke_without_command=True, no_args_is_help=True)
@click.option(
    "-d",
    "--group-dir",
    type=click.Path(),
    default="default",
    help='the directory to find the TOML file from which to run multiple jobs at the same time; defaults to the configuration directory of melmetal: "~/.melmetal" on Unix systems, and "C:\\Users\\user\\.melmetal" on Windows',
)
@click.option("-f", "--group-file", help="the TOML file name")
@click.option(
    "-n", "--group-name", help="name of the group of jobs"
)
@click.option(
    "--no-debug",
    is_flag=True,
    type=bool,
    help="prevent logging from being output to the terminal",
)
@click.pass_context
@logger.catch
def main(ctx, group_dir, group_file, group_name, no_debug):

    options = [group_file, group_name]
    group_dir = False if not any(options) else group_dir
    options.append(group_dir)

    if not any(options):
        pass
    elif not all(options):
        logger.error(
            colorize("red", "Sorry; you must use all options at once.")
        )
        exit(1)
    else:
        [...]

И второй пример:

if any(createStuff):
    if not all(createStuff):
        le(
            colorize("red", 'Sorry; you must use both the "--config-dir" and "--config-file" options at once.')
        )
        exit(1)
elif any(filtered):
    if len(filtered) is not len(drbc):
        le(
            colorize("red", 'Sorry; you must use all of "--device", "--repo-name", "--backup-type", and "--config-dir" at once.')
        )
        exit(1)
else:
    ctx = click.get_current_context()
    click.echo(ctx.get_help())
    exit(0)

Как сделать так, чтобы текст справки отображался, когда не заданы никакие подкоманды? Насколько я понимаю, это должно происходить автоматически, но для моего кода это автоматически переходит в основную функцию. Пример моего обходного пути находится во втором примере, то есть под оператором else.


person ShadowRylander    schedule 26.04.2019    source источник
comment
Вы должны спрашивать о своей проблеме, а не о своем решении, это поможет прояснить, чего вы пытаетесь достичь.   -  person Sayse    schedule 26.04.2019
comment
Ну, у меня нет проблем, скажем так; Я просто хочу знать, есть ли способ упростить код, например, с помощью аргумента ключевого слова в декораторе параметров, потому что на данный момент это довольно сложно понять, как это есть.   -  person ShadowRylander    schedule 26.04.2019
comment
Сделал бы варианты кортежем подойдет? То, как вы это описали, не похоже, что это действительно отдельные аргументы командной строки.   -  person Sayse    schedule 26.04.2019
comment
Но не сделает ли их раздельное использование более подробным? Хотя я полагаю, что это имело бы смысл, учитывая, что они находятся в той же категории, что и раньше... Это, однако, запутало бы, какой элемент в кортеже имеет какое значение.   -  person ShadowRylander    schedule 26.04.2019
comment
Это сделало бы его более подробным, но я не могу вспомнить ни одной командной строки, которую я использовал, когда аргумент зависел от другого. Но тогда, глядя на ваши примеры, вам действительно это нужно? Вы можете определить каталог из файла или наоборот   -  person Sayse    schedule 26.04.2019
comment
Это скорее проблема копирования и вставки; Я не хочу, чтобы пользователям приходилось постоянно заходить в каталог вместе с файлом; кроме того, есть и каталоги по умолчанию, которые проблематично учитывать при входе в файл.   -  person ShadowRylander    schedule 26.04.2019
comment
Итак, цель состоит в том, чтобы сгруппировать варианты?   -  person Stephen Rauch    schedule 27.04.2019
comment
В некоторой степени; сгруппировать параметры в не подкоманду, так как это сделало бы эту подкоманду подкомандой команды.   -  person ShadowRylander    schedule 27.04.2019


Ответы (2)


Вы можете принудительно использовать все параметры в группе, создав пользовательский класс, производный от click.Option, и в этом классе использовать метод click.Option.handle_parse_result(), например:

Пользовательский класс опций:

import click

class GroupedOptions(click.Option):
    def __init__(self, *args, **kwargs):
        self.opt_group = kwargs.pop('opt_group')
        assert self.opt_group, "'opt_group' parameter required"
        super(GroupedOptions, self).__init__(*args, **kwargs)

    def handle_parse_result(self, ctx, opts, args):
        if self.name in opts:
            opts_in_group = [param.name for param in ctx.command.params
                             if isinstance(param, GroupedOptions) and
                             param.opt_group == self.opt_group]

            missing_specified = tuple(name for name in opts_in_group
                                      if name not in opts)

            if missing_specified:
                raise click.UsageError(
                    "Illegal usage: When using option '{}' must also use "
                    "all of options {}".format(self.name, missing_specified)
                )

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

Использование пользовательского класса:

Чтобы использовать пользовательский класс, передайте параметр cls декоратору click.option, например:

@click.option('--opt1', cls=GroupedOptions, opt_group=1)

Кроме того, укажите номер группы опций с помощью параметра opt_group.

Как это работает?

Это работает, потому что click — это хорошо спроектированная объектно-ориентированная среда. Декоратор @click.option() обычно создает экземпляр объекта click.Option, но позволяет переопределить это поведение с помощью параметра cls. Таким образом, относительно легко наследоваться от click.Option в нашем собственном классе и переопределять нужные методы.

В этом случае мы перегружаем click.Option.handle_parse_result() и проверяем, что были указаны другие опции в нашей группе.

Примечание. Этот ответ был вдохновлен этим ответом

Тестовый код:

@click.command()
@click.option('--opt1', cls=GroupedOptions, opt_group=1)
@click.option('--opt2', cls=GroupedOptions, opt_group=1)
@click.option('--opt3', cls=GroupedOptions, opt_group=1)
@click.option('--opt4', cls=GroupedOptions, opt_group=2)
@click.option('--opt5', cls=GroupedOptions, opt_group=2)
def cli(**kwargs):
    for arg, value in kwargs.items():
        click.echo("{}: {}".format(arg, value))

if __name__ == "__main__":
    commands = (
        '--opt1=x',
        '--opt4=a',
        '--opt4=a --opt5=b',
        '--opt1=x --opt2=y --opt3=z --opt4=a --opt5=b',
        '--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)
            cli(cmd.split())

        except BaseException as exc:
            if str(exc) != '0' and \
                    not isinstance(exc, (click.ClickException, SystemExit)):
                raise

Полученные результаты:

Click Version: 6.7
Python Version: 3.6.3 (v3.6.3:2c5fed8, Oct  3 2017, 18:11:49) [MSC v.1900 64 bit (AMD64)]
-----------
> --opt1=x
Error: Illegal usage: When using option 'opt1' must also use all of options ('opt2', 'opt3')
-----------
> --opt4=a
Error: Illegal usage: When using option 'opt4' must also use all of options ('opt5',)
-----------
> --opt4=a --opt5=b
opt4: a
opt5: b
opt1: None
opt2: None
opt3: None
-----------
> --opt1=x --opt2=y --opt3=z --opt4=a --opt5=b
opt1: x
opt2: y
opt3: z
opt4: a
opt5: b
-----------
> --help
Usage: test.py [OPTIONS]

Options:
  --opt1 TEXT
  --opt2 TEXT
  --opt3 TEXT
  --opt4 TEXT
  --opt5 TEXT
  --help       Show this message and exit.
-----------
> 
opt1: None
opt2: None
opt3: None
opt4: None
opt5: None
person Stephen Rauch    schedule 27.04.2019
comment
Сердечно благодарю за это! Я проверю это как можно скорее; отдавая вам должное, могу ли я использовать это в своих программах? - person ShadowRylander; 28.04.2019
comment
Все в StackOverflow имеет лицензию CCASA. Вы можете использовать их по своему усмотрению, но если вы переиздаете, вы должны указать атрибуты. - person Stephen Rauch; 28.04.2019
comment
Еще раз спасибо! (В прошлый раз) Я хотел сначала проверить это; Я сделаю это и отмечу правильный ответ! - person ShadowRylander; 28.04.2019

Вы можете использовать Cloup. Он добавляет группы опций и позволяет определять ограничения для любой группы параметров. Он включает в себя all_or_none ограничения, которые делают именно то, что вы хотите.

Отказ от ответственности: я автор Cloup.

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


@command()
@option_group(
    'My peculiar options',
    option('--opt1'),
    option('--opt2'),
    option('--opt3'),
    constraint=all_or_none,
)
def cmd(**kwargs):
    print(kwargs)

Помощь:

Usage: cmd [OPTIONS]

My peculiar options [provide all or none]:
  --opt1 TEXT
  --opt2 TEXT
  --opt3 TEXT

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

Сообщение об ошибке:

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

Error: either all or none of the following parameters must be set:
--opt1, --opt2, --opt3

Это сообщение об ошибке, вероятно, не самое лучшее из возможных, поэтому я, вероятно, изменю его в следующих выпусках. Но вы можете легко сделать это самостоятельно в Cloup, например:

provide_all_or_none = all_or_none.rephrased(
    error='if you provide one of the following options, you need to provide '
          'all the others in the list:\n{param_list}'
)

Тогда вы получите:

Error: if you provide one of the following options, you need to provide all the others in the list:
--opt1, --opt2, --opt3
person janluke    schedule 10.02.2021
comment
Хотя я уже выбрал ответ, я обязательно проверю это; выглядит довольно интересно! - person ShadowRylander; 11.02.2021
comment
Рад это слышать! Забыл упомянуть, что я автор Cloup (добавлю). Не стесняйтесь давать мне отзывы, если вы проверите это (особенно если отрицательный!). - person janluke; 11.02.2021
comment
Сделаю! Еще раз спасибо за это! - person ShadowRylander; 11.02.2021