def create_cluster(self, name, ssh_public_key=None, **cluster_params):
cluster = models.ClusterUpdateParams(name=name,
ssh_public_key=ssh_public_key,
**cluster_params)
print("Creating cluster with params", cluster.__dict__)
result = self.client.register_cluster(new_cluster_params=cluster)
return result
def set_hosts_roles(self, cluster_id, hosts_with_roles):
log.info(
"Setting roles for hosts %s in cluster %s", hosts_with_roles, cluster_id
)
hosts = models.ClusterUpdateParams(hosts_roles=hosts_with_roles)
return self.client.update_cluster(
cluster_id=cluster_id, cluster_update_params=hosts
)