#
# 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 itertools
from heat.common import exception
from heat.engine import attributes
from heat.engine import status
[docs]
class StackDefinition(object):
"""Class representing the definition of a Stack, but not its current state.
This is the interface through which template functions will access data
about the stack definition, including the template and current values of
resource reference IDs and attributes.
This API can be considered stable by third-party Template or Function
plugins, and no part of it should be changed or removed without an
appropriate deprecation process.
"""
def __init__(self, context, template, stack_identifier, resource_data,
parent_info=None):
self._context = context
self._template = template
self._resource_data = {} if resource_data is None else resource_data
self._parent_info = parent_info
self._zones = None
self.parameters = template.parameters(stack_identifier,
template.env.params,
template.env.param_defaults)
self._resource_defns = None
self._resources = {}
self._output_defns = None
[docs]
def clone_with_new_template(self, new_template, stack_identifier,
clear_resource_data=False):
"""Create a new StackDefinition with a different template."""
res_data = {} if clear_resource_data else dict(self._resource_data)
return type(self)(self._context, new_template, stack_identifier,
res_data, self._parent_info)
@property
def t(self):
"""The stack's template."""
return self._template
@property
def env(self):
"""The stack's environment."""
return self._template.env
def _load_rsrc_defns(self):
self._resource_defns = self._template.resource_definitions(self)
[docs]
def resource_definition(self, resource_name):
"""Return the definition of the given resource."""
if self._resource_defns is None:
self._load_rsrc_defns()
return self._resource_defns[resource_name]
[docs]
def enabled_rsrc_names(self):
"""Return the set of names of all enabled resources in the template."""
if self._resource_defns is None:
self._load_rsrc_defns()
return set(self._resource_defns)
def _load_output_defns(self):
self._output_defns = self._template.outputs(self)
[docs]
def output_definition(self, output_name):
"""Return the definition of the given output."""
if self._output_defns is None:
self._load_output_defns()
return self._output_defns[output_name]
[docs]
def enabled_output_names(self):
"""Return the set of names of all enabled outputs in the template."""
if self._output_defns is None:
self._load_output_defns()
return set(self._output_defns)
[docs]
def all_rsrc_names(self):
"""Return the set of names of all resources in the template.
This includes resources that are disabled due to false conditionals.
"""
if hasattr(self._template, 'RESOURCES'):
return set(self._template.get(self._template.RESOURCES,
self._resource_defns or []))
else:
return self.enabled_rsrc_names()
[docs]
def all_resource_types(self):
"""Return the set of types of all resources in the template."""
if self._resource_defns is None:
self._load_rsrc_defns()
return set(self._resource_defns[res].resource_type
for res in self._resource_defns)
[docs]
def get_availability_zones(self):
"""Return the list of Nova availability zones."""
if self._zones is None:
nova = self._context.clients.client('nova')
zones = nova.availability_zones.list(detailed=False)
self._zones = [zone.zoneName for zone in zones]
return self._zones
def __contains__(self, resource_name):
"""Return True if the given resource name is present and enabled."""
if self._resource_defns is not None:
return resource_name in self._resource_defns
else:
# In Cfn templates, we need to know whether Ref refers to a
# resource or a parameter in order to parse the resource
# definitions
return resource_name in self._template[self._template.RESOURCES]
def __getitem__(self, resource_name):
"""Return a proxy for the given resource."""
if resource_name not in self._resources:
res_proxy = ResourceProxy(resource_name,
self.resource_definition(resource_name),
self._resource_data.get(resource_name))
self._resources[resource_name] = res_proxy
return self._resources[resource_name]
@property
def parent_resource(self):
"""Return a proxy for the parent resource.
Returns None if the stack is not a provider stack for a
TemplateResource.
"""
return self._parent_info
[docs]
class ResourceProxy(status.ResourceStatus):
"""A lightweight API for essential data about a resource.
This is the interface through which template functions will access data
about particular resources in the stack definition, such as the resource
definition and current values of reference IDs and attributes.
Resource proxies for some or all resources in the stack will potentially be
loaded for every check resource operation, so it is essential that this API
is implemented efficiently, using only the data received over RPC and
without reference to the resource data stored in the database.
This API can be considered stable by third-party Template or Function
plugins, and no part of it should be changed or removed without an
appropriate deprecation process.
"""
__slots__ = ('name', '_definition', '_resource_data')
def __init__(self, name, definition, resource_data):
self.name = name
self._definition = definition
self._resource_data = resource_data
@property
def t(self):
"""The resource definition."""
return self._definition
def _res_data(self):
assert self._resource_data is not None, "Resource data not available"
return self._resource_data
@property
def attributes_schema(self):
"""A set of the valid top-level attribute names.
This is provided for backwards-compatibility for functions that require
a container with all of the valid attribute names in order to validate
the template. Other operations on it are invalid because we don't
actually have access to the attributes schema here; hence we return a
set instead of a dict.
"""
return set(self._res_data().attribute_names())
@property
def external_id(self):
"""The external ID of the resource."""
return self._definition.external_id()
@property
def state(self):
"""The current state (action, status) of the resource."""
return self.action, self.status
@property
def action(self):
"""The current action of the resource."""
if self._resource_data is None:
return self.INIT
return self._resource_data.action
@property
def status(self):
"""The current status of the resource."""
if self._resource_data is None:
return self.COMPLETE
return self._resource_data.status
[docs]
def FnGetRefId(self):
"""For the intrinsic function get_resource."""
if self._resource_data is None:
return self.name
return self._resource_data.reference_id()
[docs]
def FnGetAtt(self, attr, *path):
"""For the intrinsic function get_attr."""
if path:
attr = (attr,) + path
try:
return self._res_data().attribute(attr)
except KeyError:
raise exception.InvalidTemplateAttribute(resource=self.name,
key=attr)
[docs]
def FnGetAtts(self):
"""For the intrinsic function get_attr when getting all attributes.
:returns: a dict of all of the resource's attribute values, excluding
the "show" attribute.
"""
all_attrs = self._res_data().attributes()
return dict((k, v) for k, v in all_attrs.items()
if k != attributes.SHOW_ATTR)
[docs]
def update_resource_data(stack_definition, resource_name, resource_data):
"""Store new resource state data for the specified resource.
This function enables the legacy (non-convergence) path to store updated
NodeData as resources are created/updated in a single StackDefinition
that lasts for the entire lifetime of the stack operation.
"""
stack_definition._resource_data[resource_name] = resource_data
stack_definition._resources.pop(resource_name, None)
# Clear the cached dep_attrs for any resource or output that directly
# depends on the resource whose data we are updating. This ensures that if
# any of the data we just updated is referenced in the path of a get_attr
# function, future calls to dep_attrs() will reflect this new data.
res_defns = stack_definition._resource_defns or {}
op_defns = stack_definition._output_defns or {}
all_defns = itertools.chain(res_defns.values(),
op_defns.values())
for defn in all_defns:
if resource_name in defn.required_resource_names():
defn._all_dep_attrs = None
[docs]
def add_resource(stack_definition, resource_definition):
"""Insert the given resource definition into the stack definition.
Add the resource to the template and store any temporary data.
"""
resource_name = resource_definition.name
stack_definition._resources.pop(resource_name, None)
stack_definition._resource_data.pop(resource_name, None)
stack_definition.t.add_resource(resource_definition)
if stack_definition._resource_defns is not None:
stack_definition._resource_defns[resource_name] = resource_definition
[docs]
def remove_resource(stack_definition, resource_name):
"""Remove the named resource from the stack definition.
Remove the resource from the template and eliminate references to it.
"""
stack_definition.t.remove_resource(resource_name)
if stack_definition._resource_defns is not None:
stack_definition._resource_defns.pop(resource_name, None)
stack_definition._resource_data.pop(resource_name, None)
stack_definition._resources.pop(resource_name, None)
