aws_cluster_destroy.py

import boto3
import configparser


def config_parser():
    """
    Set the AWS Config parameters with value from dwh.cfg config file
    """

    global KEY, SECRET, DWH_CLUSTER_TYPE, DWH_NUM_NODES
    global DWH_NODE_TYPE, DWH_CLUSTER_IDENTIFIER, DWH_DB
    global DWH_DB_USER, DWH_DB_PASSWORD, DWH_PORT, DWH_IAM_ROLE_NAME

    print("Parsing the configuration file...\n")

    config = configparser.ConfigParser()
    with open('dwh.cfg') as configfile:
        config.read_file(configfile)

        KEY = config.get('AWS', 'KEY')
        SECRET = config.get('AWS', 'SECRET')

        DWH_CLUSTER_TYPE = config.get("DWH", "DWH_CLUSTER_TYPE")
        DWH_NUM_NODES = config.get("DWH", "DWH_NUM_NODES")
        DWH_NODE_TYPE = config.get("DWH", "DWH_NODE_TYPE")

        DWH_IAM_ROLE_NAME = config.get("DWH", "DWH_IAM_ROLE_NAME")
        DWH_CLUSTER_IDENTIFIER = config.get("DWH", "DWH_CLUSTER_IDENTIFIER")

        DWH_DB = config.get("CLUSTER", "DB_NAME")
        DWH_DB_USER = config.get("CLUSTER", "DB_USER")
        DWH_DB_PASSWORD = config.get("CLUSTER", "DB_PASSWORD")
        DWH_PORT = config.get("CLUSTER", "DB_PORT")


def aws_client(service, region):
    """
    Creates an AWS client (specified by the argument) in region (specified by argument)
    :param service: The service to be created
    :param region: The region where service has to be created
    :return client: The client for AWS service
    """

    global KEY, SECRET
    client = boto3.client(service, aws_access_key_id=KEY, aws_secret_access_key=SECRET, region_name=region)

    return client


def redshift_cluster_status(redshift):
    """
    Retrieves the Redshift cluster status
    :param redshift: Boto3 client for Redshift
    :return cluster_status: The cluster status
    """

    global DWH_CLUSTER_IDENTIFIER

    cluster_props = redshift.describe_clusters(ClusterIdentifier=DWH_CLUSTER_IDENTIFIER)['Clusters'][0]
    cluster_status = cluster_props['ClusterStatus'].lower()

    return cluster_status        
        

def destroy_redshift_cluster(redshift):
    """
    Destroy the Redshift cluster
    :param redshift: Boto3 client for Redshift
    """

    global DWH_CLUSTER_IDENTIFIER

    redshift.delete_cluster(ClusterIdentifier=DWH_CLUSTER_IDENTIFIER, SkipFinalClusterSnapshot=True)

        
def main():
    config_parser()

    redshift = aws_client('redshift', "us-west-2")

    if redshift_cluster_status(redshift) == 'available':
        print('Cluster status: available')
        destroy_redshift_cluster(redshift)
        print('Cluster status: ', redshift_cluster_status(redshift))
    else:
        print("Cluster not available.")


if __name__ == '__main__':
    main()