# 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
##
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:
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):