add health and datasource proxy

Author: peekjef72

.gitignore

@@ -112,3 +112,5 @@ venv.bak/
 # mypy
 .mypy_cache/
 
+# visual code launch...
+.vscode/launch.json

grafana_client/api.py

@@ -5,6 +5,7 @@
     Dashboard,
     Datasource,
     Folder,
+    Health,
     Notifications,
     Organization,
     Organizations,
@@ -40,6 +41,7 @@ def __init__(
         self.dashboard = Dashboard(self.client)
         self.datasource = Datasource(self.client)
         self.folder = Folder(self.client)
+        self.health = Health(self.client)
         self.organization = Organization(self.client)
         self.organizations = Organizations(self.client)
         self.search = Search(self.client)

grafana_client/elements/__init__.py

@@ -3,6 +3,7 @@
 from .base import Base
 from .dashboard import Dashboard
 from .datasource import Datasource
+from .health import Health
 from .folder import Folder
 from .notifications import Notifications
 from .organization import Organization, Organizations

grafana_client/elements/datasource.py

@@ -95,3 +95,31 @@ def delete_datasource_by_name(self, datasource_name):
         delete_datasource = "/datasources/name/%s" % datasource_name
         r = self.client.DELETE(delete_datasource)
         return r
+
+    def get_datasource_proxy_data(self, datasource_id
+        , query_type='query'
+        , version='v1'
+        , expr=None
+        , time=None
+        , start=None
+        , end=None
+        , step=None
+        ):
+        """
+
+        :param datasource_id:
+        :param version: api_version currently v1
+        :param query_type: query_range |query
+        :param expr: expr to query
+
+        :return: r (dict)
+        """
+        get_datasource_path = "/datasources/proxy/{}" \
+		'/api/{}/{}?query={}'.format( datasource_id, version, query_type, expr)
+        if query_type == 'query_range':
+           get_datasource_path = get_datasource_path + '&start={}&end={}&step={}'.format(
+		start, end, step)
+        else:
+           get_datasource_path = get_datasource_path + '&time={}'.format(time)
+        r = self.api.GET(get_datasource_path)
+        return r

grafana_client/elements/health.py

@@ -0,0 +1,16 @@
+from .base import Base
+
+
+class Health(Base):
+    def __init__(self, client):
+        super(Health, self).__init__(client)
+        self.client = client
+
+    def check(self):
+        """
+
+        :return:
+        """
+        path = "/health"
+        r = self.client.GET(path)
+        return r

test/elements/test_health.py

@@ -0,0 +1,33 @@
+import unittest
+
+import requests_mock
+
+from grafana_client import GrafanaApi
+from grafana_client.client import (
+    GrafanaBadInputError,
+    GrafanaClientError,
+    GrafanaServerError,
+    GrafanaUnauthorizedError,
+)
+
+
+class HealthTestCase(unittest.TestCase):
+    def setUp(self):
+        self.grafana = GrafanaApi(("admin", "admin"), host="localhost", url_path_prefix="", protocol="http")
+
+    @requests_mock.Mocker()
+    def test_search_dashboards(self, m):
+        m.get(
+            "http://localhost/api/health",
+            json=[
+                {
+                    "commit": "6f8c1d9fe4",
+                    "database": "ok",
+                    "version": "7.5.11"
+                }
+            ],
+        )
+
+        result = self.grafana.health.check()
+        self.assertEqual(result[0]["database"], 'ok')
+        self.assertEqual(len(result), 1)