Fix INF-352 Add aggregate resource types

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

diff --git a/o2ims/service/watcher/agg_undefined_watcher.py b/o2ims/service/watcher/agg_undefined_watcher.py
new file mode 100644 (file)
index 0000000..eda07af
--- /dev/null
+++ b/o2ims/service/watcher/agg_undefined_watcher.py
@@ -0,0 +1,41 @@
+# Copyright (C) 2022 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 o2ims.service.watcher.aggregate_watcher import AggregateWatcher
+from o2ims.domain import commands
+from o2common.service.messagebus import MessageBus
+
+from o2common.helper import o2logging
+logger = o2logging.get_logger(__name__)
+
+
+class UndefinedAggWatcher(AggregateWatcher):
+    def __init__(self, client: BaseClient,
+                 bus: MessageBus) -> None:
+        super().__init__(client, bus)
+
+    def _targetname(self):
+        return "undefined_aggregate"
+
+    def _probe(self, parent: StxGenericModel, tags=None):
+        # Set a tag for children resource
+        self._tags.pool = parent.res_pool_id
+        self._set_respool_client()
+
+        resourcepoolid = parent.id
+        newmodels = self._client.list(resourcepoolid=resourcepoolid)
+        return [commands.UpdateUndefinedAgg(data=m, parentid=resourcepoolid)
+                for m in newmodels]