AWS CodeDeploy でロールバック時に削除されないリソースを自動削除するように実装してみた

################################################################################
#import library
################################################################################
from __future__ import print_function
import boto3
import json
import time
################################################################################
#import library end
################################################################################

################################################################################
#lambda start
################################################################################
def lambda_handler(event, context):
    # recieve sns data and fetch Status and DeploymentID
    rawdata = event['Records']
    data = json.loads(rawdata[0]['Sns']['Message'])
    accountid = data['accountId']
    region = data['region']
    snsinfoname = 'send-ticket-rollback'
    print("AccountID : " + accountid)
    print("region : " + region)
    print("Message :")
    print(data)

    # Create instance
    client = boto3.client("autoscaling")
    sns = boto3.client('sns')

    ################################################################################
    #common function
    ################################################################################
    def common_func():
        print("Status is " + Status + ". searching for tag: " + DEPID)
        response = client.describe_auto_scaling_groups(
            Filters = [
                {
                    'Name': 'tag:CodeDeployProvisioningDeploymentId',
                    'Values': [DEPID]
                }
            ]
        )

        ASGName = response['AutoScalingGroups'][0]['AutoScalingGroupName']
        print("ASG to be deleted is " + ASGName)
        response = client.delete_auto_scaling_group(
            AutoScalingGroupName=ASGName,
            ForceDelete=True
            )
        print(response)

        if response['ResponseMetadata']['HTTPStatusCode'] == 200:
            print("deletion has successfully started")
            max_wait_time = 780
            start_time = time.time()
            ASGNameList = []
            ASGNameList.append(ASGName)
            while True:
                elapsed_time = time.time() - start_time
                response = client.describe_auto_scaling_groups(
                    AutoScalingGroupNames=ASGNameList
                )
                if not response['AutoScalingGroups']:
                    print(f'Auto Scaling group {ASGName} deleted successfully.')
                    break
                if elapsed_time > max_wait_time:
                    print(f'Error: Timed out waiting for Auto Scaling group {ASGName} to be deleted.')
                    httpresponse = json.dumps(response['ResponseMetadata'])
                    subject = '[CodeDeploy : ASG Deletion Timeout] Deletion of ASG is taking too much time.'
                    body = 'Check the CodeDeploy Deployment of below Account.\n' + 'AccountID : ' + accountid + '\nRegion : ' + region + '\nResponse : \n' + httpresponse + '\nAutoScalingGroupName : ' + ASGName + '\nTriggering Deployment ID : ' + DEPID
                    snsresponse = sns.publish(
                        TopicArn='arn:aws:sns:' + region + ':' + accountid + ':' + snsinfoname,
                        Message=body,
                        Subject=subject
                    )
                    print(snsresponse)
                    break
                time.sleep(10)
        else:
            print("something went wrong.check with administrators about the states.")
            httpresponse = json.dumps(response['ResponseMetadata'])
            subject = '[CodeDeploy : ASG Deletion Error] Deletion API responded with Status Code other than 200.'
            body = 'Check the CodeDeploy Deployment of below Account.\n' + 'AccountID : ' + accountid + '\nRegion : ' + region + '\nResponse : \n' + httpresponse + '\nAutoScalingGroupName : ' + ASGName + '\nTriggering Deployment ID : ' + DEPID
            snsresponse = sns.publish(
                TopicArn='arn:aws:sns:' + region + ':' + accountid + ':' + snsinfoname,
                Message=body,
                Subject=subject
            )
            print(snsresponse)
    ##################################################################################
    #common function end
    ##################################################################################

    ##################################################################################
    #main
    ##################################################################################
    Status = data["status"]
    if Status == "FAILED":
        DEPID = data["deploymentId"]
        common_func()

    elif Status == "SUCCEEDED":
        RolebackInfo = json.loads(data['rollbackInformation'])
        DEPID = RolebackInfo['RollbackTriggeringDeploymentId']
        common_func()

    elif Status == "CREATED" or Status == "IN_PROGRESS":
        print("can ignore these messages, not critical.")

    else:
        print('irregular value, check with administrators')
        subject = '[CodeDeploy : unknown status] could not get value or unknown value.'
        body = 'Unknown CodeDeploy Status : ' + Status + '\nAccountID : ' + accountid + '\nRegion : ' + region + '\nMessage : \n' + json.dumps(data)
        snsresponse = sns.publish(
            TopicArn='arn:aws:sns:' + region + ':' + accountid + ':' + snsinfoname,
            Message=body,
            Subject=subject
        )
        print(snsresponse)
    ##################################################################################
    #main end
    ##################################################################################
################################################################################
#lambda end
################################################################################

def lambda_handler(event, context):
    # recieve sns data and fetch Status and DeploymentID
    rawdata = event['Records']
    data = json.loads(rawdata[0]['Sns']['Message'])
    accountid = data['accountId']
    region = data['region']
    snsinfoname = 'send-ticket-rollback'
    print("AccountID : " + accountid)
    print("region : " + region)
    print("Message :")
    print(data)

    def common_func():
        print("Status is " + Status + ". searching for tag: " + DEPID)
        response = client.describe_auto_scaling_groups(
            Filters = [
                {
                    'Name': 'tag:CodeDeployProvisioningDeploymentId',
                    'Values': [DEPID]
                }
            ]
        )

    Status = data["status"]
    if Status == "FAILED":
        DEPID = data["deploymentId"]
        common_func()

    elif Status == "SUCCEEDED":
        RolebackInfo = json.loads(data['rollbackInformation'])
        DEPID = RolebackInfo['RollbackTriggeringDeploymentId']
        common_func()

    elif Status == "CREATED" or Status == "IN_PROGRESS":
        print("can ignore these messages, not critical.")

    else:
        print('irregular value, check with administrators')
        subject = '[CodeDeploy : unknown status] could not get value or unknown value.'
        body = 'Unknown CodeDeploy Status : ' + Status + '\nAccountID : ' + accountid + '\nRegion : ' + region + '\nMessage : \n' + json.dumps(data)
        snsresponse = sns.publish(
            TopicArn='arn:aws:sns:' + region + ':' + accountid + ':' + snsinfoname,
            Message=body,
            Subject=subject
        )
        print(snsresponse)