heat/heat/engine/clients/os/ironic.py
ricolin 5b35225d16 Add ironic client plugin support
Change-Id: I2f3ee94424c4dab75fbcef6f8b32e565b45684e4
Task: 36285
2020-04-09 14:14:38 +08:00

83 lines
2.9 KiB
Python

#
# 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.
from ironicclient.common.apiclient import exceptions as ic_exc
from ironicclient.v1 import client as ironic_client
from oslo_config import cfg
from heat.common import exception
from heat.engine.clients import client_plugin
from heat.engine import constraints
CLIENT_NAME = 'ironic'
class IronicClientPlugin(client_plugin.ClientPlugin):
service_types = [BAREMETAL] = ['baremetal']
IRONIC_API_VERSION = 1.58
max_ironic_api_microversion = cfg.CONF.max_ironic_api_microversion
max_microversion = max_ironic_api_microversion if (
max_ironic_api_microversion is not None and (
IRONIC_API_VERSION > max_ironic_api_microversion)
) else IRONIC_API_VERSION
def _create(self):
interface = self._get_client_option(CLIENT_NAME, 'endpoint_type')
args = {
'interface': interface,
'service_type': self.BAREMETAL,
'session': self.context.keystone_session,
'region_name': self._get_region_name(),
'os_ironic_api_version': self.max_microversion
}
client = ironic_client.Client(**args)
return client
def is_not_found(self, ex):
return isinstance(ex, ic_exc.NotFound)
def is_over_limit(self, ex):
return isinstance(ex, ic_exc.RequestEntityTooLarge)
def is_conflict(self, ex):
return isinstance(ex, ic_exc.Conflict)
def _get_rsrc_name_or_id(self, value, entity, entity_msg):
entity_client = getattr(self.client(), entity)
try:
return entity_client.get(value).uuid
except ic_exc.NotFound:
# Ironic cli will find the value either is name or id,
# so no need to call list() here.
raise exception.EntityNotFound(entity=entity_msg,
name=value)
def get_portgroup(self, value):
return self._get_rsrc_name_or_id(value, entity='portgroup',
entity_msg='PortGroup')
def get_node(self, value):
return self._get_rsrc_name_or_id(value, entity='node',
entity_msg='Node')
class PortGroupConstraint(constraints.BaseCustomConstraint):
resource_client_name = CLIENT_NAME
resource_getter_name = 'get_portgroup'
class NodeConstraint(constraints.BaseCustomConstraint):
resource_client_name = CLIENT_NAME
resource_getter_name = 'get_node'