Remove cmd

src/mpsd_software_manager/mpsd_software.py

@@ -14,7 +14,6 @@ import time
 from pathlib import Path
 from typing import List, Tuple, Union
 import re
-import shutil
 from functools import cache
 import importlib.metadata
 
@@ -1072,23 +1071,47 @@ def install_environment(
 
 
 def remove_environment(mpsd_release, root_dir, package_sets="NONE", force_remove=False):
-    """Remove release from installation."""
+    """Remove release from installation.
+
+    Handle 3 situations :
+    1. remove dosent specify what to remove
+        -> warn and exit
+    2. remove all package_sets from release
+        -> remove release folder except logs
+    3. remove specific package_sets from release
+        -> remove spack environments via spack commands
+
+    Parameters
+    ----------
+    mpsd_release : str
+        A string representing the MPSD release version.
+    root_dir : pathlib.Path
+        A Path object representing the path to the directory where
+        the release and package_sets will be installed.
+    package_sets : list of str
+        A list of strings representing the package_sets to install
+        (e.g., "foss2021a-mpi", "global_generic", "ALL").
+    force_remove : bool, optional
+        A boolean indicating whether to force remove the release.
+        If False, the user will be prompted to confirm the removal.
+        Defaults to False.
+
+    Raises
+    ------
+    ValueError
+
+    """
     msg = (
         f"Removing release {mpsd_release}"
         f" with package_sets {package_sets} from {root_dir}"
     )
     logging.warning(msg)
-    # Handle 3 situations :
-    # 1. remove dosent specify what to remove  -> warn and exit
-    # 2. remove all package_sets from release -> remove release folder except logs
-    # 3. remove specific package_sets from release
-    #       -> remove spack environments via spack commands
 
     if package_sets == "NONE":
         logging.warning(
             "Please specify package_sets to remove, or 'ALL' to remove all toolchains"
         )
-        sys.exit(1)
+        sys.exit(50)  # TODO document this code.
     # 2nd case: remove the entire release for microarchitecture
     dir_to_remove = root_dir / mpsd_release / get_native_microarchitecture()
     if "ALL" in package_sets:
@@ -1100,14 +1123,15 @@ def remove_environment(mpsd_release, root_dir, package_sets="NONE", force_remove
         if not force_remove:
             logging.warning("do you want to continue? [y/n]")
             if input().lower() != "y":
-                sys.exit(50)  # TODO document this code.
+                sys.exit(60)  # TODO document this code.
 
         folders_to_remove = os.listdir(dir_to_remove)
         # skip logs folder
         # if "logs" in folders_to_remove:
         #     folders_to_remove.remove("logs")
         for folder in folders_to_remove:
-            shutil.rmtree(dir_to_remove / folder)
+            # shutil.rmtree(dir_to_remove / folder) #dosent delete file
+            run(f"rm -rf {dir_to_remove / folder}", shell=True, check=True)
         logging.warning(f"Removed release {mpsd_release} from {root_dir}")
         return
     # 3rd case: remove specific package_sets from release
@@ -1125,7 +1149,8 @@ def remove_spack_environment(spack_dir, environment_name):
     logging.warning(f"Removing spack environment {environment_name}")
     spack_env = spack_dir / "share" / "spack" / "setup-env.sh"
     commands_to_execute = [
-        f"source {spack_env}",
+        f"export SPACK_DIR={spack_dir}",
+        f". {spack_env}",
         f"spack env activate {environment_name}",
         f"for spec in $(spack -e {environment_name} find"  # this line continues
         r' --format "{name}@{version}%{compiler.name}@{compiler.version}");do'