From cc76673c24ef29d1450d92147f8a67fd1fda98c1 Mon Sep 17 00:00:00 2001
From: iamashwin99 <ashwin-kumar.karnad@mpsd.mpg.de>
Date: Fri, 23 Jun 2023 11:27:07 +0200
Subject: [PATCH] ask for the prompt only if force not present

---
 src/mpsd_software_manager/mpsd_software.py | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/src/mpsd_software_manager/mpsd_software.py b/src/mpsd_software_manager/mpsd_software.py
index 73b4f4d..f264cc4 100755
--- a/src/mpsd_software_manager/mpsd_software.py
+++ b/src/mpsd_software_manager/mpsd_software.py
@@ -1093,18 +1093,20 @@ def remove_environment(mpsd_release, root_dir, package_sets="NONE", force_remove
         sys.exit(1)
     if "ALL" in package_sets:
         # we need to remove the entire release folder
-        logging.warning(
-            f"Removing release {mpsd_release} from {root_dir}"
-            "do you want to continue? [y/n]"
-        )
-        if force_remove or input().lower() == "y":
-            folders_to_remove = os.listdir(root_dir / mpsd_release)
+        logging.warning(f"Removing release {mpsd_release} from {root_dir}")
+        if not force_remove:
+            logging.warning("do you want to continue? [y/n]")
+            if input().lower() == "y":
+                force_remove = True
+        if force_remove:
+            dir_to_remove = root_dir / mpsd_release / get_native_microarchitecture()
+            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(root_dir / mpsd_release / folder)
-                sys.exit(0)
+                shutil.rmtree(dir_to_remove / folder)
+            return
     for package_set in package_sets:
         # we load the spack environment and remove the package_set
         spack_env = ""
-- 
GitLab