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