add: save appwrappers to .codeflare/appwrapper/ directory.

Author: VanillaSpoon

src/codeflare_sdk/cluster/cluster.py

@@ -60,7 +60,9 @@ def __init__(self, config: ClusterConfiguration):
         """
         self.config = config
         self.app_wrapper_yaml = self.create_app_wrapper()
-        self.app_wrapper_name = self.app_wrapper_yaml.split(".")[0]
+        self.app_wrapper_name = self.app_wrapper_yaml.replace(".yaml", "").split("/")[
+            -1
+        ]
 
     def evaluate_dispatch_priority(self):
         priority_class = self.config.dispatch_priority

src/codeflare_sdk/utils/generate_yaml.py

@@ -19,6 +19,7 @@
 
 import yaml
 import sys
+import os
 import argparse
 import uuid
 from kubernetes import client, config
@@ -364,8 +365,14 @@ def disable_raycluster_tls(resources):
 
 
 def write_user_appwrapper(user_yaml, output_file_name):
+    # Create the directory if it doesn't exist
+    directory_path = os.path.dirname(output_file_name)
+    if not os.path.exists(directory_path):
+        os.makedirs(directory_path)
+
     with open(output_file_name, "w") as outfile:
         yaml.dump(user_yaml, outfile, default_flow_style=False)
+
     print(f"Written to: {output_file_name}")
 
 
@@ -433,6 +440,8 @@ def generate_appwrapper(
         enable_local_interactive(resources, cluster_name, namespace)
     else:
         disable_raycluster_tls(resources["resources"])
-    outfile = appwrapper_name + ".yaml"
+    directory_path = ".codeflare/appwrapper/"
+
+    outfile = os.path.join(directory_path, appwrapper_name + ".yaml")
     write_user_appwrapper(user_yaml, outfile)
     return outfile

tests/unit_test.py

@@ -19,6 +19,7 @@
 import re
 
 parent = Path(__file__).resolve().parents[1]
+aw_dir = ".codeflare/appwrapper/"
 sys.path.append(str(parent) + "/src")
 
 from kubernetes import client, config
@@ -241,10 +242,12 @@ def test_config_creation():
 
 def test_cluster_creation():
     cluster = createClusterWithConfig()
-    assert cluster.app_wrapper_yaml == "unit-test-cluster.yaml"
+    assert cluster.app_wrapper_yaml == f"{aw_dir}unit-test-cluster.yaml"
     assert cluster.app_wrapper_name == "unit-test-cluster"
     assert filecmp.cmp(
-        "unit-test-cluster.yaml", f"{parent}/tests/test-case.yaml", shallow=True
+        f"{aw_dir}unit-test-cluster.yaml",
+        f"{parent}/tests/test-case.yaml",
+        shallow=True,
     )
 
 
@@ -258,10 +261,12 @@ def test_cluster_creation_priority(mocker):
     config.name = "prio-test-cluster"
     config.dispatch_priority = "default"
     cluster = Cluster(config)
-    assert cluster.app_wrapper_yaml == "prio-test-cluster.yaml"
+    assert cluster.app_wrapper_yaml == f"{aw_dir}prio-test-cluster.yaml"
     assert cluster.app_wrapper_name == "prio-test-cluster"
     assert filecmp.cmp(
-        "prio-test-cluster.yaml", f"{parent}/tests/test-case-prio.yaml", shallow=True
+        f"{aw_dir}prio-test-cluster.yaml",
+        f"{parent}/tests/test-case-prio.yaml",
+        shallow=True,
     )
 
 
@@ -275,7 +280,7 @@ def test_default_cluster_creation(mocker):
     )
     cluster = Cluster(default_config)
 
-    assert cluster.app_wrapper_yaml == "unit-test-default-cluster.yaml"
+    assert cluster.app_wrapper_yaml == f"{aw_dir}unit-test-default-cluster.yaml"
     assert cluster.app_wrapper_name == "unit-test-default-cluster"
     assert cluster.config.namespace == "opendatahub"
 
@@ -285,7 +290,7 @@ def arg_check_apply_effect(group, version, namespace, plural, body, *args):
     assert version == "v1beta1"
     assert namespace == "ns"
     assert plural == "appwrappers"
-    with open("unit-test-cluster.yaml") as f:
+    with open(f"{aw_dir}unit-test-cluster.yaml") as f:
         aw = yaml.load(f, Loader=yaml.FullLoader)
     assert body == aw
     assert args == tuple()
@@ -2147,7 +2152,7 @@ def parse_j(cmd):
 
 
 def test_AWManager_creation():
-    testaw = AWManager("test.yaml")
+    testaw = AWManager(f"{aw_dir}test.yaml")
     assert testaw.name == "test"
     assert testaw.namespace == "ns"
     assert testaw.submitted == False
@@ -2171,7 +2176,7 @@ def arg_check_aw_apply_effect(group, version, namespace, plural, body, *args):
     assert version == "v1beta1"
     assert namespace == "ns"
     assert plural == "appwrappers"
-    with open("test.yaml") as f:
+    with open(f"{aw_dir}test.yaml") as f:
         aw = yaml.load(f, Loader=yaml.FullLoader)
     assert body == aw
     assert args == tuple()
@@ -2187,7 +2192,7 @@ def arg_check_aw_del_effect(group, version, namespace, plural, name, *args):
 
 
 def test_AWManager_submit_remove(mocker, capsys):
-    testaw = AWManager("test.yaml")
+    testaw = AWManager(f"{aw_dir}test.yaml")
     testaw.remove()
     captured = capsys.readouterr()
     assert (
@@ -2294,12 +2299,12 @@ def test_export_env():
 
 # Make sure to always keep this function last
 def test_cleanup():
-    os.remove("unit-test-cluster.yaml")
-    os.remove("prio-test-cluster.yaml")
-    os.remove("unit-test-default-cluster.yaml")
-    os.remove("test.yaml")
-    os.remove("raytest2.yaml")
-    os.remove("quicktest.yaml")
+    os.remove(f"{aw_dir}unit-test-cluster.yaml")
+    os.remove(f"{aw_dir}prio-test-cluster.yaml")
+    os.remove(f"{aw_dir}unit-test-default-cluster.yaml")
+    os.remove(f"{aw_dir}test.yaml")
+    os.remove(f"{aw_dir}raytest2.yaml")
+    os.remove(f"{aw_dir}quicktest.yaml")
     os.remove("tls-cluster-namespace/ca.crt")
     os.remove("tls-cluster-namespace/tls.crt")
     os.remove("tls-cluster-namespace/tls.key")