Question

我想自定义我的Python Click CLI程序的BASH完成功能，以不仅包括脚本的命令/子命令，还包括脚本创建的对象。

假设我的程序称为cli-tool，它可以使用以下命令创建对象foo：

cli-tool object create foo

为简单起见，假设命令仅将参数字符串（在这种情况下为foo连接到位于~/.cli-tool/objects/foo中的同名文本文件。然后执行cat ~/.cli-tool/objects/foo会在您的终端上打印foo。

该工具要执行的操作是当我键入以下内容：

cli-tool object get <TAB><TAB>

然后终端将列出foo和~/.cli-tool/objects内部的其他文件。

为便于记录，我已阅读Python Click 6.x stable documentation，其中明确指出：

当前，Bash完成是一项内部功能，并非可定制的。在将来的版本中，这可能会放宽。

我希望有一种方法可以从以下命令中提取 full BASH完成脚本：

eval "$(_CLI_TOOL_COMPLETE=source cli-tool)"

然后自己定制。我还看过click-completion项目，但是除了扩展Zsh和Fish shell的完成功能之外，我不确定它的作用。

有人实现了我上面提到的完成类型吗？

Answer 1

使用click-completion，这很简单。

代码：

导入并init()单击完成：

import click
import click_completion

click_completion.init()

然后实例化一个click.Choice对象：

option_type = click.Choice('obj1 obj2 obj3'.split())

对于您的选项目录，请传递适当项目的列表，而不要输入示例 obj1-obj3 。

然后将选项类型传递给click.argument()装饰器，例如：

@click.argument('option', type=option_type)

别忘了用shell激活完成。 bash的点击变化为here：

测试代码：

import click
import click_completion

click_completion.init()

option_type = click.Choice('obj1 obj2 obj3'.split())

@click.group()
def cli():
    """My Cool Tool"""

@cli.group(name='object')
def object_group():
    """Object subcommand"""

@object_group.command()
@click.argument('option', type=option_type)
def get(option):
    click.echo('option: {}'.format(option))


commands = (
    ('"" object get ""', 1),
    ('"" object get ""', 2),
    ('"" object get ""', 3),
    'object get obj1',
    '--help',
    'object --help',
    'object get --help',
)

os.environ['BASH_COMP'] = 'complete'

import sys, time
time.sleep(1)
print('Click Version: {}'.format(click.__version__))
print('Click Completion Version: {}'.format(click_completion.__version__))
print('Python Version: {}'.format(sys.version))
for cmd in commands:
    try:
        time.sleep(0.1)
        print('\n-----------')
        print('> ' + str(cmd))
        time.sleep(0.1)

        if len(cmd) == 2:
            os.environ['COMP_WORDS'] = cmd[0]
            os.environ['COMP_CWORD'] = str(cmd[1])
            cli(complete_var='BASH_COMP')
        else:
            try:
                del os.environ['COMP_WORDS']
                del os.environ['COMP_CWORD']
            except:
                pass
            cli(cmd.split())


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

结果：

Click Version: 6.7
Click Completion Version: 0.4.1
Python Version: 3.6.3 (v3.6.3:2c5fed8, Oct  3 2017, 18:11:49) [MSC v.1900 64 bit (AMD64)]

-----------
> ('"" object get ""', 1)
object
-----------
> ('"" object get ""', 2)
get
-----------
> ('"" object get ""', 3)
obj1    obj2    obj3
-----------
> object get obj1
option: obj1

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

  My Cool Tool

Options:
  --help  Show this message and exit.

Commands:
  object  Object subcommand

-----------
> object --help
Usage: test.py object [OPTIONS] COMMAND [ARGS]...

  Object subcommand

Options:
  --help  Show this message and exit.

Commands:
  get

-----------
> object get --help
Usage: test.py object get [OPTIONS] OPTION

Options:
  --help  Show this message and exit.

使用Python扩展BASH完成

1 个答案:

代码：

测试代码：

结果：