X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osmclient%2Fsol005%2Fk8scluster.py;fp=osmclient%2Fsol005%2Fk8scluster.py;h=92ef49e6bd8ee06ae9c1b4c7ea6e032f8b324af8;hb=961145b6c1ab65c8cc5953aae008781e285c1461;hp=0b99a3793ed53f91f0891362926c4867c1e86fae;hpb=8fdbc7b94c475febecf9896d9ea1e715f82c3daa;p=osm%2Fosmclient.git

diff --git a/osmclient/sol005/k8scluster.py b/osmclient/sol005/k8scluster.py
index 0b99a37..92ef49e 100644
--- a/osmclient/sol005/k8scluster.py
+++ b/osmclient/sol005/k8scluster.py
@@ -33,15 +33,18 @@ class K8scluster(object):
             self._apiName, self._apiVersion, self._apiResource
         )
 
-    def create(self, name, k8s_cluster):
-        def get_vim_account_id(vim_account):
-            vim = self._client.vim.get(vim_account)
-            if vim is None:
-                raise NotFound("cannot find vim account '{}'".format(vim_account))
-            return vim["_id"]
+    def _get_vim_account(self, vim_account):
+        vim = self._client.vim.get(vim_account)
+        if vim is None:
+            raise NotFound("cannot find vim account '{}'".format(vim_account))
+        return vim
 
+    def create(self, name, k8s_cluster):
         self._client.get_token()
-        k8s_cluster["vim_account"] = get_vim_account_id(k8s_cluster["vim_account"])
+        vim_account = self._get_vim_account(k8s_cluster["vim_account"])
+        k8s_cluster["vim_account"] = vim_account["_id"]
+        if "vca" in vim_account:
+            k8s_cluster["vca_id"] = vim_account["vca"]
         http_code, resp = self._http.post_cmd(
             endpoint=self._apiBase, postfields_dict=k8s_cluster
         )
@@ -65,6 +68,11 @@ class K8scluster(object):
     def update(self, name, k8s_cluster):
         self._client.get_token()
         cluster = self.get(name)
+        if "vim_account" in k8s_cluster:
+            vim_account = self._get_vim_account(k8s_cluster["vim_account"])
+            k8s_cluster["vim_account"] = vim_account["_id"]
+            if "vca" in vim_account:
+                k8s_cluster["vca_id"] = vim_account["vca"]
         http_code, resp = self._http.put_cmd(
             endpoint="{}/{}".format(self._apiBase, cluster["_id"]),
             postfields_dict=k8s_cluster,