o2ims/adapter/clients/ocloud_sa_client.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 # client talking to Stx standalone\r
  16 \r
  17 import uuid\r
  18 from o2ims.service.client.base_client import BaseClient\r
  19 from typing import List\r
  20 # Optional,  Set\r
  21 from o2ims.domain import stx_object as ocloudModel\r
  22 from o2ims import config\r
  23 \r
  24 # from dcmanagerclient.api import client\r
  25 from cgtsclient.client import get_client\r
  26 import logging\r
  27 logger = logging.getLogger(__name__)\r
  28 \r
  29 \r
  30 class StxSaOcloudClient(BaseClient):\r
  31     def __init__(self, driver=None):\r
  32         super().__init__()\r
  33         self.driver = driver if driver else StxSaClientImp()\r
  34 \r
  35     # def list(self) -> List[ocloudModel.StxGenericModel]:\r
  36     #     return self._list()\r
  37 \r
  38     # def get(self, id) -> ocloudModel.StxGenericModel:\r
  39     #     return self._get(id)\r
  40 \r
  41     def _get(self, id) -> ocloudModel.StxGenericModel:\r
  42         return self.driver.getInstanceInfo()\r
  43 \r
  44     def _list(self):\r
  45         return [self.driver.getInstanceInfo()]\r
  46 \r
  47 \r
  48 class StxSaResourcePoolClient(BaseClient):\r
  49     def __init__(self):\r
  50         super().__init__()\r
  51         self.driver = StxSaClientImp()\r
  52 \r
  53     def _get(self, id) -> ocloudModel.StxGenericModel:\r
  54         return self.driver.getInstanceInfo()\r
  55 \r
  56     def _list(self):\r
  57         return [self.driver.getInstanceInfo()]\r
  58 \r
  59 \r
  60 class StxSaDmsClient(BaseClient):\r
  61     def __init__(self):\r
  62         super().__init__()\r
  63         self.driver = StxSaClientImp()\r
  64 \r
  65     def _get(self, name) -> ocloudModel.StxGenericModel:\r
  66         return self.driver.getK8sDetail(name)\r
  67 \r
  68     def _list(self):\r
  69         return self.driver.getK8sList()\r
  70 \r
  71 \r
  72 class StxPserverClient(BaseClient):\r
  73     def __init__(self):\r
  74         super().__init__()\r
  75         self.driver = StxSaClientImp()\r
  76 \r
  77     def _get(self, id) -> ocloudModel.StxGenericModel:\r
  78         return self.driver.getPserver(id)\r
  79 \r
  80     def _list(self) -> List[ocloudModel.StxGenericModel]:\r
  81         return self.driver.getPserverList()\r
  82 \r
  83 \r
  84 class StxCpuClient(BaseClient):\r
  85     def __init__(self, pserver_id):\r
  86         super().__init__()\r
  87         self._pserver_id = pserver_id\r
  88         self.driver = StxSaClientImp()\r
  89 \r
  90     def _get(self, id) -> ocloudModel.StxGenericModel:\r
  91         return self.driver.getCpu(id)\r
  92 \r
  93     def _list(self) -> List[ocloudModel.StxGenericModel]:\r
  94         return self.driver.getCpuList(self._pserver_id)\r
  95 \r
  96 # internal driver which implement client call to Stx Standalone instance\r
  97 \r
  98 \r
  99 class StxSaClientImp(object):\r
 100     def __init__(self, stx_client=None):\r
 101         super().__init__()\r
 102         self.stxclient = stx_client if stx_client else self.getStxClient()\r
 103 \r
 104     def getStxClient():\r
 105         os_client_args = config.get_stx_access_info()\r
 106         config_client = get_client(**os_client_args)\r
 107         return config_client\r
 108 \r
 109     def getInstanceInfo(self) -> ocloudModel.StxGenericModel:\r
 110         systems = self.stxclient.isystem.list()\r
 111         logger.debug("systems:" + str(systems[0].to_dict()))\r
 112         return ocloudModel.StxGenericModel(systems[0]) if systems else None\r
 113 \r
 114     def getPserverList(self) -> List[ocloudModel.StxGenericModel]:\r
 115         hosts = self.stxclient.ihost.list()\r
 116         logger.debug("host 1:" + str(hosts[0].to_dict()))\r
 117         return [ocloudModel.StxGenericModel(self._hostconverter(host))\r
 118                 for host in hosts if host]\r
 119 \r
 120     def getPserver(self, id) -> ocloudModel.StxGenericModel:\r
 121         host = self.stxclient.ihost.get(id)\r
 122         logger.debug("host:" + str(host.to_dict()))\r
 123         return ocloudModel.StxGenericModel(self._hostconverter(host))\r
 124 \r
 125     def getK8sList(self) -> List[ocloudModel.StxGenericModel]:\r
 126         k8sclusters = self.stxclient.kube_cluster.list()\r
 127         logger.debug("k8sresources:" + str(k8sclusters[0].to_dict()))\r
 128 \r
 129         return [ocloudModel.StxGenericModel(self._k8sconverter(k8sres))\r
 130                 for k8sres in k8sclusters if k8sres]\r
 131 \r
 132     def getK8sDetail(self, name) -> ocloudModel.StxGenericModel:\r
 133         k8scluster = self.stxclient.kube_cluster.get(name)\r
 134         logger.debug("k8sresource:" + str(k8scluster.to_dict()))\r
 135         return ocloudModel.StxGenericModel(self._k8sconverter(k8scluster))\r
 136 \r
 137     def getCpuList(self, hostid) -> List[ocloudModel.StxGenericModel]:\r
 138         cpulist = self.stxclient.icpu.list(hostid)\r
 139         return [ocloudModel.StxGenericModel(self._cpuconverter(cpures))\r
 140                 for cpures in cpulist if cpures]\r
 141 \r
 142     def getCpu(self, id) -> ocloudModel.StxGenericModel:\r
 143         cpuinfo = self.stxclient.icpu.get(id)\r
 144         return ocloudModel.StxGenericModel(self._cpuconverter(cpuinfo))\r
 145 \r
 146     def _getIsystems(self):\r
 147         return self.stxclient.isystem.list()\r
 148 \r
 149     def _getIsystem(self, id=None):\r
 150         if id:\r
 151             return self.stxclient.isystem.get(id)\r
 152         else:\r
 153             isystems = self.stxclient.isystem.list()\r
 154             if len(isystems) != 1 and not id:\r
 155                 raise Exception('No system uuid was provided and '\r
 156                                 'more than one system exists in the account.')\r
 157             return isystems[0]\r
 158 \r
 159     @staticmethod\r
 160     def _hostconverter(host):\r
 161         setattr(host, "name", host.hostname)\r
 162         return host\r
 163 \r
 164     @staticmethod\r
 165     def _cpuconverter(cpu):\r
 166         setattr(cpu, "name", "core-"+str(cpu.core))\r
 167         return cpu\r
 168 \r
 169     @staticmethod\r
 170     def _k8sconverter(host):\r
 171         setattr(host, "name", host.cluster_name)\r
 172         setattr(host, "uuid",\r
 173                 uuid.uuid3(uuid.NAMESPACE_URL, host.cluster_name))\r
 174         setattr(host, 'updated_at', None)\r
 175         setattr(host, 'created_at', None)\r
 176         logger.debug("k8s cluster name/uuid:" +\r
 177                      host.name + "/" + str(host.uuid))\r
 178         return host\r