From 9325cad13dda7e81f24ce9e1da083b8af25d39d9 Mon Sep 17 00:00:00 2001
From: Klaus Reuter <khr@mpcdf.mpg.de>
Date: Wed, 25 Oct 2023 18:00:25 +0200
Subject: [PATCH] enable steps functionality for cleanup step 7)

---
 condainer/condainer.py | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/condainer/condainer.py b/condainer/condainer.py
index 558a951..222fdc4 100644
--- a/condainer/condainer.py
+++ b/condainer/condainer.py
@@ -323,8 +323,8 @@ def build(args):
         print(f"STOP. Mount point {env_directory} is in use, please unmount first.")
         sys.exit(1)
     else:
+        steps = {int(i) for i in args.steps.split(',')}
         try:
-            steps = {int(i) for i in args.steps.split(',')}
             if not args.quiet:
                 print(termcol.BOLD+"Starting Condainer build process ..."+termcol.ENDC)
             if not args.dryrun:
@@ -360,12 +360,13 @@ def build(args):
         except:
             raise
         finally:
-            if not args.quiet:
-                print(termcol.BOLD+termcol.CYAN+"7) Cleaning up ..."+termcol.ENDC)
-            if args.dryrun:
-                print("dryrun: skipping")
-            else:
-                shutil.rmtree(env_directory)
+            if 7 in steps:
+                if not args.quiet:
+                    print(termcol.BOLD+termcol.CYAN+"7) Cleaning up ..."+termcol.ENDC)
+                if args.dryrun:
+                    print("dryrun: skipping")
+                else:
+                    shutil.rmtree(env_directory)
             if not args.quiet:
                 print(termcol.BOLD+"Done!"+termcol.ENDC)
 
-- 
GitLab