X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=osm_mon%2Fcore%2Fcommon_db.py;h=f15ae977d991229bbf26b62ec1d3c8534ead1233;hb=1b7145f6d0d3c0090b7a33b4972861ad9ed48631;hp=8f2f552c6efef80eb429c1f69682e961f0b49f83;hpb=628df021896fa8775f9743af62a4267b617cc35c;p=osm%2FMON.git diff --git a/osm_mon/core/common_db.py b/osm_mon/core/common_db.py index 8f2f552..f15ae97 100644 --- a/osm_mon/core/common_db.py +++ b/osm_mon/core/common_db.py @@ -21,9 +21,12 @@ # For those usages not covered by the Apache License, Version 2.0 please # contact: bdiaz@whitestack.com or glavado@whitestack.com ## +from typing import List + from osm_common import dbmongo, dbmemory from osm_mon.core.config import Config +from osm_mon.core.models import Alarm class CommonDbClient: @@ -132,3 +135,16 @@ class CommonDbClient: def get_project(self, project_id: str): return self.common_db.get_one('projects', {'_id': project_id}) + + def create_alarm(self, alarm: Alarm): + return self.common_db.create('alarms', alarm.to_dict()) + + def delete_alarm(self, alarm_uuid: str): + return self.common_db.del_one('alarms', {'uuid': alarm_uuid}) + + def get_alarms(self) -> List[Alarm]: + alarms = [] + alarm_dicts = self.common_db.get_list('alarms') + for alarm_dict in alarm_dicts: + alarms.append(Alarm.from_dict(alarm_dict)) + return alarms