kalavai-client 0.5.15__py3-none-any.whl → 0.5.17__py3-none-any.whl

kalavai_client/cluster.py

@@ -10,7 +10,16 @@ from kalavai_client.utils import (
     populate_template
 )
 
-from kalavai_client.env import user_path
+from kalavai_client.env import (
+    DEFAULT_CONTAINER_NAME,
+    KUBE_VERSION,
+    DEFAULT_FLANNEL_IFACE,
+    USER_COMPOSE_FILE,
+    USER_KUBECONFIG_FILE,
+    USER_LOCAL_SERVER_FILE,
+    USER_HELM_APPS_FILE,
+    user_path
+)
 
 
 class Cluster(ABC):
@@ -125,7 +134,7 @@ class dockerCluster(Cluster):
 
     def remove_agent(self):
         try:
-            run_cmd(f'docker compose -f {self.compose_file} down')
+            run_cmd(f'docker compose -f {self.compose_file} down --volumes')
             return True
         except:
             return False
@@ -171,6 +180,7 @@ class dockerCluster(Cluster):
 
         except:
             pass
+        time.sleep(5)
         return self.is_agent_running()
 
     def get_cluster_token(self):
@@ -331,3 +341,16 @@ class k3sCluster(Cluster):
             if not validate_poolconfig(self.poolconfig_file):
                 raise ValueError("Cache missconfigured. Run 'kalavai pool stop' to clear.")
         return True
+
+####################################################
+####################################################
+
+CLUSTER = dockerCluster(
+    container_name=DEFAULT_CONTAINER_NAME,
+    kube_version=KUBE_VERSION,
+    flannel_iface=DEFAULT_FLANNEL_IFACE,
+    compose_file=USER_COMPOSE_FILE,
+    kubeconfig_file=USER_KUBECONFIG_FILE,
+    poolconfig_file=USER_LOCAL_SERVER_FILE,
+    dependencies_file=USER_HELM_APPS_FILE
+)