feat: --keep option

Signed-off-by: Lars Wirzenius <liw@liw.fi>

1 files changed, 8 insertions, 2 deletions

diff --git a/wumpus-hunter b/wumpus-hunter
index 866c15d..fc2f111 100755
--- a/wumpus-hunter
+++ b/wumpus-hunter
@@ -56,6 +56,9 @@ def parse_args():
         help="Count statistics per commit to file",
         required=True,
     )
+    p.add_argument(
+        "--keep", action="store_true", help="Keep log file of successful run"
+    )
     return p.parse_args()
 
 
@@ -135,7 +138,7 @@ def run(argv, cwd=None):
     log_output("stderr", p.stderr)
     logging.info(f"exit code {p.returncode}")
     if p.returncode != 0:
-        return Exception(f"command {argv} failed")
+        raise Exception(f"command {argv} failed")
     return p.stdout.decode()
 
 
@@ -235,7 +238,10 @@ def main():
         build(args.dir)
         run_tests(args.dir, args.test)
         record_success(args.stats, commit)
-        remove_log(args.log)
+        if args.keep:
+            rename_log(args.log, args.run_log, commit)
+        else:
+            remove_log(args.log)
     except Exception as e:
         logging.error(f"{e}", exc_info=True)
         record_failure(args.stats, commit)