Merge remote-tracking branch 'origin/agis/expose-overwite-save-profile' into agis/upd-2.30.0-0.36.0

Author: kounelisagis

tiledb/libtiledb/profile.cc

@@ -39,7 +39,7 @@ void init_profile(py::module& m) {
 
         .def("_get_param", &tiledb::Profile::get_param, py::arg("param"))
 
-        .def("_save", &tiledb::Profile::save)
+        .def("_save", &tiledb::Profile::save, py::arg("overwrite"))
 
         .def_static(
             "_load",

tiledb/profile.py

@@ -74,12 +74,13 @@ def get(self, param: str, raise_keyerror: bool = True):
 
         return val
 
-    def save(self):
+    def save(self, overwrite: bool = False):
         """Saves the profile to storage.
 
+        :param overwrite: Whether to overwrite an existing profile. Defaults to False.
         :raises tiledb.TileDBError:
         """
-        self._save()
+        self._save(overwrite)
 
     @classmethod
     def load(cls, name: str = None, dir: str = None) -> "Profile":

tiledb/tests/test_profile.py

@@ -101,6 +101,37 @@ def test_profile_save_load_remove(self):
         # remove the profile
         tiledb.Profile.remove("profile2_name", self.path("profile2_dir"))
 
+    def test_profile_save_overwrite(self):
+        token1 = "testing_the_token_1"
+        token2 = "testing_the_token_2"
+        profile_name = "overwrite_test_profile"
+        profile_dir = self.path("overwrite_test_dir")
+
+        # Create and save a profile with token1
+        profile1 = tiledb.Profile(profile_name, profile_dir)
+        profile1["rest.token"] = token1
+        profile1.save()
+
+        # Load it back to verify it was saved
+        loaded_profile1 = tiledb.Profile.load(profile_name, profile_dir)
+        assert loaded_profile1["rest.token"] == token1
+
+        # Create a new profile with the same name and try to save without overwrite
+        profile2 = tiledb.Profile(profile_name, profile_dir)
+        profile2["rest.token"] = token2
+        with pytest.raises(tiledb.TileDBError):
+            profile2.save(overwrite=False)
+
+        # Now save with overwrite=True
+        profile2.save(overwrite=True)
+
+        # Load it back to verify it was overwritten
+        loaded_profile2 = tiledb.Profile.load(profile_name, profile_dir)
+        assert loaded_profile2["rest.token"] == token2
+
+        # Clean up
+        tiledb.Profile.remove(profile_name, profile_dir)
+
 
 class ConfigWithProfileTest(ProfileTestCase):
     def test_config_with_profile(self):