auto_chaos.k8s_system

  1"""
  2☸️
  3K8s system
  4☸️
  5"""
  6from kubernetes import client, config
  7
  8from auto_chaos.base_system import BaseSystem
  9
 10
 11class K8sSystem(BaseSystem):
 12    """
 13    K8s actions classe
 14    """
 15
 16    def __init__(self, namespace: str = None) -> None:
 17        """
 18        Init
 19
 20        Args:
 21            namespace (str, optional): Namespace in which to do things. Defaults to None => cluster scope.
 22        """
 23        # Configs can be set in Configuration class directly or using helper utility
 24        config.load_kube_config()
 25        self.client = client.CoreV1Api()
 26        self.namespace = namespace
 27        super().__init__()
 28
 29    def describe(self, args: list[str] = None):
 30        """
 31        Describe a k8s cluster
 32
 33        Args:
 34            args (list[str], optional): Not used. Defaults to None.
 35        """
 36        result = {"pods": {}, "services": {}, "nodes": {}}
 37
 38        # List cluster nodes
 39        ret = self.client.list_node()
 40        for node in ret.items:
 41            result["nodes"][node.metadata.name] = {
 42                "ip": node.status.addresses[0].address,
 43                "status": node.status.phase,
 44            }
 45            print("%s" % (node.metadata.name))
 46
 47        # List namespaced pods or cluster scope
 48        if self.namespace:
 49            ret = self.client.list_namespaced_pod(self.namespace)
 50        else:
 51            ret = self.client.list_pod_for_all_namespaces(watch=False)
 52        for i in ret.items:
 53            result["pods"][i.metadata.name] = {
 54                "ip": i.status.pod_ip,
 55                "namespace": i.metadata.namespace,
 56                "name": i.metadata.name,
 57                "status": i.status.phase,
 58            }
 59            print(
 60                "%s\t%s\t%s" % (i.status.pod_ip, i.metadata.namespace, i.metadata.name)
 61            )
 62
 63        # List namespaced services or cluster scope
 64        if self.namespace:
 65            ret = self.client.list_namespaced_service(
 66                namespace=self.namespace, watch=False
 67            )
 68            for i in ret.items:
 69                result["services"][i.metadata.name] = {
 70                    "name": i.metadata.name,
 71                }
 72                print("%s" % (i.metadata.name))
 73        else:
 74            for namespace in self.client.list_namespace().items:
 75                ret = self.client.list_namespaced_service(
 76                    namespace=namespace.metadata.name, watch=False
 77                )
 78                for i in ret.items:
 79                    result["services"][i.metadata.name] = {
 80                        "name": i.metadata.name,
 81                    }
 82                    print("%s" % (i.metadata.name))
 83
 84        # Add result to results
 85        self.results.append(result)
 86
 87    def kill(self, args: list[str] = None):
 88        """
 89        Kill a k8s ressource (pod, service, node)
 90
 91        Args:
 92            args (list[str], optional): Resource type and name. Defaults to None.
 93        """
 94        resource_type = args[0]
 95        resource_name = args[1]
 96
 97        result = None
 98        try:
 99            # Delete ressource
100            if resource_type == "pod":
101                result = self.client.delete_namespaced_pod(
102                    name=resource_name, namespace=self.namespace, grace_period_seconds=0
103                )
104            if resource_type == "service":
105                result = self.client.delete_namespaced_service(
106                    name=resource_name, namespace=self.namespace, grace_period_seconds=0
107                )
108            if resource_type == "node":
109                result = self.client.delete_node(
110                    name=resource_name, grace_period_seconds=0
111                )
112            # Add result to results
113            self.results.append(result.status.phase)
114        except Exception as error:
115            self.errors.append(error)