Automation scripts for vFW_CNF_CDS usecase

[demo.git] / heat / vFW_CNF_CDS / automation / so_db_adapter.py

diff --git a/heat/vFW_CNF_CDS/automation/so_db_adapter.py b/heat/vFW_CNF_CDS/automation/so_db_adapter.py
new file mode 100755 (executable)
index 0000000..a829fbd
--- /dev/null
+++ b/heat/vFW_CNF_CDS/automation/so_db_adapter.py
@@ -0,0 +1,92 @@
+# ============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 base64
+import os
+
+from kubernetes import config, client
+from kubernetes.stream import stream
+
+
+class SoDBAdapter:
+
+    def __init__(self, cloud_region_id, complex_id, onap_kubeconfig_path):
+        self.CLOUD_REGION_ID = cloud_region_id
+        self.COMPLEX_ID = complex_id
+        self.ONAP_KUBECONFIG_PATH = onap_kubeconfig_path
+        self.MYPATH = os.path.dirname(os.path.realpath(__file__))
+
+        config.load_kube_config(config_file=os.path.join(self.MYPATH, self.ONAP_KUBECONFIG_PATH))
+        self.api_instance = client.CoreV1Api()
+        self.pod_name = self.get_mariadb_pod_name()
+        self.password = self.get_mariadb_root_username_password()
+
+    def get_mariadb_pod_name(self):
+        pods = self.api_instance.list_namespaced_pod(namespace="onap")
+        for pod in pods.items:
+            if pod.metadata.name.find("mariadb-galera-0") != -1:
+                return pod.metadata.name
+
+    def get_mariadb_root_username_password(self):
+        secrets = self.api_instance.list_namespaced_secret(namespace="onap")
+        for secret in secrets.items:
+            if secret.metadata.name.find("mariadb-galera-db-root-password") != -1:
+                base64_password = secret.data["password"]
+                base64_bytes = base64_password.encode('ascii')
+                password_bytes = base64.b64decode(base64_bytes)
+
+                return password_bytes.decode('ascii')
+
+    def run_exec_request(self, exec_command):
+        response = stream(self.api_instance.connect_get_namespaced_pod_exec,
+                          name=self.pod_name,
+                          # container="container-name",
+                          namespace="onap",
+                          command=exec_command,
+                          stdin=False,
+                          tty=False,
+                          stderr=True,
+                          stdout=True)
+        return response
+
+    def check_region_in_db(self):
+        exec_command = [
+            "/bin/sh",
+            "-c",
+            f"mysql -uroot -p{self.password} catalogdb -e 'SELECT * FROM cloud_sites;'"]
+        response = self.run_exec_request(exec_command)
+
+        is_region_found = False
+        for line in response.split("\n"):
+            if line.split("\t")[0] == self.CLOUD_REGION_ID:
+                print(line)
+                is_region_found = True
+                return is_region_found
+        return is_region_found
+
+    def add_region_to_so_db(self):
+        exec_command = [
+            "/bin/sh",
+            "-c",
+            f"mysql -uroot -p{self.password} catalogdb -e "
+            f"'insert into cloud_sites(ID, REGION_ID, IDENTITY_SERVICE_ID, CLOUD_VERSION, CLLI, ORCHESTRATOR ) "
+            f"values (\"{self.CLOUD_REGION_ID}\", \"{self.CLOUD_REGION_ID}\", \"DEFAULT_KEYSTONE\", \"2.5\", "
+            f"\"{self.COMPLEX_ID}\", \"multicloud\");'"]
+
+        response = self.run_exec_request(exec_command)
+
+        return response