Usage: ./ecs-validate-deploy.py <region> <cluster>

ecs-validate-deploy.py

#! /usr/bin/env python

import boto3
import logging
import sys


log = logging.getLogger(__name__)
ch = logging.StreamHandler()
ch.setLevel(logging.INFO)
log.addHandler(ch)

def validate_deploy_service(region, cluster, service):
    client = boto3.client('ecs', region_name=region)

    response = client.describe_services(
        cluster=cluster,
        services=[
            service,
        ]
    )

    task_definition = response['services'][0]['taskDefinition']

    response = client.list_tasks(
        cluster=cluster,
        serviceName=service,
    )

    response = client.describe_tasks(
        cluster=cluster,
        tasks=response['taskArns']
    )

    for task in response['tasks']:
        if task['taskDefinitionArn'] != task_definition:
            log.error('Old task definition {} in place when {} expected for task {}.'.format(task['taskDefinitionArn'], task_definition, task['taskArn']))
            return False
        if task['desiredStatus'] != task['lastStatus']:
            log.error('Status {} found when expected {} for task {}.'.format(task['desiredStatus'], task['lastStatus'], task['taskArn']))
            return False

    return True


def validate_deploy(region, cluster):
    for service in ('app-main', 'celery-worker', 'celery-master'):
        if not validate_deploy_service(region, cluster, service):
            return False
    return True

if __name__ == '__main__':
    if not validate_deploy(sys.argv[1], sys.argv[2]):
        sys.exit(1)