X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_mon%2Fcore%2Fcommon_db.py;h=66f81fe32cbce728d9d6ed3f250e0d2e0e39058a;hb=b3173221c25ae915eaca041ebd045f6148cc447d;hp=4f6ace0b667a487638b675e8081c2bc2a7f141b2;hpb=5d7b0d1beb1db2f0939604482e7998a6c1bc6951;p=osm%2FMON.git

diff --git a/osm_mon/core/common_db.py b/osm_mon/core/common_db.py
index 4f6ace0..66f81fe 100644
--- a/osm_mon/core/common_db.py
+++ b/osm_mon/core/common_db.py
@@ -17,7 +17,6 @@
 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 # License for the specific language governing permissions and limitations
 # under the License.
-
 # For those usages not covered by the Apache License, Version 2.0 please
 # contact: bdiaz@whitestack.com or glavado@whitestack.com
 ##
@@ -147,7 +146,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:
@@ -185,7 +184,15 @@ 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 update_alarm_extra_labels(self, alarm_labels: dict, uuid):
+        modified_count = self.common_db.set_one(
+            "alarms", {"uuid": uuid}, {"extra_labels": alarm_labels}
+        )
         return modified_count
 
     def get_alarm_by_uuid(self, uuid: str):