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 )