Source code for heat.engine.clients.os.sahara

# Copyright (c) 2014 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 logging

from oslo_utils import uuidutils
from saharaclient.api import base as sahara_base
from saharaclient import client as sahara_client
import six

from heat.common import exception
from heat.common.i18n import _
from heat.common.i18n import _LI
from heat.engine.clients import client_plugin
from heat.engine import constraints

LOG = logging.getLogger(__name__)


[docs]class SaharaClientPlugin(client_plugin.ClientPlugin): exceptions_module = sahara_base def _create(self): con = self.context endpoint_type = self._get_client_option('sahara', 'endpoint_type') endpoint = self.url_for(service_type='data-processing', endpoint_type=endpoint_type) args = { 'service_type': 'data-processing', 'input_auth_token': self.auth_token, 'auth_url': con.auth_url, 'project_name': con.tenant, 'sahara_url': endpoint, 'insecure': self._get_client_option('sahara', 'insecure'), 'cacert': self._get_client_option('sahara', 'ca_file') } client = sahara_client.Client('1.1', **args) return client
[docs] def is_not_found(self, ex): return (isinstance(ex, sahara_base.APIException) and ex.error_code == 404)
[docs] def is_over_limit(self, ex): return (isinstance(ex, sahara_base.APIException) and ex.error_code == 413)
[docs] def is_conflict(self, ex): return (isinstance(ex, sahara_base.APIException) and ex.error_code == 409)
[docs] def is_not_registered(self, ex): return (isinstance(ex, sahara_base.APIException) and ex.error_code == 400 and ex.error_name == 'IMAGE_NOT_REGISTERED')
[docs] def get_image_id(self, image_identifier): ''' Return an id for the specified image name or identifier. :param image_identifier: image name or a UUID-like identifier :returns: the id of the requested :image_identifier: :raises: exception.ImageNotFound, exception.PhysicalResourceNameAmbiguity ''' if uuidutils.is_uuid_like(image_identifier): try: image_id = self.client().images.get(image_identifier).id except sahara_base.APIException as ex: if self.is_not_registered(ex): image_id = self.get_image_id_by_name(image_identifier) else: image_id = self.get_image_id_by_name(image_identifier) return image_id
[docs] def get_image_id_by_name(self, image_identifier): ''' Return an id for the specified image name. :param image_identifier: image name :returns: the id of the requested :image_identifier: :raises: exception.ImageNotFound, exception.PhysicalResourceNameAmbiguity ''' try: filters = {'name': image_identifier} image_list = self.client().images.find(**filters) except sahara_base.APIException as ex: raise exception.Error( _("Error retrieving image list from sahara: " "%s") % six.text_type(ex)) num_matches = len(image_list) if num_matches == 0: LOG.info(_LI("Image %s was not found in sahara images"), image_identifier) raise exception.ImageNotFound(image_name=image_identifier) elif num_matches > 1: LOG.info(_LI("Multiple images %s were found in sahara with name"), image_identifier) raise exception.PhysicalResourceNameAmbiguity( name=image_identifier) else: return image_list[0].id
[docs]class ImageConstraint(constraints.BaseCustomConstraint): expected_exceptions = (exception.ImageNotFound, exception.PhysicalResourceNameAmbiguity,)
[docs] def validate_with_client(self, client, value): client.client_plugin('sahara').get_image_id(value)