added parser to dump yaml

drip_planner2/src/planner/basic_planner.py

@@ -13,6 +13,7 @@ import sys
 import pdb
 import names
 import yaml
+from utils.TOSCA_parser import *
 
 
 
@@ -54,6 +55,11 @@ class BasicPlanner:
         self.template.nodetemplates = node_templates
         
 #        print(yaml.dump(self.template.tpl))
+        
+        tp = TOSCAParser()
+        tp.tosca_template2_yaml(self.template)
+        
+        
         print('------------------')
 #            print(node.get_capabilities().keys)
 
@@ -62,7 +68,7 @@ class BasicPlanner:
 
     def get_missing_requirements(self, node):
         def_type = self.all_nodes[node.type]
-        def_requirements = def_type['requirements']        
+        def_requirements = def_type['requirements']
         missing_requirements = []
         if not node.requirements:
             missing_requirements = def_requirements

drip_planner2/src/utils/TOSCA_parser.py

+import json
+import operator
+import pdb
+import re
+from toscaparser.tosca_template import ToscaTemplate
+from toscaparser.tosca_template import TopologyTemplate
+from toscaparser.elements.nodetype import NodeType
+from toscaparser.nodetemplate import NodeTemplate
+from toscaparser.utils import yamlparser
+import urllib
+import urllib.parse
+import sys
+import pdb
+import names
+import yaml
+
+
+# TOSCA template key names
+SECTIONS = (DEFINITION_VERSION, DEFAULT_NAMESPACE, TEMPLATE_NAME,
+            TOPOLOGY_TEMPLATE, TEMPLATE_AUTHOR, TEMPLATE_VERSION,
+            DESCRIPTION, IMPORTS, DSL_DEFINITIONS, NODE_TYPES,
+            RELATIONSHIP_TYPES, RELATIONSHIP_TEMPLATES,
+            CAPABILITY_TYPES, ARTIFACT_TYPES, DATA_TYPES, INTERFACE_TYPES,
+            POLICY_TYPES, GROUP_TYPES, REPOSITORIES,INPUTS,NODE_TEMPLATES,
+            OUTPUTS,GROUPS,SUBSTITUION_MAPPINGS,POLICIES,TYPE,REQUIREMENTS) = \
+           ('tosca_definitions_version', 'tosca_default_namespace',
+            'template_name', 'tosca_template', 'template_author',
+            'template_version', 'description', 'imports', 'dsl_definitions',
+            'node_types', 'relationship_types', 'relationship_templates',
+            'capability_types', 'artifact_types', 'data_types',
+            'interface_types', 'policy_types', 'group_types', 'repositories',
+            'inputs','node_templates','outputs','groups','substitution_mappings',
+            'policies','type','requirements')    
+                
+class TOSCAParser:
+    
+    def tosca_template2_yaml(self, tosca_template):
+        
+#        print(type(tosca_template.tpl['topology_template']['node_templates']['mysql']))
+#        print((tosca_template.tpl['topology_template']['node_templates']))
+        
+        topology_dict = {}
+        topology_dict[DEFINITION_VERSION] = tosca_template.version
+        topology_dict[IMPORTS] = tosca_template._tpl_imports()
+        topology_dict[DESCRIPTION] = tosca_template.description
+        topology_dict[TOPOLOGY_TEMPLATE] = {}
+        topology_dict[TOPOLOGY_TEMPLATE][NODE_TEMPLATES] = {}
+        node_templates = tosca_template.topology_template.nodetemplates
+        for node_template in node_templates:
+            node_template_dict = self.get_node_template_dict(node_template)
+            topology_dict[TOPOLOGY_TEMPLATE][NODE_TEMPLATES][node_template.name] = node_template_dict
+            
+        
+        yaml.Dumper.ignore_aliases = lambda *args : True
+        print(yaml.dump(topology_dict,default_flow_style=False))
+        
+        
+    def get_node_template_dict(self,node_template):
+        node_template_dict = {}
+        node_template_dict[TYPE] = node_template.type
+        node_template_dict[REQUIREMENTS] = {}
+#        requirement_list = []
+        node_template_dict[REQUIREMENTS] = node_template.requirements
+            
+        return node_template_dict
\ No newline at end of file