deb-heat/heat/engine/translation.py
Peter Razumovsky dfad3ad253 Change type of raised exception in translation
Change exception type to ResourcePropertyConflict for
conflicted properties.

Change-Id: I171430ca9b90d6878e5fcaaccbafb794b5d35218
2016-04-13 16:37:33 +03:00

267 lines
12 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 oslo_utils import encodeutils
from heat.common import exception
from heat.common.i18n import _
from heat.engine.cfn import functions as cfn_funcs
from heat.engine import function
from heat.engine.hot import functions as hot_funcs
from heat.engine import properties
class TranslationRule(object):
"""Translating mechanism one properties to another.
Mechanism uses list of rules, each defines by this class, and can be
executed. Working principe: during resource creating after properties
defining resource take list of rules, specified by method
translation_rules, which should be overloaded for each resource, if it's
needed, and execute each rule using translate_properties method. Next
operations are allowed:
- ADD. This rule allows to add some value to list-type properties. Only
list-type values can be added to such properties. Using for other
cases is prohibited and will be returned with error.
- REPLACE. This rule allows to replace some property value to another. Used
for all types of properties. Note, that if property has list type,
then value will be replaced for all elements of list, where it
needed. If element in such property must be replaced by value of
another element of this property, value_name must be defined.
- DELETE. This rule allows to delete some property. If property has list
type, then deleting affects value in all list elements.
- RESOLVE. This rule allows to resolve some property using client and
the finder function. Finders may require an additional entity key.
"""
RULE_KEYS = (ADD, REPLACE,
DELETE, RESOLVE) = ('Add', 'Replace',
'Delete', 'Resolve')
def __init__(self, properties, rule, translation_path, value=None,
value_name=None, value_path=None, client_plugin=None,
finder=None, entity=None):
"""Add new rule for translating mechanism.
:param properties: properties of resource
:param rule: rule from RULE_KEYS
:param translation_path: list with path to property, which value will
be affected in rule.
:param value: value which will be involved in rule
:param value_name: value_name which used for replacing properties
inside list-type properties.
:param value_path: path to value, which should be used for translation.
:param client_plugin: client plugin that would be used to resolve
:param finder: finder method of the client plugin
:param entity: some generic finders require an entity to resolve ex.
neutron finder function.
"""
self.properties = properties
self.rule = rule
self.translation_path = translation_path
self.value = value or None
self.value_name = value_name
self.value_path = value_path
self.client_plugin = client_plugin
self.finder = finder
self.entity = entity
self.validate()
def validate(self):
if self.rule not in self.RULE_KEYS:
raise ValueError(_('There is no rule %(rule)s. List of allowed '
'rules is: %(rules)s.') % {
'rule': self.rule,
'rules': ', '.join(self.RULE_KEYS)})
elif not isinstance(self.properties, properties.Properties):
raise ValueError(_('Properties must be Properties type. '
'Found %s.') % type(self.properties))
elif not isinstance(self.translation_path, list):
raise ValueError(_('translation_path should be a list with path '
'instead of %s.') % type(self.translation_path))
elif len(self.translation_path) == 0:
raise ValueError(_('translation_path must be non-empty list with '
'path.'))
elif self.value_name and self.rule != self.REPLACE:
raise ValueError(_('Use value_name only for replacing list '
'elements.'))
elif self.rule == self.ADD and not isinstance(self.value, list):
raise ValueError(_('value must be list type when rule is Add.'))
elif (self.rule == self.RESOLVE and not (self.client_plugin or
self.finder)):
raise ValueError(_('client_plugin and finder should be specified '
'for Resolve rule'))
def execute_rule(self, client_resolve=True):
try:
(translation_key,
translation_data) = self._get_data_from_source_path(
self.translation_path)
if self.value_path:
(value_key, value_data) = self._get_data_from_source_path(
self.value_path)
value = (value_data[value_key]
if value_data and value_data.get(value_key)
else self.value)
else:
(value_key, value_data) = None, None
value = self.value
except AttributeError:
return
if (translation_data is None or
(self.rule not in (self.DELETE, self.RESOLVE) and
(value is None and self.value_name is None and
(value_data is None or value_data.get(value_key) is None)))):
return
if self.rule == TranslationRule.ADD:
self._exec_add(translation_key, translation_data, value)
elif self.rule == TranslationRule.REPLACE:
self._exec_replace(translation_key, translation_data,
value_key, value_data, value)
elif self.rule == TranslationRule.RESOLVE and client_resolve:
self._exec_resolve(translation_key, translation_data)
elif self.rule == TranslationRule.DELETE:
self._exec_delete(translation_key, translation_data)
def _get_data_from_source_path(self, path):
def get_props(props, key):
props = props.get(key)
if props.schema.schema is not None:
keys = list(props.schema.schema)
schemata = dict((k, props.schema.schema[k])
for k in keys)
props = dict((k, properties.Property(s, k))
for k, s in schemata.items())
return props
def resolve_param(param):
"""Check whether if given item is param and resolve, if it is."""
# NOTE(prazumovsky): If property uses removed in HOT function,
# we should not translate it for correct validating and raising
# validation error.
if isinstance(param, hot_funcs.Removed):
raise AttributeError(_('Property uses removed function.'))
if isinstance(param, (hot_funcs.GetParam, cfn_funcs.ParamRef)):
try:
return function.resolve(param)
except exception.UserParameterMissing as ex:
# We can't resolve parameter now. Abort translation.
err_msg = encodeutils.exception_to_unicode(ex)
raise AttributeError(
_('Can not resolve parameter '
'due to: %s') % err_msg)
elif isinstance(param, list):
return [resolve_param(param_item) for param_item in param]
else:
return param
source_key = path[0]
data = self.properties.data
props = self.properties.props
for key in path:
if isinstance(data, list):
source_key = key
elif data.get(key) is not None:
# NOTE(prazumovsky): There's no need to resolve other functions
# because we can translate all function to another path. But if
# list or map type property equals to get_param function, need
# to resolve it for correct translating.
data[key] = resolve_param(data[key])
if isinstance(data[key], (dict, list)):
data = data[key]
props = get_props(props, key)
else:
source_key = key
elif data.get(key) is None:
if (self.rule in (TranslationRule.DELETE,
TranslationRule.RESOLVE) or
(self.rule == TranslationRule.REPLACE and
self.value_name)):
return None, None
elif props.get(key).type() == properties.Schema.LIST:
data[key] = []
elif props.get(key).type() == properties.Schema.MAP:
data[key] = {}
else:
source_key = key
continue
data = data.get(key)
props = get_props(props, key)
return source_key, data
def _exec_add(self, translation_key, translation_data, value):
if isinstance(translation_data, list):
translation_data.extend(value)
else:
raise ValueError(_('Add rule must be used only for '
'lists.'))
def _exec_replace(self, translation_key, translation_data,
value_key, value_data, value):
if isinstance(translation_data, list):
for item in translation_data:
if item.get(self.value_name) and item.get(translation_key):
raise exception.ResourcePropertyConflict(
props=[translation_key, self.value_name])
elif item.get(self.value_name) is not None:
item[translation_key] = item[self.value_name]
del item[self.value_name]
elif value is not None:
item[translation_key] = value
else:
if (translation_data and translation_data.get(translation_key) and
value_data and value_data.get(value_key)):
raise exception.ResourcePropertyConflict(
props=[translation_key, value_key])
translation_data[translation_key] = value
# If value defined with value_path, need to delete value_path
# property data after it's replacing.
if value_data and value_data.get(value_key):
del value_data[value_key]
def _exec_resolve(self, translation_key, translation_data):
def resolve_and_find(translation_data, translation_value):
if isinstance(translation_value, cfn_funcs.ResourceRef):
return
if isinstance(translation_value, function.Function):
translation_value = function.resolve(translation_value)
if translation_value:
finder = getattr(self.client_plugin, self.finder)
if self.entity:
value = finder(self.entity, translation_value)
else:
value = finder(translation_value)
translation_data[translation_key] = value
if isinstance(translation_data, list):
for item in translation_data:
translation_value = item.get(translation_key)
resolve_and_find(item, translation_value)
else:
translation_value = translation_data.get(translation_key)
resolve_and_find(translation_data, translation_value)
def _exec_delete(self, translation_key, translation_data):
if isinstance(translation_data, list):
for item in translation_data:
if item.get(translation_key) is not None:
del item[translation_key]
else:
del translation_data[translation_key]