From 7493a95c33c9dbe591d2d96fa53cb5d858cf4d91 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 5689396..e8e5b4c 100755
--- a/src/mpsd_software_manager/mpsd_software.py
+++ b/src/mpsd_software_manager/mpsd_software.py
@@ -1085,18 +1085,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