Add adapter to get k8s cluster info
[pti/o2.git] / o2ims / adapter / clients / ocloud_sa_client.py
index 5d1a5f9..5e9e64c 100644 (file)
@@ -14,6 +14,7 @@
 \r
 # client talking to Stx standalone\r
 \r
+import uuid\r
 from o2ims.service.client.base_client import BaseClient\r
 from typing import List\r
 # Optional,  Set\r
@@ -61,8 +62,8 @@ class StxSaDmsClient(BaseClient):
         super().__init__()\r
         self.driver = StxSaClientImp()\r
 \r
-    def _get(self, id) -> ocloudModel.StxGenericModel:\r
-        return self.driver.getK8sDetail(id)\r
+    def _get(self, name) -> ocloudModel.StxGenericModel:\r
+        return self.driver.getK8sDetail(name)\r
 \r
     def _list(self):\r
         return self.driver.getK8sList()\r
@@ -79,6 +80,19 @@ class StxPserverClient(BaseClient):
     def _list(self) -> List[ocloudModel.StxGenericModel]:\r
         return self.driver.getPserverList()\r
 \r
+\r
+class StxCpuClient(BaseClient):\r
+    def __init__(self, pserver_id):\r
+        super().__init__()\r
+        self._pserver_id = pserver_id\r
+        self.driver = StxSaClientImp()\r
+\r
+    def _get(self, id) -> ocloudModel.StxGenericModel:\r
+        return self.driver.getCpu(id)\r
+\r
+    def _list(self) -> List[ocloudModel.StxGenericModel]:\r
+        return self.driver.getCpuList(self._pserver_id)\r
+\r
 # internal driver which implement client call to Stx Standalone instance\r
 \r
 \r
@@ -109,12 +123,56 @@ class StxSaClientImp(object):
         return ocloudModel.StxGenericModel(self._hostconverter(host))\r
 \r
     def getK8sList(self) -> List[ocloudModel.StxGenericModel]:\r
-        raise NotImplementedError\r
-\r
-    def getK8sDetail(self, id) -> ocloudModel.StxGenericModel:\r
-        raise NotImplementedError\r
+        k8sclusters = self.stxclient.kube_cluster.list()\r
+        logger.debug("k8sresources:" + str(k8sclusters[0].to_dict()))\r
+\r
+        return [ocloudModel.StxGenericModel(self._k8sconverter(k8sres))\r
+                for k8sres in k8sclusters if k8sres]\r
+\r
+    def getK8sDetail(self, name) -> ocloudModel.StxGenericModel:\r
+        k8scluster = self.stxclient.kube_cluster.get(name)\r
+        logger.debug("k8sresource:" + str(k8scluster.to_dict()))\r
+        return ocloudModel.StxGenericModel(self._k8sconverter(k8scluster))\r
+\r
+    def getCpuList(self, hostid) -> List[ocloudModel.StxGenericModel]:\r
+        cpulist = self.stxclient.icpu.list(hostid)\r
+        return [ocloudModel.StxGenericModel(self._cpuconverter(cpures))\r
+                for cpures in cpulist if cpures]\r
+\r
+    def getCpu(self, id) -> ocloudModel.StxGenericModel:\r
+        cpuinfo = self.stxclient.icpu.get(id)\r
+        return ocloudModel.StxGenericModel(self._cpuconverter(cpuinfo))\r
+\r
+    def _getIsystems(self):\r
+        return self.stxclient.isystem.list()\r
+\r
+    def _getIsystem(self, id=None):\r
+        if id:\r
+            return self.stxclient.isystem.get(id)\r
+        else:\r
+            isystems = self.stxclient.isystem.list()\r
+            if len(isystems) != 1 and not id:\r
+                raise Exception('No system uuid was provided and '\r
+                                'more than one system exists in the account.')\r
+            return isystems[0]\r
 \r
     @staticmethod\r
     def _hostconverter(host):\r
         setattr(host, "name", host.hostname)\r
         return host\r
+\r
+    @staticmethod\r
+    def _cpuconverter(cpu):\r
+        setattr(cpu, "name", "core-"+str(cpu.core))\r
+        return cpu\r
+\r
+    @staticmethod\r
+    def _k8sconverter(host):\r
+        setattr(host, "name", host.cluster_name)\r
+        setattr(host, "uuid",\r
+                uuid.uuid3(uuid.NAMESPACE_URL, host.cluster_name))\r
+        setattr(host, 'updated_at', None)\r
+        setattr(host, 'created_at', None)\r
+        logger.debug("k8s cluster name/uuid:" +\r
+                     host.name + "/" + str(host.uuid))\r
+        return host\r