这种python-click行为正确吗？

Question

我正在尝试使用托盘单击来制作一个命令行程序，该程序接受一系列输入参数和一个可选的输出参数。

在带有Python的Ubuntu 18.04 Python 3.6和Windows 10 python 3.7上的行为相同-单击版本7.0。

我制作了一个测试文件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中。

注意:虽然此行为满足了问题中的请求，但它仍然没有回答为什么只有在只有一个dst参数时才不需要src。

测试代码：

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.