def setUp(self):
self.api = RedshiftConnection()
self.cluster_prefix = 'boto-redshift-cluster-%s'
self.node_type = 'dw.hs1.xlarge'
self.parent_username = '******'
self.parent_password = '******'
self.db_name = 'simon'
# Redshift was taking ~20 minutes to bring clusters up in testing.
self.wait_time = 60 * 20
def __init__(self, parameters):
try:
self.access_key = parameters["access_key"]
self.secret_key = parameters["secret_key"]
except:
logging.error("Something went wrong initializing RedshiftManager")
sys.exit()
# Create connection
self.connection = RedshiftConnection(aws_access_key_id = self.access_key, aws_secret_access_key = self.secret_key)
# Create clusters
# Delete cluster
def connect_redshift(aws_access_key_id=None,
aws_secret_access_key=None,
**kwargs):
"""
:type aws_access_key_id: string
:param aws_access_key_id: Your AWS Access Key ID
:type aws_secret_access_key: string
:param aws_secret_access_key: Your AWS Secret Access Key
:rtype: :class:`boto.redshift.layer1.RedshiftConnection`
:return: A connection to Amazon's Redshift service
"""
from boto.redshift.layer1 import RedshiftConnection
return RedshiftConnection(aws_access_key_id=aws_access_key_id,
aws_secret_access_key=aws_secret_access_key,
**kwargs)
def __init__(self, connection=None, **kw_args):
self.poll_interval_sec = kw_args.pop('poll_interval_sec', 30)
if connection is None:
self._conn = RedshiftConnection(**kw_args)
else:
self._conn = connection
