Exclude scratch folder when clean multitest runpath.

morganstanley · Jun 27, 2024 · 0a4ba7c · 0a4ba7c
1 parent 0ff900a
commit 0a4ba7c
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/testplan/common/utils/helper.py b/testplan/common/utils/helper.py
@@ -222,7 +222,16 @@ def clean_runpath_if_passed(
     """
     multitest = env.parent
     if multitest.report.passed:
-        shutil.rmtree(multitest.runpath, ignore_errors=True)
+        for subfile in os.listdir(multitest.runpath):
+            # TODO: Define scratch as a constant
+            if subfile != "scratch":
+                path = os.path.join(
+                    os.path.abspath(multitest.runpath), subfile
+                )
+                if os.path.isfile(path) or os.path.islink(path):
+                    os.remove(path)
+                elif os.path.isdir(path):
+                    shutil.rmtree(path, ignore_errors=True)
 
 
 @testsuite