Merge pull request #155 from robusta-dev/kubernetes-proxy-fix

Patch fix for proxy not working from kubeconfig

robusta_krr/core/integrations/kubernetes.py → robusta_krr/core/integrations/kubernetes/__init__.py

@@ -22,6 +22,7 @@
 from robusta_krr.core.models.result import ResourceAllocations
 from robusta_krr.utils.configurable import Configurable
 
+from . import config_patch as _
 from .rollout import RolloutAppsV1Api
 
 AnyKubernetesAPIObject = Union[V1Deployment, V1DaemonSet, V1StatefulSet, V1Pod, V1Job]

robusta_krr/core/integrations/kubernetes/config_patch.py

@@ -0,0 +1,37 @@
+# NOTE: This is a workaround for the issue described here:
+# https://github.com/kubernetes-client/python/pull/1863
+
+from __future__ import annotations
+
+from kubernetes.client import configuration
+from kubernetes.config import kube_config
+
+
+class KubeConfigLoader(kube_config.KubeConfigLoader):
+    def _load_cluster_info(self):
+        super()._load_cluster_info()
+
+        if "proxy-url" in self._cluster:
+            self.proxy = self._cluster["proxy-url"]
+
+    def _set_config(self, client_configuration: Configuration):
+        super()._set_config(client_configuration)
+
+        key = "proxy"
+        if key in self.__dict__:
+            setattr(client_configuration, key, getattr(self, key))
+
+
+class Configuration(configuration.Configuration):
+    def __init__(
+        self,
+        proxy: str | None = None,
+        **kwargs,
+    ):
+        super().__init__(**kwargs)
+
+        self.proxy = proxy
+
+
+configuration.Configuration = Configuration
+kube_config.KubeConfigLoader = KubeConfigLoader