Automation scripts for vFW_CNF_CDS usecase
[demo.git] / heat / vFW_CNF_CDS / automation / create_k8s_region.py
diff --git a/heat/vFW_CNF_CDS/automation/create_k8s_region.py b/heat/vFW_CNF_CDS/automation/create_k8s_region.py
new file mode 100755 (executable)
index 0000000..cfbec8c
--- /dev/null
@@ -0,0 +1,127 @@
+# ============LICENSE_START=======================================================
+# Copyright (C) 2020 Orange
+# ================================================================================
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# ============LICENSE_END=========================================================
+
+import logging
+import os
+from uuid import uuid4
+
+from config import Config
+from k8s_client import K8sClient
+from so_db_adapter import SoDBAdapter
+from onapsdk.aai.business import Customer
+from onapsdk.aai.cloud_infrastructure import Complex, CloudRegion
+from onapsdk.msb.k8s import ConnectivityInfo
+
+logger = logging.getLogger("")
+logger.setLevel(logging.DEBUG)
+fh = logging.StreamHandler()
+fh_formatter = logging.Formatter('%(asctime)s %(levelname)s %(lineno)d:%(filename)s(%(process)d) - %(message)s')
+fh.setFormatter(fh_formatter)
+logger.addHandler(fh)
+
+MYPATH = os.path.dirname(os.path.realpath(__file__))
+
+#### Create complex if not exists ####
+logger.info("******** Complex *******")
+try:
+    complex = list(Complex.get_all(physical_location_id=Config.COMPLEX_ID))[0]
+    logger.info("Complex exists")
+except IndexError:
+    logger.info("Complex does not exists")
+    complex = Complex.create(physical_location_id=Config.COMPLEX_ID,
+                             name=Config.COMPLEX_ID,
+                             physical_location_type="office",
+                             street1="DummyStreet 1",
+                             city="DummyCity",
+                             postal_code="00-000",
+                             country="DummyCountry",
+                             region="DummyRegion")
+    logger.info("Complex created")
+
+#### Create cloud region if not exists ####
+logger.info("******** Cloud Region *******")
+try:
+    cloud_region = list(CloudRegion.get_all(cloud_owner=Config.CLOUD_OWNER, cloud_region_id=Config.CLOUD_REGION))[0]
+    logger.info("Cloud region exists")
+except IndexError:
+    logger.info("Cloud region does not exists")
+    cloud_region = CloudRegion.create(cloud_owner=Config.CLOUD_OWNER,
+                                      cloud_region_id=Config.CLOUD_REGION,
+                                      cloud_type="k8s",
+                                      owner_defined_type="t1",
+                                      cloud_region_version="1.0",
+                                      complex_name=complex.physical_location_id,
+                                      cloud_zone="CloudZone",
+                                      sriov_automation="false",
+                                      orchestration_disabled=False,
+                                      in_maint=False)
+    logger.info("Cloud region created")
+
+logger.info("******** Cloud regiongion <-> Complex *******")
+cloud_region.link_to_complex(complex)
+
+logger.info("******** Availability zone *******")
+cloud_region.add_availability_zone(availability_zone_name=Config.AVAILABILITY_ZONE_NAME,
+                                   availability_zone_hypervisor_type=Config.HYPERVISOR_TYPE)
+
+logger.info("******** Tenant *******")
+cloud_region.add_tenant(str(uuid4()), Config.TENANT_NAME)
+
+#### Update or create connectivity info ####
+logger.info("******** Connectivity Info *******")
+with open(os.path.join(MYPATH, Config.CLUSTER_KUBECONFIG_PATH), 'rb') as kubeconfig_file:
+    kubeconfig = kubeconfig_file.read()
+try:
+    connectivity_info = ConnectivityInfo.get_connectivity_info_by_region_id(cloud_region_id=Config.CLOUD_REGION)
+    logger.info("Connectivity Info exists ")
+    logger.info("Delete Connectivity Info ")
+    connectivity_info.delete()
+    connectivity_info = ConnectivityInfo.create(cloud_region_id=Config.CLOUD_REGION,
+                                                cloud_owner=Config.CLOUD_OWNER,
+                                                kubeconfig=kubeconfig)
+    logger.info("Connectivity Info created ")
+except:
+    logger.info("Connectivity Info does not exists ")
+    connectivity_info = ConnectivityInfo.create(cloud_region_id=Config.CLOUD_REGION,
+                                                cloud_owner=Config.CLOUD_OWNER,
+                                                kubeconfig=kubeconfig)
+    logger.info("Connectivity Info created ")
+
+#### Add Custom Resource Definitions ####
+k8s_client = K8sClient(kubeconfig_path=Config.CLUSTER_KUBECONFIG_PATH)
+for crd in Config.CUSTOMER_RESOURCE_DEFINITIONS:
+    k8s_client.create_custom_object(crd)
+
+#### Create customer if not exists ####
+logger.info("******** Customer *******")
+try:
+    customer = Customer.get_by_global_customer_id(Config.GLOBAL_CUSTOMER_ID)
+    logger.info("Customer exists")
+except:
+    logger.info("Customer exists")
+    customer = Customer.create(Config.GLOBAL_CUSTOMER_ID, Config.GLOBAL_CUSTOMER_ID, "INFRA")
+    logger.info("Customer created")
+
+#### Add region to SO db ####
+logger.info("******** SO Database *******")
+so_db_adapter = SoDBAdapter(cloud_region_id=Config.CLOUD_REGION,
+                            complex_id=Config.COMPLEX_ID,
+                            onap_kubeconfig_path=Config.ONAP_KUBECONFIG_PATH)
+is_region_in_so = so_db_adapter.check_region_in_db()
+
+if not is_region_in_so:
+    so_db_adapter.add_region_to_so_db()