commit 15f3a60234bf40179052ffda8edc49dd1a53d1bb
author Srikrishna Iyer <sriyer@google.com> Tue Mar 30 23:57:27 2021 -0700
committer Srikrishna Iyer <46467186+sriyerg@users.noreply.github.com> Fri Apr 02 10:30:48 2021 -0700
tree 9625cb8c2202bce12c81333cd0a1d7212da2a6c2
parent 8ca91da429ee1faf65ccaed1e1a656b16372f287

[dvsim] Minor fix in clean_odirs function

The `clean_odirs` method takes `max_odirs` as an arg (which limits the
number of output directories to that number). This fixes the code to
handle `max_odirs = 1|0`.

Signed-off-by: Srikrishna Iyer <sriyer@google.com>

util/dvsim/utils.py

diff --git a/util/dvsim/utils.py b/util/dvsim/utils.py
index dd2cce2..dedcf75 100644
--- a/util/dvsim/utils.py
+++ b/util/dvsim/utils.py
@@ -579,7 +579,7 @@
                   key=os.path.getctime,
                   reverse=True)
 
-    for old in dirs[max_odirs - 1:]:
+    for old in dirs[max(0, max_odirs - 1):]:
         shutil.rmtree(old, ignore_errors=True)
 
-    return dirs[0:max_odirs - 2]
+    return [] if max_odirs == 0 else dirs[:max_odirs - 1]