boroivanov/ecs-tools

import os
import sys
import boto3
import click

from botocore.exceptions import ProfileNotFound, NoRegionError
from ecstools.lib.cli import AliasedGroup, Subcommand

version = '0.2.7'

commands_dir = os.path.join(os.path.dirname(__file__), 'commands')


class ClusterCommand(Subcommand):
    plugin_folder = os.path.join(commands_dir, 'cluster')


class ServiceCommand(Subcommand):
    plugin_folder = os.path.join(commands_dir, 'service')


class TaskDefinitionCommand(Subcommand):
    plugin_folder = os.path.join(commands_dir, 'task-definition')


@click.command(cls=AliasedGroup)
@click.pass_context
@click.version_option(version=version, message=version)
@click.option('-p', '--profile', help='AWS profile')
@click.option('-r', '--region', help='AWS region')
def cli(ctx, region, profile):
    """AWS ECS deploy tools"""
    try:
        sess = boto3.session.Session(profile_name=profile, region_name=region)
    except ProfileNotFound as e:
        click.echo(e, err=True)
        sys.exit(1)

    try:
        ecs = sess.client('ecs')
        ecr = sess.client('ecr')
        elbv2 = sess.client('elbv2')
    except NoRegionError as e:
        click.echo(e, err=True)
        sys.exit(1)

    ctx.obj = {
        'region': region,
        'profile': profile,
        'ecs': ecs,
        'ecr': ecr,
        'elbv2': elbv2
    }


@cli.group(cls=ClusterCommand)
@click.pass_context
def cluster(ctx):
    """Manage clusters"""
    pass


@cli.group(cls=ServiceCommand)
@click.pass_context
def service(ctx):
    """Manage and deploy services"""
    pass


@cli.group(cls=TaskDefinitionCommand, name='task-definition')
@click.pass_context
def task_definition(ctx):
    """Manage task definitions"""
    pass


if __name__ == '__main__':
    cli()