时间戳

首页 > 解决方案 > 使用 python click 的 SSH 或类似 sudo 的行为

python - 使用 python click 的 SSH 或类似 sudo 的行为

问题描述

我click用于我的 Python 脚本命令行处理。我想实现类似于 SSH 或 Sudo 的行为，即解析一些已知参数并按原样处理其他所有内容，无需任何处理。

例如，考虑这个命令行：

ssh -v myhost echo -n -e foo bar

请注意，-v它将由 SSH 处理，但echo它之后的所有内容都不会作为选项处理。

这是我当前的实现：

@click.command()
@click.option('-v', '--verbose', is_flag=True)
@click.argument('target')
@click.argument('command', nargs=-1)
def my_command(verbose, target, command):
    print('verbose:', verbose)
    print('target:', target)
    print('command:', command)

它没有按预期工作：

$ python test.py -v hostname echo -e foo
Usage: test.py [OPTIONS] TARGET [COMMAND]...
Try "test.py --help" for help.

Error: no such option: -e

我可以添加--分隔符来强制执行预期的行为：

$ python /tmp/test.py -v hostname -- echo -e foo
verbose: True
target: hostname
command: ('echo', '-e', 'foo')

但这不是我想要的。

我还可以添加ignore_unknown_options=True：

@click.command(context_settings=dict(ignore_unknown_options=True))
...

$ python /tmp/test.py -v hostname echo -e foo
verbose: True
target: hostname
command: ('echo', '-e', 'foo')

但它不适用于已知选项，就像-v在这种情况下一样。

所以问题是：遇到某些选项后如何指示click停止处理任何选项argument？

标签： pythoncommand-line-interfacepython-click

解决方案

您可以使用自定义click.Command类自动插入--需要的位置，例如：

自定义类：

class RealNargsMinusOne(click.Command):

    def parse_args(self, ctx, args):
        orig_args = list(args)
        try:
            return super(RealNargsMinusOne, self).parse_args(ctx, args)
        except click.NoSuchOption as exc:
            first_unknown = str(exc).split()[-1]
            position_unknown = orig_args.index(first_unknown)
            orig_args.insert(position_unknown, '--')
            return super(RealNargsMinusOne, self).parse_args(ctx, orig_args)

使用自定义类：

要使用自定义类，请将其作为cls参数传递给命令装饰器，例如：

@click.command(cls=RealNargsMinusOne)
....
def my_command():
    ....

这是如何运作的？

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

在这种情况下，我们覆盖click.Command.parse_args()并捕获click.NoSuchOption异常。然后我们在有问题的选项前用 a 修改参数列表--并重试解析。

测试代码：

import click

@click.command(cls=RealNargsMinusOne)
@click.option('-v', '--verbose', is_flag=True)
@click.argument('target')
@click.argument('command', nargs=-1)
def my_command(verbose, target, command):
    print('verbose:', verbose)
    print('target:', target)
    print('command:', command)


if __name__ == "__main__":
    commands = (
        '-v hostname echo -e foo',
        '-v hostname -x echo -e foo',
        '-x hostname -x echo -e foo',
        '--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)
            my_command(cmd.split(), allow_extra_args=True)

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

结果：

Click Version: 6.7
Python Version: 3.6.3 (v3.6.3:2c5fed8, Oct  3 2017, 18:11:49) [MSC v.1900 64 bit (AMD64)]
-----------
> -v hostname echo -e foo
verbose: True
target: hostname
command: ('echo', '-e', 'foo')
-----------
> -v hostname -x echo -e foo
verbose: True
target: hostname
command: ('-x', 'echo', '-e', 'foo')
-----------
> -x hostname -x echo -e foo
verbose: False
target: -x
command: ('hostname', '-x', 'echo', '-e', 'foo')
-----------
> --help
Usage: test.py [OPTIONS] TARGET [COMMAND]...

Options:
  -v, --verbose
  --help         Show this message and exit.
-----------
> 
Usage: test.py [OPTIONS] TARGET [COMMAND]...

Error: Missing argument "target".

推荐阅读