Fix INF-359 and INF-352 about the alarm event record
[pti/o2.git] / o2ims / service / watcher / alarm_watcher.py
1 # Copyright (C) 2021 Wind River Systems, Inc.
2 #
3 #  Licensed under the Apache License, Version 2.0 (the "License");
4 #  you may not use this file except in compliance with the License.
5 #  You may obtain a copy of the License at
6 #
7 #      http://www.apache.org/licenses/LICENSE-2.0
8 #
9 #  Unless required by applicable law or agreed to in writing, software
10 #  distributed under the License is distributed on an "AS IS" BASIS,
11 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 #  See the License for the specific language governing permissions and
13 #  limitations under the License.
14
15 from o2common.domain import tags
16 from o2common.service.messagebus import MessageBus
17 from o2common.service.watcher.base import BaseWatcher
18 from o2common.service.client.base_client import BaseClient
19
20 from o2ims.domain import commands
21 from o2ims.domain.stx_object import StxGenericModel
22
23 from o2common.helper import o2logging
24 logger = o2logging.get_logger(__name__)
25
26
27 class AlarmWatcher(BaseWatcher):
28     def __init__(self, fault_client: BaseClient,
29                  bus: MessageBus) -> None:
30         super().__init__(fault_client, bus)
31         self._tags = tags.Tag()
32         self.poolid = None
33
34     def _targetname(self):
35         return "alarm"
36
37     def _probe(self, parent: StxGenericModel, tags: object = None):
38         # Set a tag for children resource
39         self._tags.pool = parent.res_pool_id
40         self._set_respool_client()
41
42         resourcepoolid = parent.id
43         newmodels = self._client.list()
44         return [commands.UpdateAlarm(m, resourcepoolid) for m in newmodels] \
45             if len(newmodels) > 0 else []
46
47     def _set_respool_client(self):
48         self.poolid = self._tags.pool
49         self._client.set_pool_driver(self.poolid)