Fix INF-352 Add aggregate resource types
[pti/o2.git] / o2ims / service / watcher / agg_undefined_watcher.py
1 # Copyright (C) 2022 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 o2ims.domain.stx_object import StxGenericModel
16 from o2common.service.client.base_client import BaseClient
17 from o2ims.service.watcher.aggregate_watcher import AggregateWatcher
18 from o2ims.domain import commands
19 from o2common.service.messagebus import MessageBus
20
21 from o2common.helper import o2logging
22 logger = o2logging.get_logger(__name__)
23
24
25 class UndefinedAggWatcher(AggregateWatcher):
26     def __init__(self, client: BaseClient,
27                  bus: MessageBus) -> None:
28         super().__init__(client, bus)
29
30     def _targetname(self):
31         return "undefined_aggregate"
32
33     def _probe(self, parent: StxGenericModel, tags=None):
34         # Set a tag for children resource
35         self._tags.pool = parent.res_pool_id
36         self._set_respool_client()
37
38         resourcepoolid = parent.id
39         newmodels = self._client.list(resourcepoolid=resourcepoolid)
40         return [commands.UpdateUndefinedAgg(data=m, parentid=resourcepoolid)
41                 for m in newmodels]