-
Notifications
You must be signed in to change notification settings - Fork 14
/
gf_config.py
61 lines (48 loc) · 2.15 KB
/
gf_config.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
import logging
from django.shortcuts import render
from django.shortcuts import redirect
from django.views.generic import TemplateView
from mercury.forms import GFConfigForm
from mercury.models import GFConfig
from mercury.grafanaAPI.grafana_api import Grafana
from django.contrib import messages
log = logging.getLogger(__name__)
log.setLevel(logging.ERROR)
def update_config(request, gf_id=None):
GFConfig.objects.all().update(gf_current=False)
GFConfig.objects.filter(id=gf_id).update(gf_current=True)
return redirect("/gfconfig")
def delete_config(request, gf_id=None):
GFConfig.objects.get(id=gf_id).delete()
return redirect("/gfconfig")
class GFConfigView(TemplateView):
template_name = "gf_configs.html"
def get(self, request, *args, **kwargs):
configs = GFConfig.objects.all().order_by("id")
config_form = GFConfigForm()
context = {"config_form": config_form, "configs": configs}
return render(request, self.template_name, context)
def post(self, request, *args, **kwargs):
if "submit" in request.POST:
config_data = GFConfig(
gf_name=request.POST.get("gf_name"),
gf_host=request.POST.get("gf_host"),
gf_token=request.POST.get("gf_token"),
)
# Create Grafana instance with host and token
grafana = Grafana(config_data.gf_host, config_data.gf_token)
try:
grafana.validate_credentials()
config_data.gf_current = True
# Only save the config if credentials were validated
config_data.save()
except ValueError as error:
messages.error(request, f"Grafana initial set up failed: {error}")
try:
grafana.create_postgres_datasource()
except ValueError as error:
messages.error(request, f"Datasource couldn't be created. {error}")
configs = GFConfig.objects.all().order_by("id")
config_form = GFConfigForm()
context = {"config_form": config_form, "configs": configs}
return render(request, self.template_name, context)