Class RedshiftWrapper

python/example_code/redshift/redshift.py:16–153

Encapsulates Amazon Redshift cluster operations.

# snippet-start:[python.example_code.redshift.RedshiftWrapper]
class RedshiftWrapper:
    """
    Encapsulates Amazon Redshift cluster operations.
    """

    def __init__(self, redshift_client):
        """
        :param redshift_client: A Boto3 Redshift client.
        """
        self.client = redshift_client

    # snippet-end:[python.example_code.redshift.RedshiftWrapper]

    @classmethod
    def from_client(cls):
        """
        Creates a wrapper object from a Boto3 client.

        :return: The wrapper object.
        """
        client = boto3.client("redshift")
        return cls(client)

    # snippet-start:[python.example_code.redshift.CreateCluster]
    def create_cluster(
        self,
        cluster_identifier,
        node_type,
        master_username,
        master_user_password,
        publicly_accessible,
        number_of_nodes,
    ):
        """
        Creates a cluster.

        :param cluster_identifier: The name of the cluster.
        :param node_type: The type of node in the cluster.
        :param master_username: The master username.
        :param master_user_password: The master user password.
        :param publicly_accessible: Whether the cluster is publicly accessible.
        :param number_of_nodes: The number of nodes in the cluster.
        :return: The cluster.
        """

        try:
            cluster = self.client.create_cluster(
                ClusterIdentifier=cluster_identifier,
                NodeType=node_type,
                MasterUsername=master_username,
                MasterUserPassword=master_user_password,
                PubliclyAccessible=publicly_accessible,
                NumberOfNodes=number_of_nodes,
            )
            return cluster
        except ClientError as err:
            logging.error(
                "Couldn't create a cluster. Here's why: %s: %s",