1 # Licensed under the Apache License, Version 2.0 (the "License"); you may
2 # not use this file except in compliance with the License. You may obtain
3 # a copy of the License at
5 # http://www.apache.org/licenses/LICENSE-2.0
7 # Unless required by applicable law or agreed to in writing, software
8 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
9 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
10 # License for the specific language governing permissions and limitations
13 from toscaparser.common.exception import ExceptionCollector
14 from toscaparser.common.exception import InvalidTypeError
15 from toscaparser.elements.attribute_definition import AttributeDef
16 from toscaparser.elements.entity_type import EntityType
17 from toscaparser.elements.property_definition import PropertyDef
18 from toscaparser.unsupportedtype import UnsupportedType
21 class StatefulEntityType(EntityType):
22 '''Class representing TOSCA states.'''
24 interfaces_node_lifecycle_operations = ['create',
28 interfaces_relationship_configure_operations = ['post_configure_source',
29 'post_configure_target',
33 def __init__(self, entitytype, prefix, custom_def=None):
34 entire_entitytype = entitytype
35 if UnsupportedType.validate_type(entire_entitytype):
38 if entitytype.startswith(self.TOSCA + ":"):
39 entitytype = entitytype[(len(self.TOSCA) + 1):]
40 entire_entitytype = prefix + entitytype
41 if not entitytype.startswith(self.TOSCA):
42 entire_entitytype = prefix + entitytype
43 if entire_entitytype in list(self.TOSCA_DEF.keys()):
44 self.defs = self.TOSCA_DEF[entire_entitytype]
45 entitytype = entire_entitytype
46 elif custom_def and entitytype in list(custom_def.keys()):
47 self.defs = custom_def[entitytype]
50 ExceptionCollector.appendException(
51 InvalidTypeError(what=entitytype))
52 self.type = entitytype
54 def get_properties_def_objects(self):
55 '''Return a list of property definition objects.'''
57 props = self.get_definition(self.PROPERTIES)
59 for prop, schema in props.items():
60 properties.append(PropertyDef(prop, None, schema))
63 def get_properties_def(self):
64 '''Return a dictionary of property definition name-object pairs.'''
65 return {prop.name: prop
66 for prop in self.get_properties_def_objects()}
68 def get_property_def_value(self, name):
69 '''Return the property definition associated with a given name.'''
70 props_def = self.get_properties_def()
71 if props_def and name in props_def.keys():
72 return props_def[name].value
74 def get_attributes_def_objects(self):
75 '''Return a list of attribute definition objects.'''
76 attrs = self.get_value(self.ATTRIBUTES, parent=True)
78 return [AttributeDef(attr, None, schema)
79 for attr, schema in attrs.items()]
82 def get_attributes_def(self):
83 '''Return a dictionary of attribute definition name-object pairs.'''
84 return {attr.name: attr
85 for attr in self.get_attributes_def_objects()}
87 def get_attribute_def_value(self, name):
88 '''Return the attribute definition associated with a given name.'''
89 attrs_def = self.get_attributes_def()
90 if attrs_def and name in attrs_def.keys():
91 return attrs_def[name].value