Add dashboard template
[scripts.git] / elastic / generate-config.py
index e5e8547..49a6209 100755 (executable)
@@ -1,32 +1,34 @@
 #!/usr/bin/env python3
 
+import json
 import pystache
 
 #print(pystache.render('Hi {{person}}!', {'person': 'Mom'}))
 
-tests = [
-    'test_encrypt_raw_10M',
-     'test_encrypt_raw_100k',
-    # 'test_encrypt_raw_10k',
-    # 'test_decrypt_raw_100k',
-    # 'test_decrypt_raw_10M',
-    # 'test_decrypt_raw_10k',
-    # 'test_decrypt_raw_1M',
-    # 'test_decrypt_raw_500k',
-    # 'test_encrypt_raw_1M',
-    # 'test_encrypt_raw_500k',
-    # 'test_initialization',
-    # 'test_async_create_1000_10k',
-    # 'test_async_create_100_100k',
-    # 'test_create_1000_10k',
-    # 'test_create_100_100k',
-    # 'test_create_20_500k',
-    # 'test_decrypt_doc_100k',
-    # 'test_decrypt_doc_10M',
-    # 'test_decrypt_doc_10k',
-    #'test_decrypt_doc_1M'
-    ]
+tests_json = '{ "tests":[\
+    {"name": "test_encrypt_raw_10M"},\
+    {"name": "test_encrypt_raw_100k"},\
+    {"name": "test_encrypt_raw_10k"},\
+    {"name": "test_decrypt_raw_100k"},\
+    {"name": "test_decrypt_raw_10M"},\
+    {"name": "test_decrypt_raw_10k"},\
+    {"name": "test_decrypt_raw_1M"},\
+    {"name": "test_decrypt_raw_500k"},\
+    {"name": "test_encrypt_raw_1M"},\
+    {"name": "test_encrypt_raw_500k"},\
+    {"name": "test_initialization"},\
+    {"name": "test_async_create_1000_10k"},\
+    {"name": "test_async_create_100_100k"},\
+    {"name": "test_create_1000_10k"},\
+    {"name": "test_create_100_100k"},\
+    {"name": "test_create_20_500k"},\
+    {"name": "test_decrypt_doc_100k"},\
+    {"name": "test_decrypt_doc_10M"},\
+    {"name": "test_decrypt_doc_10k"},\
+    {"name": "test_decrypt_doc_1M"}\
+    ]}'
 
+tests = json.loads(tests_json)
 
 
 # Generate visualizations
@@ -34,16 +36,17 @@ def generate_visualizations():
     dir = './visualization/'
     template = open(dir + 'template.mustache', 'r').read()
 
-    for test in tests:
-        out_file = dir + test + '.json'
+    # test_names = tests.
+    for test in tests['tests']:
+        test_name = test['name']
+        out_file = dir + test_name + '.json'
         print('Generating ' + out_file)
         context = {
-           'title': test,
-           'query': "commit_info.project:soledad AND commit_info.branch='master' AND machine_info.host='weasel' AND name='" + test + "'"
+           'title': test_name,
+           'query': "commit_info.project:soledad AND commit_info.branch='master' AND machine_info.host='weasel' AND name='" + test_name + "'"
         }
 
         rendered = pystache.render(template, context)
-        #print(context)
 
         out = open(out_file, 'w')
         out.write(rendered)
@@ -53,16 +56,18 @@ def generate_visualizations():
 def generate_dashboards():
     dir = './dashboard/'
     template = open(dir + 'template.mustache', 'r').read()
-    out_file = dir + test + '.json'
+    out_file = dir + 'soledad-benchmarks.json'
     print('Generating ' + out_file)
 
-    item = q
+    rendered = pystache.render(template, tests)
+    #print(rendered)
 
-    for test in tests:
-        dashboards = dashboards + ''
+    out = open(out_file, 'w')
+    out.write(rendered)
+    out.flush()
 
-    rendered = pystache.render(template, context)
-    print(rendered)
 
+# Main
+
+generate_dashboards()
 generate_visualizations()
-#generate_dashboards()