49b25508c8ad492eddf6e0e5e750a45eb688193b
[pti/o2.git] / o2ims / domain / ocloud.py
1 # Copyright (C) 2021 Wind River Systems, Inc.\r
2 #\r
3 #  Licensed under the Apache License, Version 2.0 (the "License");\r
4 #  you may not use this file except in compliance with the License.\r
5 #  You may obtain a copy of the License at\r
6 #\r
7 #      http://www.apache.org/licenses/LICENSE-2.0\r
8 #\r
9 #  Unless required by applicable law or agreed to in writing, software\r
10 #  distributed under the License is distributed on an "AS IS" BASIS,\r
11 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
12 #  See the License for the specific language governing permissions and\r
13 #  limitations under the License.\r
14 \r
15 from __future__ import annotations\r
16 \r
17 from o2ims.domain.base import AgRoot\r
18 # from dataclasses import dataclass\r
19 # from datetime import date\r
20 # from typing import Optional, List, Set\r
21 from .resource_type import ResourceTypeEnum\r
22 # from uuid import UUID\r
23 \r
24 \r
25 class Subscription(AgRoot):\r
26     def __init__(self, id: str, callback: str, consumersubid: str = '',\r
27                  filter: str = '') -> None:\r
28         self.subscriptionId = id\r
29         self.callback = callback\r
30         self.consumerSubscriptionId = consumersubid\r
31         self.filter = filter\r
32 \r
33 \r
34 class DeploymentManager:\r
35     def __init__(self, id: str, name: str, ocloudid: str,\r
36                  dmsendpoint: str, description: str = '',\r
37                  supportedLocations: str = '', capabilities: str = '',\r
38                  capacity: str = '') -> None:\r
39         self.deploymentManagerId = id\r
40         self.oCloudId = ocloudid\r
41         self.name = name\r
42         self.description = description\r
43         self.deploymentManagementServiceEndpoint = dmsendpoint\r
44         self.supportedLocations = supportedLocations\r
45         self.capabilities = capabilities\r
46         self.capacity = capacity\r
47         self.extensions = []\r
48 \r
49 \r
50 class ResourcePool(AgRoot):\r
51     def __init__(self, id: str, name: str, location: str,\r
52                  ocloudid: str, gLocationId: str = '',\r
53                  description: str = '') -> None:\r
54         self.resourcePoolId = id\r
55         self.oCloudId = ocloudid\r
56         self.globalLocationId = gLocationId\r
57         self.name = name\r
58         self.location = location\r
59         self.description = description\r
60         self.extensions = []\r
61 \r
62 \r
63 class ResourceType(AgRoot):\r
64     def __init__(self, typeid: str, name: str, typeEnum: ResourceTypeEnum,\r
65                  ocloudid: str, vender: str = '', model: str = '',\r
66                  version: str = '',\r
67                  description: str = '') -> None:\r
68         self.resourceTypeId = typeid\r
69         self.oCloudId = ocloudid\r
70         self.resourceTypeEnum = typeEnum.value\r
71         self.name = name\r
72         self.vender = vender\r
73         self.model = model\r
74         self.version = version\r
75         self.description = description\r
76         self.extensions = []\r
77 \r
78 \r
79 class Resource(AgRoot):\r
80     def __init__(self, resourceId: str, resourceTypeId: str,\r
81                  resourcePoolId: str, oCloudId: str = '',\r
82                  parentId: str = '', elements: list = [],\r
83                  description: str = '') -> None:\r
84         self.resourceId = resourceId\r
85         self.oCloudId = oCloudId\r
86         self.resourceTypeId = resourceTypeId\r
87         self.resourcePoolId = resourcePoolId\r
88         self.parentId = parentId\r
89         self.elements = elements\r
90         self.description = description\r
91         self.extensions = []\r
92 \r
93 \r
94 class Ocloud(AgRoot):\r
95     def __init__(self, ocloudid: str, name: str, imsendpoint: str,\r
96                  globalcloudId: str = '',\r
97                  description: str = '', version_number: int = 0) -> None:\r
98         super().__init__()\r
99         self.oCloudId = ocloudid\r
100         self.globalcloudId = globalcloudId\r
101         self.version_number = version_number\r
102         self.name = name\r
103         self.description = description\r
104         self.infrastructureManagementServiceEndpoint = imsendpoint\r
105         self.resourcePools = []\r
106         self.deploymentManagers = []\r
107         self.resourceTypes = []\r
108         self.extensions = []\r
109         # self.events = []\r
110 \r
111     def addDeploymentManager(self,\r
112                              deploymentManager: DeploymentManager):\r
113 \r
114         deploymentManager.oCloudId = self.oCloudId\r
115         old = filter(\r
116             lambda x: x.deploymentManagerId ==\r
117             deploymentManager.deploymentManagerId,\r
118             self.deploymentManagers)\r
119         for o in old or []:\r
120             self.deploymentManagers.remove(o)\r
121         self.deploymentManagers.append(deploymentManager)\r