Use of yaml.safe_load instead of yaml_load

[osm/MON.git] / osm_mon / core / common_db.py

diff --git a/osm_mon/core/common_db.py b/osm_mon/core/common_db.py
index 1e78bc8..a9fcfbe 100644 (file)
--- a/osm_mon/core/common_db.py
+++ b/osm_mon/core/common_db.py
@@ -147,7 +147,7 @@ class CommonDbClient:
     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", {"_id": vim_account_id}, update_dict)
             # self.common_db.set_one('vim_accounts', {"name": "test-vim"}, update_dict)
             return True
         except Exception:
@@ -165,6 +165,9 @@ class CommonDbClient:
     def get_project(self, project_id: str):
         return self.common_db.get_one("projects", {"_id": project_id})
 
+    def get_k8sclusters(self):
+        return self.common_db.get_list("k8sclusters")
+
     def create_alarm(self, alarm: Alarm):
         action_data = {"uuid": alarm.uuid, "action": alarm.action}
         self.common_db.create("alarms_action", action_data)
@@ -182,7 +185,9 @@ class CommonDbClient:
         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})
+        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):