Monitoring Restoration Progress of a Redshift Cluster

import boto3 import time import botocore.exceptions creds = _get_creds(cred_label)['creds'] access_key = creds['username'] secret_key = creds['password'] def monitor_restore_progress(cluster_id, region): # Initialize the boto3 Redshift client with the specified region redshift = boto3.client('redshift', aws_access_key_id=access_key,aws_secret_access_key=secret_key,region_name=region) # Mark the start time start_time = time.time() # Start an infinite loop to continuously monitor the cluster's status while True: try: # Fetch the current status of the specified Redshift cluster response = redshift.describe_clusters(ClusterIdentifier=cluster_id) # Check if the 'Clusters' list is not empty if not response['Clusters']: print(f"No cluster found with identifier: {cluster_id}") break cluster_status = response['Clusters'][0]['ClusterStatus'] # Check the cluster's status and provide appropriate feedback if cluster_status in ['creating', 'restoring']: print(f"Cluster {cluster_id} status: {cluster_status}. Restoration is in progress...") elif cluster_status == 'available': elapsed_time = time.time() - start_time mins, secs = divmod(elapsed_time, 60) hours, mins = divmod(mins, 60) print(f"Cluster {cluster_id} is now available. Restoration completed successfully in {int(hours)}h {int(mins)}m {int(secs)}s.") break else: print(f"Cluster {cluster_id} status: {cluster_status}.") break # Wait for 30 seconds before checking the status again time.sleep(30) except botocore.exceptions.ClientError as e: print(f"ClientError: {e.response['Error']['Message']}") break except Exception as e: print(f"An unexpected error occurred: {e}") break # Example usage with hardcoded values. You should replace these with actual values received from the previous task. # cluster_identifier = "redshift-cluster-restored" # Replace this with actual cluster ID received from previous task # aws_region = "us-west-2" # Replace this with actual AWS region received from previous task # Begin monitoring the restoration progress of the specified cluster in the specified region if new_cluster_id: cluster_identifier = new_cluster_id monitor_restore_progress(cluster_identifier, aws_region) else: print("No Cluster Id provided for monitoring")