Commit f1e1e884 authored by Spiros Koulouzis's avatar Spiros Koulouzis

added parser to dump yaml

parent d66a4806
...@@ -13,6 +13,7 @@ import sys ...@@ -13,6 +13,7 @@ import sys
import pdb import pdb
import names import names
import yaml import yaml
from utils.TOSCA_parser import *
...@@ -54,6 +55,11 @@ class BasicPlanner: ...@@ -54,6 +55,11 @@ class BasicPlanner:
self.template.nodetemplates = node_templates self.template.nodetemplates = node_templates
# print(yaml.dump(self.template.tpl)) # print(yaml.dump(self.template.tpl))
tp = TOSCAParser()
tp.tosca_template2_yaml(self.template)
print('------------------') print('------------------')
# print(node.get_capabilities().keys) # print(node.get_capabilities().keys)
......
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
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment