# -*- coding: utf-8 -*- # # Copyright 2016 Mirantis, Inc. # # Licensed under the Apache License, Version 2.0 (the "License"); you may # not use this file except in compliance with the License. You may obtain # a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the # License for the specific language governing permissions and limitations # under the License. import six from fuelclient.cli import error from fuelclient import objects from fuelclient.v1 import base_v1 class GraphClient(base_v1.BaseV1Client): _entity_wrapper = objects.Environment related_graphs_list_api_path = "{related_model}/{related_model_id}" \ "/deployment_graphs/" related_graph_api_path = "{related_model}/{related_model_id}" \ "/deployment_graphs/{graph_type}" graphs_list_api = "graphs/" cluster_deploy_api_path = "graphs/execute/" cluster_own_tasks_api_path = "clusters/{env_id}/deployment_tasks/own/" merged_cluster_tasks_api_path = "clusters/{env_id}/deployment_tasks/" merged_plugins_tasks_api_path = "clusters/{env_id}/deployment_tasks/" \ "plugins/" cluster_release_tasks_api_path = "clusters/{env_id}/deployment_tasks/" \ "release/" def update_graph_for_model( self, data, related_model, related_model_id, graph_type): return self.connection.put_request( self.related_graph_api_path.format( related_model=related_model, related_model_id=related_model_id, graph_type=graph_type), data ) def create_graph_for_model( self, data, related_model, related_model_id, graph_type): return self.connection.post_request( self.related_graph_api_path.format( related_model=related_model, related_model_id=related_model_id, graph_type=graph_type), data ) def get_graph_for_model( self, related_model, related_model_id, graph_type): return self.connection.get_request( self.related_graph_api_path.format( related_model=related_model, related_model_id=related_model_id, graph_type=graph_type)) def upload(self, data, related_model, related_id, graph_type): # create or update if not isinstance(data, dict): data = {'tasks': data} method = self.update_graph_for_model # FIXME(bgaifullin) need to remove loading whole graph only # for detecting that it does not exist try: self.get_graph_for_model(related_model, related_id, graph_type) except error.HTTPError as exc: if '404' in exc.message: method = self.create_graph_for_model # could accept {tasks: [], metadata: {}} or just tasks list method(data, related_model, related_id, graph_type) def execute(self, env_id, nodes=None, graph_types=None, task_names=None, dry_run=False, noop_run=False, force=False): request_data = {'cluster': env_id} def map_args_to_graph_types(graph): result = dict() result['type'] = graph if nodes: result['nodes'] = nodes if task_names: result['tasks'] = task_names return result if graph_types: request_data['graphs'] = list(six.moves.map( map_args_to_graph_types, graph_types )) if dry_run: request_data['dry_run'] = True if noop_run: request_data['noop_run'] = True if force: request_data['force'] = True deploy_data = self.connection.post_request( self.cluster_deploy_api_path, request_data ) return objects.DeployTask.init_with_data(deploy_data) def get_merged_cluster_tasks(self, env_id, graph_type=None): params = {} if graph_type is not None: params['graph_type'] = graph_type return self.connection.get_request( self.merged_cluster_tasks_api_path.format(env_id=env_id), params=params ) def get_merged_plugins_tasks(self, env_id, graph_type=None): params = {} if graph_type is not None: params['graph_type'] = graph_type return self.connection.get_request( self.merged_plugins_tasks_api_path.format(env_id=env_id), params=params ) def get_release_tasks_for_cluster(self, env_id, graph_type=None): params = {} if graph_type is not None: params['graph_type'] = graph_type return self.connection.get_request( self.cluster_release_tasks_api_path.format(env_id=env_id), params=params ) def get_own_tasks_for_cluster(self, env_id, graph_type=None): params = {} if graph_type is not None: params['graph_type'] = graph_type return self.connection.get_request( self.cluster_own_tasks_api_path.format(env_id=env_id), params=params ) def download(self, env_id, level, graph_type): tasks_levels = { 'all': lambda: self.get_merged_cluster_tasks( env_id=env_id, graph_type=graph_type), 'cluster': lambda: self.get_own_tasks_for_cluster( env_id=env_id, graph_type=graph_type), 'plugins': lambda: self.get_merged_plugins_tasks( env_id=env_id, graph_type=graph_type), 'release': lambda: self.get_release_tasks_for_cluster( env_id=env_id, graph_type=graph_type) } return tasks_levels[level]() def get_env_release_graphs_list(self, env_id): """Get list of graphs related to the environment's release. :param env_id: environment ID :type env_id: int :return: list of graphs records :rtype: list[dict] """ data = self.get_by_id(env_id) release_id = data['release_id'] return self.connection.get_request( self.related_graphs_list_api_path.format( related_model='releases', related_model_id=release_id ), params={'fetch_related': '0'} ) def get_env_cluster_graphs_list(self, env_id, fetch_related=True): """Get list of graphs related to the environment. :param env_id: environment ID :type env_id: int :param fetch_related: fetch graphs related to cluster plugins and release :type fetch_related: bool :return: list of graphs records :rtype: list[dict] """ return self.connection.get_request( self.related_graphs_list_api_path.format( related_model='clusters', related_model_id=env_id, ), params={'fetch_related': '1' if fetch_related else '0'} ) def get_env_plugins_graphs_list(self, env_id): """Get list of graphs related to plugins active for the given environment. :param env_id: environment ID :type env_id: int :return: list of graphs records :rtype: list[dict] """ env = objects.Environment(env_id) enabled_plugins_ids = env.get_enabled_plugins() result = [] for plugin_id in enabled_plugins_ids: result += self.connection.get_request( self.related_graphs_list_api_path.format( related_model='plugins', related_model_id=plugin_id ), params={'fetch_related': '0'} ) return result def get_all_graphs_list(self): return self.connection.get_request(self.graphs_list_api) def list(self, env_id=None, filters=None): """Get graphs list. If all filter flags are set to False, then it fill be considered as 'show all' and all filter flags will be toggled to True. :param env_id: environment ID :type env_id: int :param filters: the name of models which graphs will be included to result :return: list of graphs records :rtype: list[dict] """ # we cannot use dict here, because order is important handlers = ( ('release', self.get_env_release_graphs_list), ('plugins', self.get_env_plugins_graphs_list), ('cluster', self.get_env_cluster_graphs_list) ) graphs_list = [] filters = filters and set(filters) if env_id: for relation, handler in handlers: if not filters or relation in filters: graphs_list.extend(handler(env_id=env_id)) else: all_graphs_list = self.get_all_graphs_list() for graph in all_graphs_list: for relation in graph['relations']: if not filters or relation['model'] in filters: graphs_list.append(graph) break return graphs_list def get_client(connection): return GraphClient(connection)