Fix INF-364 alarmEventRecord does not comply with spec
[pti/o2.git] / o2ims / adapter / clients / aggregate_client.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 import uuid
15 from typing import List
16
17 from o2common.service.client.base_client import BaseClient
18 from o2ims.domain import stx_object as ocloudModel
19 from o2ims.domain.resource_type import ResourceTypeEnum
20
21 from o2common.helper import o2logging
22 logger = o2logging.get_logger(__name__)
23
24
25 class ComputeAggClient(BaseClient):
26     def __init__(self, driver=None):
27         super().__init__()
28         self.driver = AggClientImp()
29
30     def _get(self, id) -> ocloudModel.StxGenericModel:
31         return self.driver.getComputeList(res_pool=self._pool_id)[0]
32
33     def _list(self, **filters):
34         return self.driver.getComputeList(res_pool=self._pool_id)
35
36     def _set_stx_client(self):
37         pass
38
39
40 class NetworkAggClient(BaseClient):
41     def __init__(self, driver=None):
42         super().__init__()
43         self.driver = AggClientImp()
44
45     def _get(self, id) -> ocloudModel.StxGenericModel:
46         return self.driver.getNetworkList(res_pool=self._pool_id)[0]
47
48     def _list(self, **filters):
49         return self.driver.getNetworkList(res_pool=self._pool_id)
50
51     def _set_stx_client(self):
52         pass
53
54
55 class StorageAggClient(BaseClient):
56     def __init__(self, driver=None):
57         super().__init__()
58         self.driver = AggClientImp()
59
60     def _get(self, id) -> ocloudModel.StxGenericModel:
61         return self.driver.getStorageList(res_pool=self._pool_id)[0]
62
63     def _list(self, **filters):
64         return self.driver.getStorageList(res_pool=self._pool_id)
65
66     def _set_stx_client(self):
67         pass
68
69
70 class UndefinedAggClient(BaseClient):
71     def __init__(self, driver=None):
72         super().__init__()
73         self.driver = AggClientImp()
74
75     def _get(self, id) -> ocloudModel.StxGenericModel:
76         return self.driver.getUndefinedList(res_pool=self._pool_id)[0]
77
78     def _list(self, **filters):
79         return self.driver.getUndefinedList(res_pool=self._pool_id)
80
81     def _set_stx_client(self):
82         pass
83
84
85 class AggClientImp(object):
86     def __init__(self):
87         super().__init__()
88
89     def getComputeList(self, **filters) -> List[ocloudModel.StxGenericModel]:
90         compute = ComputeAggregate(filters['res_pool'])
91         return [ocloudModel.StxGenericModel(
92             ResourceTypeEnum.COMPUTE_AGGREGATE, compute)]
93
94     def getNetworkList(self, **filters) -> List[ocloudModel.StxGenericModel]:
95         network = NetworkAggregate(filters['res_pool'])
96         return [ocloudModel.StxGenericModel(
97             ResourceTypeEnum.NETWORK_AGGREGATE, network)]
98
99     def getStorageList(self, **filters) -> List[ocloudModel.StxGenericModel]:
100         storage = StorageAggregate(filters['res_pool'])
101         return [ocloudModel.StxGenericModel(
102             ResourceTypeEnum.STORAGE_AGGREGATE, storage)]
103
104     def getUndefinedList(self, **filters) -> List[ocloudModel.StxGenericModel]:
105         undefined = UndefinedAggregate(filters['res_pool'])
106         return [ocloudModel.StxGenericModel(
107             ResourceTypeEnum.UNDEFINED_AGGREGATE, undefined)]
108
109
110 class Aggregate:
111     def __init__(self, res_pool_id: str, name: str) -> None:
112         self.name = name
113         setattr(self, 'name', self.name)
114         setattr(self, 'uuid',
115                 uuid.uuid3(uuid.NAMESPACE_URL, res_pool_id + self.name))
116         setattr(self, 'updated_at', None)
117         setattr(self, 'created_at', None)
118
119     def to_dict(self):
120         return {}
121
122
123 class ComputeAggregate(Aggregate):
124     def __init__(self, res_pool_id: str) -> None:
125         super().__init__(res_pool_id, 'compute_aggregate')
126
127
128 class NetworkAggregate(Aggregate):
129     def __init__(self, res_pool_id: str) -> None:
130         super().__init__(res_pool_id, 'network_aggregate')
131
132
133 class StorageAggregate(Aggregate):
134     def __init__(self, res_pool_id: str) -> None:
135         super().__init__(res_pool_id, 'storage_aggregate')
136
137
138 class UndefinedAggregate(Aggregate):
139     def __init__(self, res_pool_id: str) -> None:
140         super().__init__(res_pool_id, 'undefined_aggregate')