如何使用python-click对同一组中的cmd使用相同的选项

Question

我有如下代码，foo和bar都有一个用户选项，我需要编写：

@click.option('--user', default='*')

每个功能两次。

但实际上我有很多这样的cmds，所以这是很多重复的代码。

@click.group(help="cmd group")
def main():
    pass


@click.command(name='foo')
@click.option('--user', default='*')
def foo(user):
    click.secho(user, fg='green')


@click.command(name='bar')
@click.option('--user', default='*')
def bar(user):
    click.secho(user, fg='green')


main.add_command(foo)
main.add_command(bar)

我想要的是向cmd组中的一个位置添加相同的选项。现在我可以用点击来做这个了吗？

Answer 1

在上下文中传递：

@click.group(help="cmd group")
@click.option('--user', default='*')
@click.pass_context
def main(ctx, user):
    ctx.obj = {'user': user}


@click.command(name='foo')
@click.pass_context
def foo(ctx):
    click.secho(ctx.obj['user'], fg='green')


@click.command(name='bar')
@click.pass_context
def bar(ctx):
    click.secho(ctx.obj['user'], fg='green')


main.add_command(foo)
main.add_command(bar)

mycli --user cats bar将从bar子命令中回显猫

Answer 2

方法1.我假设@click.option()的第一个参数名是para。

USER_DEFAULT = {'para' : '--user', 'default' : '*'}

@click.option(**USER_DEFAULT)

方法2.将其包装在您自己的装饰器中，其中填充了默认参数。