1 files changed, 40 insertions, 0 deletions
diff --git a/app/toscalib/utils/tosca_export.py b/app/toscalib/utils/tosca_export.py
new file mode 100644
index 0000000..c4399ef
--- /dev/null
+++ b/app/toscalib/utils/tosca_export.py
@@ -0,0 +1,40 @@
+#Author: Shu Shi
+#emaiL: shushi@research.att.com
+
+
+import yaml
+import re
+from toscalib.templates.constant import *
+
+def _yaml_export(filename, content):    
+    out_str = yaml.safe_dump(content,  default_flow_style=False, width=float("inf"))
+    out_str = re.sub(YMO_PREFIX, '', out_str)
+    if filename == 'WEB':
+        return out_str
+    with open(filename, 'w') as outfile:
+        outfile.write(out_str)
+    return out_str
+
+def _heat_export(filename, content):
+    heat_ver = YMO_PREFIX + 'heat_template_version'
+    heat_content = content[YMO_TOPOLOGY]
+    if heat_content is None:
+        heat_content = {}
+    heat_content[heat_ver] = '2013-05-23'
+#    if heat_content.has_key(YMO_TOPO_OUTPUTS):
+    if YMO_TOPO_OUTPUTS in heat_content:
+        heat_content.pop(YMO_TOPO_OUTPUTS)
+        
+    out_str = yaml.dump(heat_content,  default_flow_style=False)
+    out_str = re.sub(YMO_TOPO_INPUTS, YMO_PREFIX+'parameters', out_str)
+    out_str = re.sub(YMO_TOPO_NODE_TEMPLATES, YMO_PREFIX+'resources', out_str)
+    
+    out_str = re.sub('get_input', 'get_param', out_str)
+    out_str = re.sub('get_attribute', 'get_attr', out_str)
+    out_str = re.sub('get_id', 'get_resource', out_str)
+    out_str = re.sub('get_property', 'get_attr', out_str)
+    out_str = re.sub('type: integer', 'type: number', out_str)
+        
+    out_str = re.sub(YMO_PREFIX, '', out_str)
+    with open(filename, 'w') as outfile:
+        outfile.write(out_str)