diff options
Diffstat (limited to 'share/starlingx_base/registration/registration.py')
| -rw-r--r-- | share/starlingx_base/registration/registration.py | 126 |
1 files changed, 126 insertions, 0 deletions
diff --git a/share/starlingx_base/registration/registration.py b/share/starlingx_base/registration/registration.py index 565799a0..c45db66b 100644 --- a/share/starlingx_base/registration/registration.py +++ b/share/starlingx_base/registration/registration.py @@ -17,6 +17,10 @@ import json import uuid import traceback +import requests + +from ruamel import yaml + from django.conf import settings from newton_base.registration import registration as newton_registration @@ -252,6 +256,14 @@ class RegistryHelper(newton_registration.RegistryHelper): except Exception as e: self._logger.debug("update cloud region fails %s" % str(e)) + # update k8s connectivity + try: + self._update_k8s_info(cloud_owner, cloud_region_id, viminfo, + cloud_extra_info, sess) + except Exception as e: + self.__logger.debug("update k8s info failes %s" % str(e)) + # continue the registration without reporting error + try: return super(RegistryHelper, self).registryV0(vimid, project_idorname) except Exception as e: @@ -290,6 +302,7 @@ class RegistryHelper(newton_registration.RegistryHelper): }) return instruction_capability + def _update_cloud_region(self, cloud_owner, cloud_region_id, openstack_region_id, viminfo, session=None): if cloud_owner and cloud_region_id: self._logger.debug( @@ -419,6 +432,119 @@ class RegistryHelper(newton_registration.RegistryHelper): return [] + def _update_k8s_info(self, cloud_owner, cloud_region_id, + viminfo, cloud_extra_info, session=None): + try: + vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id) + + # check system version of starlingx + systeminfo = self._get_list_resources( + "/isystems", "platform", session, viminfo, vimid, + "isystems") + systemversion = systeminfo[0].get("software_version", None) if systeminfo else None + if not systemversion: + self._logger.warn("query system version fails") + return + + # check if a k8s platform + is_k8s_cluster = False + # check WRCP versions: + if systemversion == "19.12": + is_k8s_cluster = True + elif systemversion == "19.10": + is_k8s_cluster = True + + if not is_k8s_cluster: + self._logger.info("%s, %s is not a k8s platform, system version: %s" + % (cloud_owner, cloud_region_id, systemversion)) + return + + # check if user token provided to access k8s platform + k8s_apitoken = cloud_extra_info.get( + "k8s-apitoken", None) if cloud_extra_info else None + k8s_apiserver = cloud_extra_info.get( + "k8s-apiserver", None) if cloud_extra_info else None + if not k8s_apitoken: + self._logger.warn("k8s-apitoken is not provided,"\ + "k8s connectivity must be provisioned in other ways") + return + + if not k8s_apiserver: + self._logger.warn("k8s-apiserver is not provided,"\ + "k8s connectivity must be provisioned in other ways") + return + + # now create kube config + kubecfgdata = { + "apiVersion": "v1", + "clusters": + [ + {"cluster": { + "insecure-skip-tls-verify": True, + "server": k8s_apiserver + }, + "name": "wrcpcluster"} + ], + "contexts": + [ + {"context": { + "cluster": "wrcpcluster", + "namespace": "default", + "user": "admin-user"}, + "name": "wrcpcluster-admin"} + ], + "current-context": "wrcpcluster-admin", + "kind": "Config", + "preferences": {}, + "users": + [ + {"name": "admin-user", + "user":{ + "token": k8s_apitoken + }} + ] + } + + kubecfgfilepath = "/tmp/k8sconfig_%s_%s" % (cloud_owner, cloud_region_id) + + # encoding = utf-8 by default + with open(kubecfgfilepath, "w") as kubecfgfile: + yaml.dump(kubecfgdata, kubecfgfile, Dumper=yaml.RoundTripDumper) + + # now create connectivity to multicloud-k8s + multicloudK8sUrl = "%s://%s:%s/api/multicloud-k8s/v1" % ( + settings.MSB_SERVICE_PROTOCOL, settings.MSB_SERVICE_ADDR, settings.MSB_SERVICE_PORT) + auth_api_url = "/v1/connectivity-info" + + metadata1 = { + "cloud-owner" : cloud_owner, + "cloud-region" : cloud_region_id, + "other-connectivity-list" : {} + } + + with open(kubecfgfilepath, "rb") as kubecfgfile: + files = { + 'metadata': (None, json.dumps(metadata1)), + 'file': kubecfgfile + } + + resp = requests.post(multicloudK8sUrl+auth_api_url, files=files, verify=False) + if resp.status_code == 201: + self._logger.info("create k8sconnectivity for %s, %s, succeeds: %s" + % (cloud_owner, cloud_region_id, resp.content)) + else: + self._logger.warn("create k8sconnectivity for %s, %s, fails:%s" + % (cloud_owner, cloud_region_id, resp.content)) + #print(resp.content, resp.status_code) + + except HttpError as e: + self._logger.error("HttpError: status:%s, response:%s" + % (e.http_status, e.response.json())) + return [] + except Exception: + self._logger.error(traceback.format_exc()) + return [] + class InfraResourceAuditor(newton_registration.RegistryHelper): def __init__(self, multicloud_prefix, aai_base_url): |