Gmim 553

smsdk/client.py

@@ -238,6 +238,116 @@ def get_data_v1(self, ename, util_name, normalize=True, *args, **kwargs):
 
         return data
 
+    @version_check_decorator
+    def list_alerts(self, alert_type=""):
+        """
+        Main data fetching function for all the entities.  Note this is the general data fetch function.  You probably want to use the model-specific functions such as get_cycles().
+        :param ename: Name of the entities
+        :param util_name: Name of the utility function
+        :param normalize: Flatten nested data structures
+        :return: pandas dataframe
+        """
+        base_url = get_url(
+            self.config["protocol"],
+            self.tenant,
+            self.config["site.domain"],
+            self.config["port"],
+        )
+
+        cls = smsdkentities.get("alert")(self.session, base_url)
+        alert_info = getattr(cls, "list_alerts")(alert_type)
+        return alert_info
+
+    @version_check_decorator
+    def get_alert_dataframe(self, alert_type=""):
+        """
+        Main data fetching function for all the entities.  Note this is the general data fetch function.  You probably want to use the model-specific functions such as get_cycles().
+        :param ename: Name of the entities
+        :param util_name: Name of the utility function
+        :param normalize: Flatten nested data structures
+        :return: pandas dataframe
+        """
+        base_url = get_url(
+            self.config["protocol"],
+            self.tenant,
+            self.config["site.domain"],
+            self.config["port"],
+        )
+
+        cls = smsdkentities.get("alert")(self.session, base_url)
+        alert_dataframe = getattr(cls, "get_alert_dataframe")(alert_type)
+        return alert_dataframe
+
+    def update_alert(self, alert_id, params):
+        base_url = get_url(
+            self.config["protocol"],
+            self.tenant,
+            self.config["site.domain"],
+            self.config["port"],
+        )
+        cls = smsdkentities.get("alert")(self.session, base_url)
+        getattr(cls, "update_alert")(alert_id, params)
+
+    def delete_alert(self, alert_id=None, delete_all=False, alert_group=""):
+        """
+        Main data fetching function for all the entities.  Note this is the general data fetch function.  You probably want to use the model-specific functions such as get_cycles().
+        :param ename: Name of the entities
+        :param util_name: Name of the utility function
+        :param normalize: Flatten nested data structures
+        :return: pandas dataframe
+        """
+        base_url = get_url(
+            self.config["protocol"],
+            self.tenant,
+            self.config["site.domain"],
+            self.config["port"],
+        )
+        if alert_id is None and not delete_all and not alert_group:
+            print(
+                "Invalid input please provide alert group or alert id or delete_all flag as true if you want to delete all the alerts."
+            )
+            return
+        cls = smsdkentities.get("alert")(self.session, base_url)
+        getattr(cls, "delete_alert")(alert_id, delete_all, alert_group)
+
+    @version_check_decorator
+    def create_alerts(self, dataframe, alert_type=""):
+        """
+        Main data fetching function for all the entities.  Note this is the general data fetch function.  You probably want to use the model-specific functions such as get_cycles().
+        :param ename: Name of the entities
+        :param util_name: Name of the utility function
+        :param normalize: Flatten nested data structures
+        :return: pandas dataframe
+        """
+        base_url = get_url(
+            self.config["protocol"],
+            self.tenant,
+            self.config["site.domain"],
+            self.config["port"],
+        )
+
+        cls = smsdkentities.get("alert")(self.session, base_url)
+        getattr(cls, "create_alert")(alert_type, dataframe)
+
+    @version_check_decorator
+    def update_alert_group(self, dataframe):
+        """
+        Main data fetching function for all the entities.  Note this is the general data fetch function.  You probably want to use the model-specific functions such as get_cycles().
+        :param ename: Name of the entities
+        :param util_name: Name of the utility function
+        :param normalize: Flatten nested data structures
+        :return: pandas dataframe
+        """
+        base_url = get_url(
+            self.config["protocol"],
+            self.tenant,
+            self.config["site.domain"],
+            self.config["port"],
+        )
+
+        cls = smsdkentities.get("alert")(self.session, base_url)
+        getattr(cls, "update_alert_group")(dataframe)
+
     @version_check_decorator
     @ClientV0.validate_input
     @ClientV0.cycle_decorator

smsdk/config/api_endpoints.json

@@ -2,6 +2,9 @@
   "Auth": {
     "url": "/auth/password/login"
   },
+  "Alert": {
+    "url": "/v1/obj/alert_config"
+  },
   "Cycle": {
     "url_v1": "/v1/datatab/cycle",
     "url": "/api/cycle",