summaryrefslogtreecommitdiffstats
path: root/dashboard/dashboard/elastic2kibana/dashboard_assembler.py
blob: 651168bbb1fe477976bfae1f0c2b868fc72d19f1 (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
import json

import utility
from dashboard.common import elastic_access


class DashboardAssembler(object):
    def __init__(self,
                 project,
                 case,
                 family,
                 installer,
                 pod,
                 visualizations,
                 es_url,
                 es_creds):
        super(DashboardAssembler, self).__init__()
        self.project = project
        self.case = case
        self.test_family = family
        self.installer = installer
        self.pod = pod
        self.visualizations = visualizations
        self.es_url = es_url
        self.es_creds = es_creds
        self._assemble()
        self._publish()

    def _assemble(self):
        db = {
            "query": {
                "project_name": self.project,
                "case_name": self.case,
                "installer": self.installer,
                "metric": self.visualizations[0].vis_state_title,
                "pod": self.pod
            },
            "test_family": self.test_family,
            "ids": [visualization.id for visualization in self.visualizations]
        }
        template = utility.env.get_template('dashboard.json')
        self.dashboard = json.loads(template.render(db=db))
        utility.dumps(self.dashboard,
                      ['description',
                       'uiStateJSON',
                       'panelsJSON',
                       'optionsJSON'])
        utility.dumps_2depth(self.dashboard,
                             'kibanaSavedObjectMeta',
                             'searchSourceJSON')
        self.id = self.dashboard['title'].replace(' ', '-').replace('/', '-')
        return self

    def _publish(self):
        elastic_access.publish_kibana(self.es_url,
                                      self.es_creds,
                                      'dashboard',
                                      self.id,
                                      self.dashboard)