-
Notifications
You must be signed in to change notification settings - Fork 12
/
model.py
112 lines (94 loc) · 3.68 KB
/
model.py
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
# -*- coding: utf-8 -*-
# (c) 2021 Andreas Motl <andreas@hiveeyes.org>
# License: GNU Affero General Public License, Version 3
import dataclasses
from collections import OrderedDict
from typing import Dict, List, Optional
from urllib.parse import urljoin
from munch import Munch
@dataclasses.dataclass
class GrafanaDataModel:
admin_stats: Optional[Dict] = dataclasses.field(default_factory=dict)
dashboards: Optional[List[Munch]] = dataclasses.field(default_factory=list)
dashboard_list: Optional[List[Munch]] = dataclasses.field(default_factory=list)
datasources: Optional[List[Munch]] = dataclasses.field(default_factory=list)
folders: Optional[List[Munch]] = dataclasses.field(default_factory=list)
organizations: Optional[List[Munch]] = dataclasses.field(default_factory=list)
users: Optional[List[Munch]] = dataclasses.field(default_factory=list)
teams: Optional[List[Munch]] = dataclasses.field(default_factory=list)
annotations: Optional[List[Munch]] = dataclasses.field(default_factory=list)
snapshots: Optional[List[Munch]] = dataclasses.field(default_factory=list)
notifications: Optional[List[Munch]] = dataclasses.field(default_factory=list)
@dataclasses.dataclass
class DashboardDetails:
dashboard: Dict
@property
def panels(self) -> List:
return self.dashboard.dashboard.get("panels", [])
@property
def annotations(self) -> List:
return self.dashboard.dashboard.get("annotations", {}).get("list", [])
@property
def templating(self) -> List:
return self.dashboard.dashboard.get("templating", {}).get("list", [])
@dataclasses.dataclass
class DatasourceItem:
uid: Optional[str] = None
name: Optional[str] = None
type: Optional[str] = None
url: Optional[str] = None
@classmethod
def from_payload(cls, payload: any):
if isinstance(payload, Munch):
payload = dict(payload)
if isinstance(payload, dict):
return cls(**payload)
if isinstance(payload, str):
return cls(name=payload)
raise TypeError(f"Unknown payload type for DatasourceItem: {type(payload)}")
@dataclasses.dataclass
class DatasourceExplorationItem:
datasource: Munch
used_in: List[Munch]
grafana_url: str
def format_compact(self):
dsshort = OrderedDict(
uid=self.datasource.uid,
name=self.datasource.name,
type=self.datasource.type,
url=self.datasource.url,
)
item = OrderedDict(datasource=dsshort)
for dashboard in self.used_in:
item.setdefault("dashboards", [])
dbshort = OrderedDict(
title=dashboard.dashboard.title,
uid=dashboard.dashboard.uid,
path=dashboard.meta.url,
url=urljoin(self.grafana_url, dashboard.meta.url),
)
item["dashboards"].append(dbshort)
return item
@dataclasses.dataclass
class DashboardExplorationItem:
dashboard: Munch
datasources: List[Munch]
grafana_url: str
def format_compact(self):
dbshort = OrderedDict(
title=self.dashboard.dashboard.title,
uid=self.dashboard.dashboard.uid,
path=self.dashboard.meta.url,
url=urljoin(self.grafana_url, self.dashboard.meta.url),
)
item = OrderedDict(dashboard=dbshort)
for datasource in self.datasources:
item.setdefault("datasources", [])
dsshort = OrderedDict(
uid=datasource.uid,
name=datasource.name,
type=datasource.type,
url=datasource.url,
)
item["datasources"].append(dsshort)
return item