DCAE-D tosca-lab initial commit

DCAE-D tosca-lab initial commit

Change-Id: Ia42934ce7c75abe05fa106585c9334c8b048ee36
Issue-ID: SDC-1218
Signed-off-by: Stone, Avi (as206k) <as206k@att.com>

1 files changed, 99 insertions, 0 deletions

diff --git a/app/map_sup_enrich_compose.py b/app/map_sup_enrich_compose.py
new file mode 100644
index 0000000..815a258
--- /dev/null
+++ b/app/map_sup_enrich_compose.py
@@ -0,0 +1,99 @@
+#Author: Shu Shi
+#emaiL: shushi@research.att.com
+
+from toscalib.tosca_workbook import ToscaWorkBook
+from toscalib.tosca_builder import ToscaBuilder
+
+import getopt, sys, json, logging
+
+def usage():
+    print('OPTIONS:')
+    print('\t-h|--help: print this help message')
+    print('\t-i|--input: The home folder where all spec files are')
+    print('\t-o|--output: the output file name')
+    print('\t-v|--value: the json value file')
+
+    
+def main():
+    try:
+        opts, args = getopt.getopt(sys.argv[1:], "hi:o:v:", ["help", "input=", "output=", "value="])
+    except getopt.GetoptError as err:
+        # print help information and exit:
+        logging.error( str(err)) # will print something like "option -a not recognized"
+        usage()
+        sys.exit(2)
+
+    spec_prefix = None
+    output_file = None
+    value_file = None
+    
+    for o, a in opts:
+        if  o in ("-h", "--help"):
+            usage()
+            sys.exit()
+        elif o in ("-i", "--input"):
+            spec_prefix = a
+        elif o in ("-o", "--output"):
+            output_file = a
+        elif o in ("-v", "--value"):
+            value_file = a
+        else:
+            logging.error( 'Unrecognized option: ' + o)
+            usage()
+            sys.exit(2)
+            
+    if spec_prefix is None or output_file is None:
+        logging.error( 'Incorrect arguments!')
+        usage()
+        sys.exit(2)
+
+    model_prefix = './data/tosca_model'
+    meta_model = './data/meta_model/meta_tosca_schema.yaml'
+        
+    for ms in ['map', 'enrich', 'supplement']:
+            
+        builder = ToscaBuilder()
+        
+        builder.import_schema(meta_model)
+        builder.import_spec(spec_prefix+'/dcae-event-proc/dcae-event-proc-cdap-' + ms+ '\\' + ms+ '_spec.json')
+        builder.create_node_type()
+        builder.export_schema(model_prefix+'/' + ms + '/schema.yaml')
+        builder.import_schema(model_prefix+'/' + ms + '/schema.yaml')
+        builder.create_model(ms)
+        builder.export_model(model_prefix+'/' + ms + '/template.yaml')
+        builder.create_translate(ms)
+        builder.export_translation(model_prefix+'/' + ms + '/translate.yaml')
+        
+    workbook = ToscaWorkBook()
+    
+    workbook._import_dir(model_prefix)
+    workbook._import_dir('./data/shared_model/')
+    workbook._use('map','NO_PREFIX')
+    workbook._use('supplement','NO_PREFIX')
+    workbook._use('enrich','NO_PREFIX')
+    
+    if value_file is not None: 
+        try: 
+            with open(value_file) as data_file:
+                data = json.load(data_file)
+                for ms in ['map', 'enrich', 'supplement']:
+#                    if data.has_key(ms):
+                    if ms in data:
+                        prop_sec = data[ms]
+                        for key in prop_sec.keys():
+                            workbook._assign(ms, key, prop_sec[key])
+        except err :
+            logging.error( "Unable to read " +value_file)
+            logging.error( str(err))
+    workbook._add_shared_node([{'dcae.capabilities.cdapHost':'cdap_host'}, {'dcae.capabilities.dockerHost': 'docker_host'}, {'dcae.capabilities.composition.host': 'composition_virtual'}])
+    
+    workbook._assign('supplement', 'stream_publish_0', 'map')
+    workbook._assign('enrich', 'stream_publish_0', 'supplement')
+         
+    workbook.tran_db = workbook.db
+         
+    workbook._export_yaml('event_proc.yaml', 'no_expand,main')
+    workbook._export_yaml(output_file, 'cloudify,main')
+    
+if __name__ == "__main__":
+    main()
\ No newline at end of file