X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_mon%2Fcore%2Fcommon_db.py;h=1e78bc8f4c383d86114c79b2857a47e362362550;hb=345e73dd34a166bea88158d8c0270c436d384858;hp=df8db6079d7626985021a1d2d23770be463f5e02;hpb=8e4179facf22c8096992f0a83caeec9f2f4996c7;p=osm%2FMON.git

diff --git a/osm_mon/core/common_db.py b/osm_mon/core/common_db.py
index df8db60..1e78bc8 100644
--- a/osm_mon/core/common_db.py
+++ b/osm_mon/core/common_db.py
@@ -144,6 +144,15 @@ class CommonDbClient:
                     )
         return vim_account
 
+    def set_vim_account(self, vim_account_id: str, update_dict: dict) -> bool:
+        try:
+            # Set vim_account resources in mongo
+            self.common_db.set_one('vim_accounts', {"_id": vim_account_id}, update_dict)
+            # self.common_db.set_one('vim_accounts', {"name": "test-vim"}, update_dict)
+            return True
+        except Exception:
+            return False
+
     def get_sdncs(self):
         return self.common_db.get_list("sdns")
 
@@ -157,9 +166,12 @@ class CommonDbClient:
         return self.common_db.get_one("projects", {"_id": project_id})
 
     def create_alarm(self, alarm: Alarm):
+        action_data = {"uuid": alarm.uuid, "action": alarm.action}
+        self.common_db.create("alarms_action", action_data)
         return self.common_db.create("alarms", alarm.to_dict())
 
     def delete_alarm(self, alarm_uuid: str):
+        self.common_db.del_one("alarms_action", {"uuid": alarm_uuid})
         return self.common_db.del_one("alarms", {"uuid": alarm_uuid})
 
     def get_alarms(self) -> List[Alarm]:
@@ -169,6 +181,13 @@ class CommonDbClient:
             alarms.append(Alarm.from_dict(alarm_dict))
         return alarms
 
+    def update_alarm_status(self, alarm_state: str, uuid):
+        modified_count = self.common_db.set_one("alarms", {"uuid": uuid}, {"alarm_status": alarm_state})
+        return modified_count
+
+    def get_alarm_by_uuid(self, uuid: str):
+        return self.common_db.get_one("alarms", {"uuid": uuid})
+
     def get_user(self, username: str):
         return self.common_db.get_one("users", {"username": username})