Question

我想在配置中使用一个值（将其加载到上下文中）作为单击命令选项的默认值。我已经阅读了this section的文档，但我认为我不了解我需要做什么。

这是我的示例脚本：

import sys
import click

@click.group()
@click.pass_context
def cli(ctx):
    """
    CLI
    """
    ctx.ensure_object(dict)
    ctx.obj['DEFAULT_ENVIRONMENT'] = "dev"


@cli.command()
@click.option('-e', '--environment', required=True, default=click.get_current_context().obj['DEFAULT_ENVIRONMENT'])
def show_env(environment):
    click.echo(environment)


if __name__ == '__main__':
    cli()

如果我运行python cli.py show-env，目标是使它输出dev（因为从上下文中加载参数时我没有传递参数）。

此操作失败

Traceback (most recent call last):
  File "testcli.py", line 15, in <module>
    @click.option('-e', '--environment', required=True, default=click.get_current_context().obj['DEFAULT_ENVIRONMENT'])
  File "/home/devuser/.virtualenvs/cli/lib/python3.6/site-packages/click/globals.py", line 26, in get_current_context
    raise RuntimeError('There is no active click context.')
RuntimeError: There is no active click context.

我也尝试在@pass_context命令上使用show_env，如下所示：

@cli.command()
@click.option('-e', '--environment', required=True, default=ctx.obj['DEFAULT_ENVIRONMENT'])
@click.pass_context
def show_env(ctx, environment):
    click.echo(environment)

这失败了，因为此时未定义ctx。

Traceback (most recent call last):
  File "testcli.py", line 15, in <module>
    @click.option('-e', '--environment', required=True, default=ctx.obj['DEFAULT_ENVIRONMENT'])
NameError: name 'ctx' is not defined

我可以使用上下文设置命令选项的默认值吗？

Answer 1

如您所述，在尝试检查上下文时，该上下文尚不存在。您可以使用以下自定义类，从上下文延迟默认查询，直到上下文确实存在：

自定义类

def default_from_context(default_name):

    class OptionDefaultFromContext(click.Option):

        def get_default(self, ctx):
            self.default = ctx.obj[default_name]
            return super(OptionDefaultFromContext, self).get_default(ctx)

    return OptionDefaultFromContext

使用自定义类

要使用自定义类，请通过click.option参数将其传递到cls，例如：

@click.option('-e', '--environment', required=True,
              cls=default_from_context('DEFAULT_ENVIRONMENT'))

这是如何工作的？

之所以可行，是因为click是一个设计良好的OO框架。 @click.option()装饰器通常会实例化click.Option对象，但允许使用cls参数来覆盖此行为。因此，在我们自己的类中继承click.Option并超越所需的方法是相对容易的事情。

在这种情况下，我们将覆盖click.Option.get_default()。在get_default()中，我们检查上下文并设置默认值。然后，我们将父级称为get_default()，以继续进行进一步的处理。

测试代码：

import click

@click.group()
@click.pass_context
def cli(ctx):
    """
    CLI
    """
    ctx.ensure_object(dict)
    ctx.obj['DEFAULT_ENVIRONMENT'] = "dev"


@cli.command()
@click.option('-e', '--environment', required=True,
              cls=default_from_context('DEFAULT_ENVIRONMENT'))
def show_env(environment):
    click.echo(environment)


if __name__ == "__main__":
    commands = (
        'show_env',
        '--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.5 (v3.6.5:f59c0932b4, Mar 28 2018, 05:52:31) 
[GCC 4.2.1 Compatible Apple LLVM 6.0 (clang-600.0.57)]
-----------
> show_env
dev
-----------
> --help
Usage: click_prog.py [OPTIONS] COMMAND [ARGS]...

  CLI

Options:
  --help  Show this message and exit.

Commands:
  show_env

我可以使用上下文值作为click.option（）默认值吗？

1 个答案:

自定义类

使用自定义类

这是如何工作的？

测试代码：

结果：