Using same alerting API as the UI.

Author: Maharacha

grafana_client/api.py

@@ -12,6 +12,7 @@
 from .client import GrafanaClient
 from .elements import (
     Admin,
+    Alerting,
     AlertingProvisioning,
     Annotations,
     Dashboard,
@@ -57,6 +58,7 @@ def __init__(
         )
         self.url = None
         self.admin = Admin(self.client)
+        self.alerting = Alerting(self.client)
         self.alertingprovisioning = AlertingProvisioning(self.client)
         self.dashboard = Dashboard(self.client)
         self.dashboard_versions = DashboardVersions(self.client)

grafana_client/elements/__init__.py

@@ -1,4 +1,5 @@
 from .admin import Admin
+from .alerting import Alerting
 from .alertingprovisioning import AlertingProvisioning
 from .annotations import Annotations
 from .base import Base

grafana_client/elements/alerting.py

@@ -0,0 +1,49 @@
+from .base import Base
+
+
+class Alerting(Base):
+    def __init__(self, client):
+        super(Alerting, self).__init__(client)
+        self.client = client
+
+    def get_alertrule(self, folder_name, alertrule_name):
+        """
+        :param folder_name:
+        :param alertrule_name:
+        :return:
+        """
+        get_alertrule_path = "/ruler/grafana/api/v1/rules/%s/%s" % (folder_name, alertrule_name)
+        r = self.client.GET(get_alertrule_path)
+        return r
+
+    def create_alertrule(self, folder_name, alertrule):
+        """
+        :param folder_name:
+        :param alertrule:
+        :return:
+        """
+        create_alertrule_path = "/ruler/grafana/api/v1/rules/%s" % folder_name
+        r = self.client.POST(create_alertrule_path, json=alertrule)
+        return r
+
+    def update_alertrule(self, folder_name, alertrule):
+        """
+        @param folder_name:
+        @param alertrule:
+        @return:
+        """
+
+        update_alertrule_path = "/ruler/grafana/api/v1/rules/%s" % folder_name
+        r = self.client.POST(update_alertrule_path, json=alertrule)
+        return r
+
+    def delete_alertrule(self, folder_name, alertrule_name):
+        """
+        :param folder_name:
+        :param alertrule_name:
+        @return:
+        """
+
+        delete_alertrule_path = "/ruler/grafana/api/v1/rules/%s/%s" % (folder_name, alertrule_name)
+        r = self.client.DELETE(delete_alertrule_path)
+        return r