adding validation for local_queue provided in cluster config

Author: Fiona-Waters

Diff for: src/codeflare_sdk/utils/generate_yaml.py

@@ -308,6 +308,24 @@ def get_default_kueue_name(namespace: str):
     )
 
 
+def local_queue_exists(namespace, local_queue_name: str):
+    try:
+        config_check()
+        api_instance = client.CustomObjectsApi(api_config_handler())
+        local_queues = api_instance.list_namespaced_custom_object(
+            group="kueue.x-k8s.io",
+            version="v1beta1",
+            namespace=namespace,
+            plural="localqueues",
+        )
+    except Exception as e:  # pragma: no cover
+        return _kube_api_error_handling(e)
+    for lq in local_queues["items"]:
+        if lq["metadata"]["name"] == local_queue_name:
+            return True
+    return False
+
+
 def write_components(
     user_yaml: dict,
     output_file_name: str,
@@ -324,24 +342,29 @@ def write_components(
     open(output_file_name, "w").close()
     lq_name = local_queue or get_default_kueue_name(namespace)
     cluster_labels = labels
-    with open(output_file_name, "a") as outfile:
-        for component in components:
-            if "generictemplate" in component:
-                if (
-                    "workload.codeflare.dev/appwrapper"
-                    in component["generictemplate"]["metadata"]["labels"]
-                ):
-                    del component["generictemplate"]["metadata"]["labels"][
+    if local_queue_exists(namespace, lq_name):
+        with open(output_file_name, "a") as outfile:
+            for component in components:
+                if "generictemplate" in component:
+                    if (
                         "workload.codeflare.dev/appwrapper"
-                    ]
-                    labels = component["generictemplate"]["metadata"]["labels"]
-                    labels.update({"kueue.x-k8s.io/queue-name": lq_name})
-                    labels.update(cluster_labels)
-                outfile.write("---\n")
-                yaml.dump(
-                    component["generictemplate"], outfile, default_flow_style=False
-                )
-    print(f"Written to: {output_file_name}")
+                        in component["generictemplate"]["metadata"]["labels"]
+                    ):
+                        del component["generictemplate"]["metadata"]["labels"][
+                            "workload.codeflare.dev/appwrapper"
+                        ]
+                        labels = component["generictemplate"]["metadata"]["labels"]
+                        labels.update({"kueue.x-k8s.io/queue-name": lq_name})
+                        labels.update(cluster_labels)
+                    outfile.write("---\n")
+                    yaml.dump(
+                        component["generictemplate"], outfile, default_flow_style=False
+                    )
+        print(f"Written to: {output_file_name}")
+    else:
+        raise ValueError(
+            "local_queue provided does not exist. Please provide the correct local_queue name in Cluster Configuration"
+        )
 
 
 def load_components(
@@ -355,26 +378,31 @@ def load_components(
     components = user_yaml.get("spec", "resources")["resources"].get("GenericItems")
     lq_name = local_queue or get_default_kueue_name(namespace)
     cluster_labels = labels
-    for component in components:
-        if "generictemplate" in component:
-            if (
-                "workload.codeflare.dev/appwrapper"
-                in component["generictemplate"]["metadata"]["labels"]
-            ):
-                del component["generictemplate"]["metadata"]["labels"][
+    if local_queue_exists(namespace, lq_name):
+        for component in components:
+            if "generictemplate" in component:
+                if (
                     "workload.codeflare.dev/appwrapper"
-                ]
-                labels = component["generictemplate"]["metadata"]["labels"]
-                labels.update({"kueue.x-k8s.io/queue-name": lq_name})
-                labels.update(cluster_labels)
-            component_list.append(component["generictemplate"])
-
-    resources = "---\n" + "---\n".join(
-        [yaml.dump(component) for component in component_list]
-    )
-    user_yaml = resources
-    print(f"Yaml resources loaded for {name}")
-    return user_yaml
+                    in component["generictemplate"]["metadata"]["labels"]
+                ):
+                    del component["generictemplate"]["metadata"]["labels"][
+                        "workload.codeflare.dev/appwrapper"
+                    ]
+                    labels = component["generictemplate"]["metadata"]["labels"]
+                    labels.update({"kueue.x-k8s.io/queue-name": lq_name})
+                    labels.update(cluster_labels)
+                component_list.append(component["generictemplate"])
+
+        resources = "---\n" + "---\n".join(
+            [yaml.dump(component) for component in component_list]
+        )
+        user_yaml = resources
+        print(f"Yaml resources loaded for {name}")
+        return user_yaml
+    else:
+        raise ValueError(
+            "local_queue provided does not exist. Please provide the correct local_queue name in Cluster Configuration"
+        )
 
 
 def load_appwrapper(user_yaml: dict, name: str):