1 files changed, 58 insertions, 0 deletions
diff --git a/sdv/docker/sdvsecurity/nfvsec/utils/k8sclient.py b/sdv/docker/sdvsecurity/nfvsec/utils/k8sclient.py
new file mode 100644
index 0000000..b1eba5a
--- /dev/null
+++ b/sdv/docker/sdvsecurity/nfvsec/utils/k8sclient.py
@@ -0,0 +1,58 @@
+"""
+Kubernetes cluster api helper functions
+"""
+
+
+import time
+
+from kubernetes import client, config
+from kubernetes.client import Configuration
+from kubernetes.client.api import core_v1_api
+from kubernetes.client.rest import ApiException
+from kubernetes.stream import stream
+
+from kubernetes.stream import stream
+import logging
+from conf import settings    # pylint: disable=import-error
+
+
+class K8sClient():
+    """
+    Class for controlling the pod through PAPI
+    """
+
+    def __init__(self):
+        """
+        Initialisation function.
+        """
+        self._logger = logging.getLogger(__name__)
+        config.load_kube_config(settings.getValue('K8S_CONFIG_FILEPATH'))
+        self.api = client.CoreV1Api()
+
+    def get_pod(self, namespace, name):
+        """
+        Returns json details any one pod with matching label
+
+        :param namespace: namespace to use
+        :param namespace: name of the pod (Longest possible).
+        :return: pod details
+        """
+        api_response = self.api.list_namespaced_pod(namespace)
+        for pod in api_response.items:
+            #print(pod.metadata.name)
+            if pod.metadata.name.startswith(name):
+                return pod
+        return None
+
+
+    def execute(self, pod, cmd):
+        """
+        Executes `cmd` inside `pod` and returns response
+        :param pod: pod object
+        :param cmd: command to execute inside pod
+        :return: response from pod
+        """
+        response = stream(self.api.connect_get_namespaced_pod_exec,
+                        pod.metadata.name, pod.metadata.namespace, command=cmd,
+                        stderr=True, stdin=False, stdout=True, tty=False)
+        return response