def list_clusters(self): try: return [ models.Cluster(pool) for pool in self.__list_clusters(aztk.models.Software.spark) ] except batch_error.BatchErrorException as e: raise error.AztkError(helpers.format_batch_exception(e))
def wait_until_cluster_is_ready(self, cluster_id: str): try: util.wait_for_master_to_be_ready(self, cluster_id) pool = self.batch_client.pool.get(cluster_id) nodes = self.batch_client.compute_node.list(pool_id=cluster_id) return models.Cluster(pool, nodes) except batch_error.BatchErrorException as e: raise error.AztkError(helpers.format_batch_exception(e))
def list_clusters(core_cluster_operations): try: software_metadata_key = base_models.Software.spark return [ models.Cluster(cluster) for cluster in core_cluster_operations.list(software_metadata_key) ] except batch_error.BatchErrorException as e: raise error.AztkError(helpers.format_batch_exception(e))
def get_cluster(self, cluster_id: str): try: pool, nodes = self.__get_pool_details(cluster_id) return models.Cluster(pool, nodes) except batch_error.BatchErrorException as e: raise error.AztkError(helpers.format_batch_exception(e))
def get_cluster(core_cluster_operations, cluster_id: str): try: cluster = core_cluster_operations.get(cluster_id) return models.Cluster(cluster) except BatchErrorException as e: raise error.AztkError(helpers.format_batch_exception(e))