1 files changed, 85 insertions, 0 deletions
diff --git a/sdv/docker/sdvstate/tools/kube_utils.py b/sdv/docker/sdvstate/tools/kube_utils.py
new file mode 100644
index 0000000..3468ac6
--- /dev/null
+++ b/sdv/docker/sdvstate/tools/kube_utils.py
@@ -0,0 +1,85 @@
+# Copyright 2020 University Of Delhi.
+#
+# 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.
+
+
+
+"""
+Kubernetes cluster api helper functions
+"""
+
+from kubernetes import client, config
+from kubernetes.stream import stream
+
+from tools.conf import settings    # pylint: disable=import-error
+
+
+def load_kube_api():
+    """
+    Loads kubernetes api
+    """
+    config.load_kube_config(settings.getValue('kube_config'))
+    api = client.CoreV1Api()
+    settings.setValue('kube_api', api)
+
+
+def kube_api():
+    """
+    Returns kube_api object
+    """
+    return settings.getValue('kube_api')
+
+
+def get_pod_with_labels(labels):
+    """
+    Returns json details any one pod with matching labels
+
+    :param labels: labels to find matching pod
+    :return: pod details
+    """
+    api = kube_api()
+    pod = api.list_pod_for_all_namespaces(label_selector=labels).items[0]
+    return pod
+
+
+def kube_exec(pod, cmd):
+    """
+    Executes `cmd` inside `pod` and returns response
+
+    :param pod: pod object
+    :param cmd: command to execute inside pod
+    :return: response from pod
+    """
+    api = kube_api()
+    response = stream(api.connect_get_namespaced_pod_exec,
+                      pod.metadata.name, pod.metadata.namespace, command=cmd,
+                      stderr=True, stdin=False, stdout=True, tty=False)
+    return response
+
+
+def kube_curl(*args):
+    """
+    executes curl cmd in kubernetes network
+
+    :param args: comma separated list of args to pass to curl
+    :return: http response
+    """
+    args = list(args)
+    args.insert(0, "curl")
+
+    labels = "application=prometheus-openstack-exporter,component=exporter"
+    pod = get_pod_with_labels(labels)
+
+    response = kube_exec(pod, args)
+
+    return response