8547f5d4a125b1a0a10327ae746c51e086c1b992
[pti/o2.git] / o2ims / domain / ocloud.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 __future__ import annotations
16 import json
17
18 from o2common.domain.base import AgRoot, Serializer
19 # from dataclasses import dataclass
20 # from datetime import date
21 # from typing import Optional, List, Set
22 from .resource_type import ResourceTypeEnum
23 # from uuid import UUID
24
25
26 class DeploymentManager(AgRoot, Serializer):
27     def __init__(self, id: str, name: str, ocloudid: str,
28                  dmsendpoint: str, description: str = '',
29                  supportedLocations: str = '', capabilities: str = '',
30                  capacity: str = '', profile: str = '') -> None:
31         super().__init__()
32         self.deploymentManagerId = id
33         self.version_number = 0
34         self.oCloudId = ocloudid
35         self.name = name
36         self.description = description
37         self.deploymentManagementServiceEndpoint = dmsendpoint
38         self.supportedLocations = supportedLocations
39         self.capabilities = capabilities
40         self.capacity = capacity
41         self.profile = profile
42         self.extensions = []
43
44     def serialize(self):
45         d = Serializer.serialize(self)
46
47         if 'profile' in d and d['profile'] != '':
48             d['profile'] = json.loads(d['profile'])
49
50         return d
51
52
53 class ResourcePool(AgRoot, Serializer):
54     def __init__(self, id: str, name: str, location: str,
55                  ocloudid: str, gLocationId: str = '',
56                  description: str = '') -> None:
57         super().__init__()
58         self.resourcePoolId = id
59         self.version_number = 0
60         self.oCloudId = ocloudid
61         self.globalLocationId = gLocationId
62         self.name = name
63         self.location = location
64         self.description = description
65         self.extensions = []
66
67
68 class ResourceType(AgRoot, Serializer):
69     def __init__(self, typeid: str, name: str, typeEnum: ResourceTypeEnum,
70                  ocloudid: str, vender: str = '', model: str = '',
71                  version: str = '',
72                  description: str = '') -> None:
73         super().__init__()
74         self.resourceTypeId = typeid
75         self.version_number = 0
76         self.oCloudId = ocloudid
77         self.resourceTypeEnum = typeEnum
78         self.name = name
79         self.vender = vender
80         self.model = model
81         self.version = version
82         self.description = description
83         self.extensions = []
84
85
86 class Resource(AgRoot, Serializer):
87     def __init__(self, resourceId: str, resourceTypeId: str,
88                  resourcePoolId: str, name: str, parentId: str = '',
89                  gAssetId: str = '', elements: str = '',
90                  description: str = '') -> None:
91         super().__init__()
92         self.resourceId = resourceId
93         self.version_number = 0
94         self.resourceTypeId = resourceTypeId
95         self.resourcePoolId = resourcePoolId
96         self.name = name
97         self.globalAssetId = gAssetId
98         self.parentId = parentId
99         self.elements = elements
100         self.description = description
101         self.children = []
102         self.extensions = []
103
104     def set_children(self, children: list):
105         self.children = children
106
107     def set_resource_type_name(self, resource_type_name: str):
108         self.resourceTypeName = resource_type_name
109
110     def serialize(self):
111         d = Serializer.serialize(self)
112
113         if 'elements' in d and d['elements'] != '':
114             d['elements'] = json.loads(d['elements'])
115
116         if not hasattr(self, 'children') or len(self.children) == 0:
117             return d
118         else:
119             d['children'] = []
120
121         for child in self.children:
122             d['children'].append(child.serialize())
123         return d
124
125
126 class Ocloud(AgRoot, Serializer):
127     def __init__(self, ocloudid: str, name: str, imsendpoint: str,
128                  globalcloudId: str = '',
129                  description: str = '', version_number: int = 0) -> None:
130         super().__init__()
131         self.oCloudId = ocloudid
132         self.globalcloudId = globalcloudId
133         self.version_number = version_number
134         self.name = name
135         self.description = description
136         self.infrastructureManagementServiceEndpoint = imsendpoint
137         self.resourcePools = []
138         self.deploymentManagers = []
139         self.resourceTypes = []
140         self.extensions = []
141
142     # def addDeploymentManager(self,
143     #                          deploymentManager: DeploymentManager):
144
145     #     deploymentManager.oCloudId = self.oCloudId
146     #     old = filter(
147     #         lambda x: x.deploymentManagerId ==
148     #         deploymentManager.deploymentManagerId,
149     #         self.deploymentManagers)
150     #     for o in old or []:
151     #         self.deploymentManagers.remove(o)
152     #     self.deploymentManagers.append(deploymentManager)