summaryrefslogtreecommitdiff
path: root/elastic/generate-config.py
blob: 9d5358c7f8ad43f6ed8c51914ca7912011720b79 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
#!/usr/bin/env python3

import pystache
import requests
import json

URL='https://moose.leap.se:9200'


# Generate visualizations
def generate_visualizations():
    dir = './visualization/'
    template = open(dir + 'template.mustache', 'r').read()

    # test_names = tests.
    for test_name in tests:
        out_file = dir + test_name + '.json'
        print('Generating ' + out_file)
        context = {
           '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)

        # verify we generated valid JSON visualizations
        content = json.loads(rendered)
        visState = content['visState']
        searchSourceJSON = content['kibanaSavedObjectMeta']['searchSourceJSON']
        assert json.loads(visState)
        assert json.loads(searchSourceJSON)

        out = open(out_file, 'w')
        out.write(rendered)
        out.flush()

# Generate Dashboard
def generate_dashboards():
    dir = './dashboard/'
    template = open(dir + 'template.mustache', 'r').read()
    out_file = dir + 'soledad-benchmarks.json'
    print('Generating ' + out_file)

    panels = '['
    count = 0
    for test_name in tests:
        count += 1
        panels += '{\\"id\\":\\"' + test_name + '\\",\\"panelIndex\\":' + str(count) + ',\\"col\\":1,\\"row\\":' + str(count) + ',\\"size_x\\":6,\\"size_y\\":3,\\"type\\":\\"visualization\\"}, '
    panels = panels[:-2]
    panels += ']'

    panels_json = { 'panels_json': panels }
    rendered = pystache.render(template, panels_json)
    #print(rendered)

    out = open(out_file, 'w')
    out.write(rendered)
    out.flush()


# Main

# Get all test names
with open('./query/test_names.json') as f:
    response = requests.get(URL + '/_search', data=f)
    buckets = response.json()['aggregations']['test_names']['buckets']

tests = []
for d in buckets:
    name = d['key']
    tests.append(name)

generate_dashboards()
generate_visualizations()