Add Accelerator resource watcher; fix bug of the alarm dictionary

[pti/o2.git] / o2ims / service / watcher / pserver_dev_watcher.py

diff --git a/o2ims/service/watcher/pserver_dev_watcher.py b/o2ims/service/watcher/pserver_dev_watcher.py
new file mode 100644 (file)
index 0000000..446aa26
--- /dev/null
+++ b/o2ims/service/watcher/pserver_dev_watcher.py
@@ -0,0 +1,42 @@
+# Copyright (C) 2021 Wind River Systems, Inc.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+from o2ims.domain.stx_object import StxGenericModel
+from o2common.service.client.base_client import BaseClient
+# from o2common.service.unit_of_work import AbstractUnitOfWork
+from o2ims.service.watcher.resource_watcher import ResourceWatcher
+from o2ims.domain import commands
+from o2common.service.messagebus import MessageBus
+
+from o2common.helper import o2logging
+logger = o2logging.get_logger(__name__)
+
+
+class PServerDevWatcher(ResourceWatcher):
+    def __init__(self, client: BaseClient,
+                 bus: MessageBus) -> None:
+        super().__init__(client, bus)
+
+    def _targetname(self):
+        return "pserver_dev"
+
+    def _probe(self, parent: StxGenericModel, tags):
+        # Set a tag for children resource
+        self._tags.pool = tags.pool
+        self._set_respool_client()
+
+        hostid = parent.id
+        newmodels = self._client.list(hostid=hostid)
+        return [commands.UpdatePserverDev(data=m, parentid=hostid)
+                for m in newmodels]