Флаг функции для команд щелчка Python

В моем скрипте python-click-CLI я создаю некоторые команды для функций, которые не должны быть видны пользователям (чтобы не путать их), но видны, например. Разработчики.

Можно ли использовать флаги функций для команд Python-щелчка?

Я хотел бы иметь возможность настроить (через файл конфигурации и т. д.), если команда доступна или нет. Если функция-команда отключена, команду нельзя вызывать, и справка не должна ее показывать.

Как это:

FLAG_ENABLED = False

# This command should not be shown and not be callable as long as the flag is disabled
@cli.command(name='specialfeature', active=FLAG_ENABLED)
def special_feature_command()
    ....

Очевидно, я мог бы изменить тело своей функции:

@cli.command(name='specialfeature', active=FLAG_ENABLED)
def special_feature_command()
    if FLAG_ENABLED:
        ...
    else:
        ...

Но тогда моя команда все равно будет отображаться в справке, чего я хотел бы избежать.

python python-click

Wolkenarchitekt 27.03.2019 источник

comment

Я не могу найти активную в документах, я нашел скрытую: скрытую — скрыть эту команду из выходных данных справки.click.palletsprojects.com/en/7.x/api/#click.Command - geckos 27.03.2019

Ответы (1)

arrow_upward
1
arrow_downward

Вы можете добавить функцию флаг функции с помощью пользовательского класса, например:

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

Этот класс переопределяет метод click.Group.command(), который используется для оформления командных функций. Он добавляет возможность передавать флаг active, который при False пропускает добавление команды в группу.

import click

class FeatureFlagCommand(click.Group):

    def command(self, *args, active=True, **kwargs):
        """Behaves the same as `click.Group.command()` except added an
        `active` flag which can be used to disable to command.
        """
        if active:
            return super(FeatureFlagCommand, self).command(*args, **kwargs)
        else:
            return lambda f: f

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

Передав параметр cls декоратору click.group(), любые команды, добавленные в группу через group.command(), будут закрыты флагом active.

@click.group(cls=FeatureFlagCommand)
def cli():
    ...

@cli.command(name='specialfeature', active=FLAG_ENABLED)
def special_feature_command()
    ...

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

Это работает, потому что click — это хорошо спроектированная объектно-ориентированная среда. Легко наследоваться от click.Group и создать новый декоратор command(). В новом декораторе command(), если активен флаг False, мы возвращаем необработанную функцию вместо добавления функции в группу.

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

@click.group(cls=FeatureFlagCommand)
def cli():
    """My Awesome Click Program"""

@cli.command(active=False)
def command1():
    click.echo('Command 1')

@cli.command(active=True)
def command2():
    click.echo('Command 2')

@cli.command()
def command3():
    click.echo('Command 3')


if __name__ == "__main__":
    commands = (
        'command1',
        'command2',
        'command3',
        '--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)]
-----------
> command1
Usage: test.py [OPTIONS] COMMAND [ARGS]...

Error: No such command "command1".
-----------
> command2
Command 2
-----------
> command3
Command 3
-----------
> --help
Usage: test.py [OPTIONS] COMMAND [ARGS]...

  My Awesome Click Program

Options:
  --help  Show this message and exit.

Commands:
  command2
  command3
-----------
> 
Usage: test.py [OPTIONS] COMMAND [ARGS]...

  My Awesome Click Program

Options:
  --help  Show this message and exit.

Commands:
  command2
  command3

Stephen Rauch 27.03.2019