Использование click.MultiCommand с методами класса

Если вы не чувствуете себя особенно амбициозным (или, возможно, мазохистом), вы, вероятно, захотите использовать библиотеку, например DevIL , которая уже поддерживает это. Текущая версия может загружать и / или сохранять в форматах PNG и BMP вместе с несколькими десятками других.

По сравнению с чем-то вроде IJG, это гораздо больше ориентировано на работу с OpenGL или DirectX (например, , он может загружать файл достаточно прямо в текстуру или наоборот).

1
задан Stephen Rauch 26 April 2019 в 05:03
поделиться

1 ответ

Чтобы использовать classmethod в качестве команды щелчка, необходимо иметь возможность заполнить параметр cls при вызове команды. Это можно сделать с помощью пользовательского класса click.Command, например:

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

import click

class ClsMethodClickCommand(click.Command):
    def __init__(self, *args, **kwargs):
        self._cls = [None]
        super(ClsMethodClickCommand, self).__init__(*args, **kwargs)

    def main(self, *args, **kwargs):
        self._cls[0] = args[0]
        return super(ClsMethodClickCommand, self).main(*args[1:], **kwargs)

    def invoke(self, ctx):
        ctx.params['cls'] = self._cls[0]
        return super(ClsMethodClickCommand, self).invoke(ctx)

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

class MyClassWithAClickCommand:

    @classmethod
    @click.command(cls=ClsMethodClickCommand)
    ....
    def cli(cls, ....):
        ....

И затем в [ 118] класс, вам нужно заполнить атрибут _cls, так как command.main не вызывается в этом случае:

def get_command(self, ctx, name):
    # this is hard coded in this example but presumably
    #   would be done with a lookup via name
    cmd = MyClassWithAClickCommand.cli

    # Tell the click command which class it is associated with
    cmd._cls[0] = MyClassWithAClickCommand
    return cmd

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

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

В этом случае мы переопределяем click.Command.invoke() и затем добавляем содержащий класс к диктовке ctx.params как cls перед вызовом обработчика команды.

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

class MyClassWithAClickCommand:

    @classmethod
    @click.command(cls=ClsMethodClickCommand)
    @click.argument('arg')
    def cli(cls, arg):
        click.echo('cls: {}'.format(cls.__name__))
        click.echo('cli: {}'.format(arg))


class ConversionCLI(click.MultiCommand):
    def list_commands(self, ctx):
        return ['converter_x']

    def get_command(self, ctx, name):
        cmd = MyClassWithAClickCommand.cli
        cmd._cls[0] = MyClassWithAClickCommand
        return cmd


@click.command(cls=ConversionCLI)
def convert():
    """Convert files using specified converter"""



if __name__ == "__main__":
    commands = (
        'converter_x an_arg',
        'converter_x --help',
        'converter_x',
        '--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)
            convert(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)]
-----------
> converter_x an_arg
class: MyClassWithAClickCommand
cli: an_arg
-----------
> converter_x --help
Usage: test.py converter_x [OPTIONS] ARG

Options:
  --help  Show this message and exit.
-----------
> converter_x
Usage: test.py converter_x [OPTIONS] ARG

Error: Missing argument "arg".
-----------
> --help
Usage: test.py [OPTIONS] COMMAND [ARGS]...

  Convert files using specified converter

Options:
  --help  Show this message and exit.

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

  Convert files using specified converter

Options:
  --help  Show this message and exit.

Commands:
  converter_x
0
ответ дан Stephen Rauch 26 April 2019 в 05:03
поделиться
Другие вопросы по тегам:

Похожие вопросы: