Bug 2215 fixed
[osm/MON.git] / osm_mon / core / models.py
index 8fe3b6b..d076e5e 100644 (file)
@@ -32,7 +32,10 @@ class Alarm:
         operation: str = None,
         statistic: str = None,
         metric: str = None,
+        action: str = None,
         tags: dict = {},
+        alarm_status: str = "ok",
+        extra_labels: dict = {},
     ):
         self.uuid = str(uuid.uuid4())
         self.name = name
@@ -41,7 +44,10 @@ class Alarm:
         self.operation = operation
         self.statistic = statistic
         self.metric = metric
+        self.action = action
         self.tags = tags
+        self.alarm_status = alarm_status
+        self.extra_labels = extra_labels
 
     def to_dict(self) -> dict:
         alarm = {
@@ -53,6 +59,8 @@ class Alarm:
             "metric": self.metric,
             "tags": self.tags,
             "operation": self.operation,
+            "alarm_status": self.alarm_status,
+            "extra_labels": self.extra_labels,
         }
         return alarm
 
@@ -67,4 +75,6 @@ class Alarm:
         alarm.metric = data.get("metric")
         alarm.tags = data.get("tags")
         alarm.operation = data.get("operation")
+        alarm.alarm_status = data.get("alarm_status")
+        alarm.extra_labels = data.get("extra_labels")
         return alarm