Python 点击多个群组名称

Question

是否可以用 Python Click 做这样的事情？我想对同一个 click.Group.

使用不同的名称

import click

class CustomMultiGroup(click.Group):

    def group(self, *args, **kwargs):
        """Behaves the same as `click.Group.group()` except if passed
        a list of names, all after the first will be aliases for the first.
        """
        def decorator(f):
            if isinstance(args[0], list):
                _args = [args[0][0]] + list(args[1:])
                for alias in args[0][1:]:
                    cmd = super(CustomMultiCommand, self).group(
                        alias, *args[1:], **kwargs)(f)
                    cmd.short_help = "Alias for '{}'".format(_args[0])
            else:
                _args = args
            cmd = super(CustomMultiCommand, self).group(
                *_args, **kwargs)(f)
            return cmd

        return decorator

@click.group(cls=CustomMultiGroup)
def mycli():
    pass

@cli.group(['my-group', 'my-grp'])
def my_group():
    pass

@my_group.command()
def my_command():
    pass

我希望我的命令行是这样的：

mycli my-group my-command

和

mycli my-grp my-command

但引用相同的函数。

此 post 是对

的引用

Answer 1

click.Group 和 click.Command 的行为不同，因此您需要修改以允许别名组访问别名组的命令：

自定义Class

此 class 覆盖了用于修饰组函数的 click.Group.group() 方法。它增加了传递组别名列表的能力。此 class 还添加了引用别名组的简短帮助。

import click

class CustomMultiGroup(click.Group):

    def group(self, *args, **kwargs):
        """Behaves the same as `click.Group.group()` except if passed
        a list of names, all after the first will be aliases for the first.
        """
        def decorator(f):
            aliased_group = []
            if isinstance(args[0], list):
                # we have a list so create group aliases
                _args = [args[0][0]] + list(args[1:])
                for alias in args[0][1:]:
                    grp = super(CustomMultiGroup, self).group(
                        alias, *args[1:], **kwargs)(f)
                    grp.short_help = "Alias for '{}'".format(_args[0])
                    aliased_group.append(grp)
            else:
                _args = args

            # create the main group
            grp = super(CustomMultiGroup, self).group(*_args, **kwargs)(f)

            # for all of the aliased groups, share the main group commands
            for aliased in aliased_group:
                aliased.commands = grp.commands

            return grp

        return decorator

测试代码：

@click.group(cls=CustomMultiGroup)
def cli():
    pass

@cli.group(['my-group', 'my-grp'])
def my_group():
    """ My Sub Command """
    pass

@my_group.command('my-command')
def my_command():
    click.echo("My Command")

cli('--help'.split())
cli('my-grp --help'.split())
cli('my-group --help'.split())
cli('my-grp my-command'.split())
cli('my-group my-command'.split())

测试结果：

Usage: my_cli [OPTIONS] COMMAND [ARGS]...

Options:
  --help  Show this message and exit.

Commands:
  my-group  My Sub Command
  my-grp    Alias for 'my-group'

Process finished with exit code 0

Answer 2

在 Click 中使用命令别名的最简单方法是使用以下软件包之一：

click-aliases
Cloup, which also adds several other features other than aliases.

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

这两个实现略有不同。

Cloup 将命令的别名存储在命令本身中。这允许 Cloup 在命令本身的 --help 中记录命令的别名（也）。
click-aliases 始终在组的“命令”部分显示（子）命令的别名。在 Cloup 中，您可以根据自己的喜好 enable/disable 此行为（默认情况下，别名不显示）。

Python 点击多个群组名称

Python Click multiple group names

python

command-line-interface

alias

python-click

自定义Class

测试代码：

测试结果：