much work done

src/kuberclient.py

@@ -0,0 +1,99 @@
+#!/usr/bin/env python3
+
+"""
+Kubernetes Client Tools
+"""
+
+__author__ = "@tormakris"
+__copyright__ = "Copyright 2020, videON Team"
+__module_name__ = "kuberclient"
+__version__text__ = "1"
+
+from kubernetes import client, config
+
+
+class Kubectl:
+    def __init__(self, name: str, resourcetype: str = "ingest", stream_key: str = "test", push_urls: str = "",
+                 ffmpeg_args: str = "", encode_push_urls: str = ""):
+        config.load_incluster_config()
+        self.name = name
+        self.streamtype = resourcetype.lower(),
+        self.stream_key = stream_key,
+        self.push_urls = push_urls,
+        self.ffmpeg_args = ffmpeg_args,
+        self.encode_push_url = encode_push_urls
+
+    def __delete_pod(self):
+        foundpods = client.CoreV1Api().list_namespaced_pod("videon", label_selector=f"app in ({self.name})")
+        for pod in foundpods.items:
+            client.CoreV1Api().delete_namespaced_pod(pod.metadata.name, "videon")
+
+    def __create_configmap(self):
+        configmap = client.V1ConfigMap(api_version="v1", kind="ConfigMap",
+                                       data={"TYPE": self.streamtype, "STREAM_KEY": self.stream_key,
+                                             "PUSH_URLS": self.push_urls, "FFMPEG_ARGS": self.ffmpeg_args,
+                                             "ENCODE_PUSH_URL": self.encode_push_url})
+        client.CoreV1Api().create_namespaced_config_map(namespace="videon", name=self.name, body=configmap)
+
+    def __update_configmap(self):
+        configmap = client.V1ConfigMap(api_version="v1", kind="ConfigMap",
+                                       data={"TYPE": self.streamtype, "STREAM_KEY": self.stream_key,
+                                             "PUSH_URLS": self.push_urls, "FFMPEG_ARGS": self.ffmpeg_args,
+                                             "ENCODE_PUSH_URL": self.encode_push_url})
+        client.CoreV1Api().patch_namespaced_config_map(namespace="videon",name=self.name, body=configmap)
+
+    def __create_deployment(self):
+        envs = [client.V1EnvFromSource(config_map_ref=client.V1ConfigMapEnvSource(name=self.name))]
+        container = client.V1Container(
+            name=self.name,
+            image="registry.kmlabz.com/videon/nginx-streamer",
+            image_pull_policy="Always",
+            ports=[client.V1ContainerPort(container_port=1935)],
+            env_from=envs,
+        )
+        # Template
+        template = client.V1PodTemplateSpec(
+            metadata=client.V1ObjectMeta(labels={"app": self.name}),
+            spec=client.V1PodSpec(containers=[container]))
+        # Spec
+        spec = client.V1DeploymentSpec(
+            replicas=1,
+            template=template)
+        # Deployment
+        deployment = client.V1Deployment(
+            api_version="apps/v1",
+            kind="Deployment",
+            metadata=client.V1ObjectMeta(name=self.name),
+            spec=spec)
+        # Creation of the Deployment in specified namespace
+        client.AppsV1Api().create_namespaced_deployment(
+            namespace="videon", body=deployment
+        )
+
+    def __create_service(self):
+        core_v1_api = client.CoreV1Api()
+        body = client.V1Service(
+            api_version="v1",
+            kind="Service",
+            metadata=client.V1ObjectMeta(
+                name=self.name
+            ),
+            spec=client.V1ServiceSpec(
+                selector={"app": self.name},
+                ports=[client.V1ServicePort(
+                    port=1935,
+                    target_port=1935
+                )]
+            )
+        )
+        # Creation of the Service in specified namespace
+        core_v1_api.create_namespaced_service(namespace="videon", body=body)
+
+    def create_resource(self):
+        self.__create_configmap()
+        self.__create_deployment()
+        self.__create_service()
+
+    def update_resource(self):
+        self.__update_configmap()
+        self.__delete_pod()