EntityTemplate has no property of parent_type

[parser.git] / tosca2heat / tosca-parser / toscaparser / entity_template.py

diff --git a/tosca2heat/tosca-parser/toscaparser/entity_template.py b/tosca2heat/tosca-parser/toscaparser/entity_template.py
index 281012b..f416c99 100644 (file)
--- a/tosca2heat/tosca-parser/toscaparser/entity_template.py
+++ b/tosca2heat/tosca-parser/toscaparser/entity_template.py
@@ -81,6 +81,15 @@ class EntityTemplate(object):
     def type(self):
         if self.type_definition:
             return self.type_definition.type
+        else:
+            return None
+
+    @property
+    def parent_type(self):
+        if self.type_definition:
+            return self.type_definition.parent_type
+        else:
+            return None
 
     @property
     def requirements(self):