Question

我正在尝试使用Pallet Click制作一个命令行程序，该程序需要一个输入参数列表和一个可选的输出参数。

在具有python-click版本7.0的Ubuntu 18.04 Python 3.6和Windows 10 Python 3.7上，行为相同。

我制作了一个测试文件click_test.py：

import click


@click.command()
@click.argument('src', nargs=-1, required=True)
@click.argument('dst', required=False)
def copy(src, dst):
    print(f'{src!r}')
    print(f'{dst!r}')


if __name__ == '__main__':
    copy()

运行python click_test.py first-argument会得到以下输出：

Usage: click_test.py [OPTIONS] SRC... [DST]
Try "click_test.py --help" for help.

Error: Missing argument "SRC...".

用法说明描述了我的期望。 SRC是必需的，但DST是可选的。但是，错误消息仍然显示缺少SRC。

这是正确的行为，还是错误？

Answer 1

您所需的命令行有点含糊。单击如何知道src是src还是最后一个dst？因此，使用nargs=-1，解析器会将值从src推迟到dst。

但是，通过以自定义click.Argument类的形式进行一些重复设置，可以实现您对问题的期望。

自定义类别：

def take_empty_from(other_param_name):

    class EmptyFrom(click.Argument):

        def consume_value(self, ctx, opts):
            value = opts.get(self.name)
            if value == () and opts.get(other_param_name):
                value = opts[self.name] = (opts.get(other_param_name), )
                opts[other_param_name] = None
                return value
            else:
                return super(EmptyFrom, self).consume_value(ctx, opts)

    return EmptyFrom

使用自定义类：

@click.command()
@click.argument('src', nargs=-1, required=True, cls=take_empty_from('dst'))
@click.argument('dst', required=False)
def copy(src, dst):

这是如何工作的？

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

在这种情况下，我们将覆盖click.Argument.consume_value()，然后在仅一个参数的情况下，从dst抓取参数并将其放入src。

请注意：尽管此行为符合问题中的要求，但仍未得到答案，为什么仅在只有一个src参数时才不需要dst。

测试代码：

import click


@click.command()
@click.argument('src', nargs=-1, required=True, cls=take_empty_from('dst'))
@click.argument('dst', required=False)
def copy(src, dst):
    click.echo('src: {}'.format(src))
    click.echo('dst: {}'.format(dst))


if __name__ == "__main__":
    commands = (
        '',
        'a',
        'a b',
        'a b c',
        '--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)
            copy(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)]
-----------
> 
Usage: click_prog.py [OPTIONS] SRC... [DST]

Error: Missing argument "src".
-----------
> a
src: ('a',)
dst: None
-----------
> a b
src: ('a',)
dst: b
-----------
> a b c
src: ('a', 'b')
dst: c
-----------
> --help
Usage: click_prog.py [OPTIONS] SRC... [DST]

Options:
  --help  Show this message and exit.

这种python-click行为正确吗？

1 个答案:

自定义类别：

使用自定义类：

这是如何工作的？

测试代码：

结果：