首页 > 解决方案 > 有没有办法可以返回数据并提供额外的参数,python3 @click

问题描述

我正在组合一个 Python3 工具,该工具允许我与 AWS 开发工具包进行交互并从 AWS 中提取某些信息,但我无法让它提供输出,并根据选项提供不同的输出。解释代码可能有助于理解。

import boto3
import click

@click.group()
@click.option('--profile', default=None, help='Specify AWS Profile')
@click.option('--region', default=None, help='Specify AWS Region')
@click.pass_context
def cli(ctx, profile, region):
    ctx.obj = Params(profile, region)

@cli.command(name='ec2')
@click.pass_obj
def ec2(ctx, update):
    print("list_instances")

@ec2.command()
@click.pass_obj
def update_instances(ctx):
     print("update instances")

if __name__ == '__main__':
    cli()

理想情况下,我希望能够运行以下命令:

# app.py ec2<-- 这将返回实例列表

# app.py ec2 --update instance_id/all<-- 这将在所有返回的 Id 或特定实例 Id 上更新 SSM 代理。

我可以让它返回实例列表但没有其他选项,或者我可以让它返回更新命令,但没有实例列表。

非常感谢,扎克

标签: pythonamazon-web-servicesboto3python-click

解决方案


import boto3
import click

@click.group()
@click.option('--profile', default=None, help='Specify AWS Profile')
@click.option('--region', default=None, help='Specify AWS Region')
@click.pass_context
def cli(ctx, profile, region):
    ctx.obj = Params(profile, region)

@cli.command(name='ec2')
@click.option('--update', default=None)
@click.pass_obj
def ec2(ctx, update):
    if update is not None:
      update_instances(ctx, update)
    else:
      print("list_instances")
 
def update_instances(ctx, update):
     print("update instances")

if __name__ == '__main__':
    cli()

需要注意的是,在您的示例中,变量Params未定义,不确定它是否是故意的,或者您只是忘记将其包含在问题中。


推荐阅读