我有一个类似的脚本:
#myscript.py
import click
def step1(arg1):
print('Step 1: ' + arg1)
def step2(arg2):
print('Step 2: ' + arg2)
def main(arg1, arg2):
step1(arg1)
step2(arg2)
大部分时间我想用myscript arg1 arg2
运行脚本,但有时我可能只想运行一步:例如myscript step1 arg1
。 如何设置点击以执行此操作?有没有办法让一个默认命令,然后是其他可选命令?
有时,从另一个命令调用一个命令可能会很有趣。这种模式通常不受Click的阻碍,但仍有可能。
我是否需要使用此click.invoke()
模式?
答案 0 :(得分:1)
我认为Multi Command Chaining和Multi Command Pipelines功能旨在涵盖这种情况。流水线提供了所请求的确切行为(在命令行中没有给出任何step1
和step2
),但它更详细,并且chain=True
没有任何参数可以是可选的;您必须(a)始终同时提供arg1
和arg2
,即使仅调用step2
也是如此;或(b)将这些论点转换为选项(--arg1 foo
而不是foo
)。
import click
@click.group(chain=True)
def cli():
pass
@cli.command()
@click.argument('arg1')
def step1(arg1):
click.echo('Step 1: ' + arg1)
@cli.command()
@click.argument('arg2')
def step2(arg2):
click.echo('Step 2: ' + arg2)
cli()
然后:
$ python3 chain.py step1 foo step2 bar
Step 1: foo
Step 2: bar
$ python3 chain.py step2 bar
Step 2: bar
import click
@click.group(chain=True, invoke_without_command=True)
@click.argument('arg1')
@click.argument('arg2')
def cli(arg1, arg2):
pass
@cli.resultcallback()
def process_pipeline(processors, **kwargs):
# If no commands given, invoke step1 then step2
processors = processors if len(processors) else [step1, step2]
for processor in processors:
processor(**kwargs)
def step1(**kwargs):
click.echo('Step 1: ' + kwargs['arg1'])
def step2(**kwargs):
click.echo('Step 2: ' + kwargs['arg2'])
@cli.command('step1')
def make_step1():
return step1
@cli.command('step2')
def make_step2():
return step2
cli()
然后
$ python3 pipeline.py foo bar
Step 1: foo
Step 2: bar
$ python3 pipeline.py foo bar step2
Step 2: bar
答案 1 :(得分:1)
我没有说出问题来完全解释我想要做什么,因为每一步都需要前一步的输出。非常感谢Paul让我走上了正确的道路。
我的解决方案如下:
@click.group(invoke_without_command=True)
@click.option('--arg1')
@click.option('--arg2')
@click.pass_context
def cli(ctx, arg1, arg2):
'''Description
'''
if ctx.invoked_subcommand is None:
do_everything(ctx, arg1, arg2)
@cli.command()
@click.option('--arg1')
def step_1(arg1):
return do_something(arg1)
@cli.command()
@click.argument('step_one_result')
@click.option('--arg2')
def step_2(step_one_result, arg2):
do_something_else(step_one_result, arg2)
def do_everything(ctx, arg1, arg2):
step_one_result = ctx.invoke(step_1, arg1=arg1)
ctx.invoke(do_something_else, step_one_result=step_one_result, arg2=arg2)
#and because of weirdness with pass_context and using setuptools
def main():
cli(obj={})
if __name__ == '__main__':
main()
编辑:您会注意到ctx.invoke()
的使用,这是调用函数所必需的,而不会出现以下错误
line 619, in make_context
ctx = Context(self, info_name=info_name, parent=parent, **extra)
TypeError: __init__() got an unexpected keyword argument 'arg1'