121 lines
4.3 KiB
Python
121 lines
4.3 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.
|
|
|
|
import sys
|
|
|
|
from heatclient import exc as heatException
|
|
from oslo_log import log as logging
|
|
from oslo_utils import timeutils
|
|
|
|
from tacker.common import clients
|
|
from tacker.extensions import vnfm
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class HeatClient(object):
|
|
def __init__(self, auth_attr, region_name=None):
|
|
# context, password are unused
|
|
self.heat = clients.OpenstackClients(auth_attr, region_name).heat
|
|
self.stacks = self.heat.stacks
|
|
self.resource_types = self.heat.resource_types
|
|
self.resources = self.heat.resources
|
|
|
|
def _stack_ids(self, stack_id):
|
|
filters = {"owner_id": stack_id,
|
|
"show_nested": True}
|
|
|
|
for stack in self.stacks.list(**{"filters": filters}):
|
|
yield stack.id
|
|
if stack.parent and stack.parent == stack_id:
|
|
for x in self._stack_ids(stack.id):
|
|
yield x
|
|
|
|
def find_stack(self, **kwargs):
|
|
stack = None
|
|
for sub_stack in self.stacks.list(**{"filters": kwargs}):
|
|
stack = sub_stack
|
|
return stack
|
|
|
|
def get_cooldown(self, stack_id, rsc_name):
|
|
metadata = self.resource_metadata(stack_id, rsc_name)
|
|
if 'cooldown_end' in metadata:
|
|
now = timeutils.utcnow()
|
|
cooldown_end = timeutils.parse_strtime(
|
|
next(iter(metadata['cooldown_end'].keys())))
|
|
return max(0, timeutils.delta_seconds(now, cooldown_end))
|
|
return None
|
|
|
|
def create(self, fields):
|
|
fields = fields.copy()
|
|
fields['disable_rollback'] = True
|
|
if 'password' in fields.get('template', {}):
|
|
fields['password'] = fields['template']['password']
|
|
|
|
fields.setdefault('timeout_mins')
|
|
|
|
try:
|
|
return self.stacks.create(**fields)
|
|
except heatException.HTTPException:
|
|
type_, value, tb = sys.exc_info()
|
|
raise vnfm.HeatClientException(msg=value)
|
|
|
|
def delete(self, stack_id):
|
|
try:
|
|
self.stacks.delete(stack_id)
|
|
except heatException.HTTPNotFound:
|
|
LOG.warning("Stack %(stack)s created by service chain driver is "
|
|
"not found at cleanup", {'stack': stack_id})
|
|
|
|
def get(self, stack_id):
|
|
return self.stacks.get(stack_id)
|
|
|
|
def get_stack_nested_depth(self, stack_id):
|
|
stack_ids = self._stack_ids(stack_id)
|
|
if stack_ids:
|
|
return len(list(stack_ids))
|
|
|
|
return 0
|
|
|
|
def update(self, stack_id, **kwargs):
|
|
try:
|
|
return self.stacks.update(stack_id, **kwargs)
|
|
except heatException.HTTPException:
|
|
type_, value, tb = sys.exc_info()
|
|
raise vnfm.HeatClientException(msg=value)
|
|
|
|
def resource_attr_support(self, resource_name, property_name):
|
|
resource = self.resource_types.get(resource_name)
|
|
return property_name in resource['attributes']
|
|
|
|
def resource_get_list(self, stack_id, nested_depth=0):
|
|
return self.heat.resources.list(stack_id,
|
|
nested_depth=nested_depth)
|
|
|
|
def resource_signal(self, stack_id, rsc_name):
|
|
return self.heat.resources.signal(stack_id, rsc_name)
|
|
|
|
def resource_get(self, stack_id, rsc_name):
|
|
return self.heat.resources.get(stack_id, rsc_name)
|
|
|
|
def resource_event_list(self, stack_id, rsc_name, **kwargs):
|
|
return self.heat.events.list(stack_id, rsc_name, **kwargs)
|
|
|
|
def resource_metadata(self, stack_id, rsc_name):
|
|
return self.heat.resources.metadata(stack_id, rsc_name)
|
|
|
|
def resource_mark_unhealthy(self, stack_id, resource_name,
|
|
mark_unhealthy, resource_status_reason):
|
|
return self.heat.resources.mark_unhealthy(stack_id, resource_name,
|
|
mark_unhealthy,
|
|
resource_status_reason)
|